yaml 和 jinja2 阅读器

Question

我希望能够在 python 中读取 YAML jinja 配置文件，就像使用 PyYAML 库一样，但我收到错误：

{% set name = "abawaca" %}
{% set version = "1.00" %}

package:
   name: {{ name }}
   version: {{ version }}

source:
   fn: {{ name }}-{{ version }}.tar.gz
   url: https://github.com/CK7/abawaca/archive/v{{ version }}.tar.gz
   sha256: 57465bb291c3a9af93605ffb11d704324079036205e5ac279601c9e98c467529

build:
   number: 0

requirements:
   build:
        - gcc   # [not osx]
        - llvm  # [osx]

Answer 1

您的输入不是有效的 YAML，您可以轻松检查，例如此处 您应该首先扩展{% %}构造，然后处理 YAML，或者您应该将文件转换为有效的 YAML。

这是选择 jinja2 的部分结果，因为该宏序列{% ... %}以在 YAML 中具有特殊含义的字符 ( {) 开头。

如果您需要更改 YAML 并重新编写它，您可以定义自己的分隔符并选择它们，以便在 YAML 中没有特殊含义。

您{% %}应该放入 YAML 注释块，因为在顶层您有一个映射，并且应该只有键值对。实现这一目标的一种方法是将开始重新定义为#% %#（您不一定必须更改结束，但我更喜欢对称性）。

然后，更新后，通过一个小脚本运行正确的 YAML，该脚本处理文件并将分隔符替换为 jinja2 可以理解的分隔符，或者调整环境，以更改 jinja2 使用的实际定义。

更正data.yaml：

#% set name = "abawaca" %#
#% set version = "1.00" %#

package:
   name: <{ name }>
   version: 42

source:
   fn: <{ name }>-<{ version }>.tar.gz
   url: https://github.com/CK7/abawaca/archive/v<{ version }>.tar.gz
   sha256: 57465bb291c3a9af93605ffb11d704324079036205e5ac279601c9e98c467529

build:
   number: 0

requirements:
   build:
      - gcc   # [not osx]
      - llvm  # [osx]

这可以通过以下方式处理：

import jinja2
from ruamel import yaml

yaml_file = 'data.yaml'
tmp_file = 'tmp.yaml'

data = yaml.round_trip_load(open(yaml_file))
data['package']['version'] = '<{ version }>'
with open(tmp_file, 'w') as fp:
    yaml.round_trip_dump(data, fp)

environment = jinja2.Environment(
    loader=jinja2.FileSystemLoader(searchpath='.'),
    trim_blocks=True,
    block_start_string='#%', block_end_string='%#',
    variable_start_string='<{', variable_end_string='}>')

    print(environment.get_template(tmp_file).render())

给予：

package:
  name: abawaca
  version: 1.00

source:
  fn: abawaca-1.00.tar.gz
  url: https://github.com/CK7/abawaca/archive/v1.00.tar.gz
  sha256: 57465bb291c3a9af93605ffb11d704324079036205e5ac279601c9e98c467529

build:
  number: 0

requirements:
  build:
  - gcc       # [not osx]
  - llvm      # [osx]

请注意，您必须使用`ruamel.yaml（免责声明：我是该包的作者），您不能使用 PyYAML 轻松地做到这一点，因为它会丢弃读取 YAML 文件的注释。由于注释中的所有 jinja2 都出现在文件的开头，因此您可以通过此特定示例解决此问题，但通常情况并非如此。