Question

我已经能够将YAML文档中的数据加载到配置文件中。

YAML：

id: 1
x: key1
y: key2

Python脚本：

#Import necessary functions from Jinja2 module
from jinja2 import Environment, FileSystemLoader

#Import YAML module
import yaml

import sys

ALLOWED_ARGS = set(['template', 'data'])

def print_usage():
    print('Wrong arguments: Usage: configplate.py --template=<template.jinja> --data=<data.yaml>')

def load_args(args):
    out = {}
    for arg in args:
        if arg == __file__: #ignore the filename argument
            continue
        args_splited = arg.split('=')
        out[args_splited[0].lstrip('-')] = args_splited[1]
    if set(out.keys()) != set(ALLOWED_ARGS):
        print_usage()
        raise ValueError('Required argument not present.')
    return out

def make(template, data):
    #Load data from YAML into Python dictionary
    config_data = yaml.load(open(data))

    #Load Jinja2 templates
    env = Environment(loader = FileSystemLoader('.'), trim_blocks=True, lstrip_blocks=True)
    template = env.get_template(template)

    #Render the template with data and print the output
    return template.render(config_data)

if __name__ == '__main__':
    if len(sys.argv) != 3:
        print_usage()
    else:
        args = load_args(sys.argv)
        res = make(args['template'], args['data'])
        print(res)

配置文件：

<add name="Cloud" connectionString="{{ x }}" />

我现在面临的问题是我无法将相同的YAML加载到多个配置文件中。我有第二个配置文件，我想同时填写。

配置文件2：

<add name="Cloud" connectionString="{{ y }}" />

如何在同一个python脚本中添加它？我也不需要在控制台上设置配置文件，我可以简单地将它们的路径放入脚本中。

Answer 1

由于您要处理多个模板，因此需要使load_args能够处理多个--template=参数/选项。我调整了你的load_args以返回一个带有列表值的dict，但是对允许的参数数量的检查（多个模板，单个YAML文件）是粗略的。

对于处理命令行，您应该查看标准arparse模块，您可以在其中指示每个参数/选项可以多次出现。

由于您要重用YAML中的数据，因此只能在调用make()之外加载一次。 没有理由不使用 safe_load() 而不是记录在案的不安全 load()。

使用：

#Import necessary functions from Jinja2 module
from jinja2 import Environment, FileSystemLoader

#Import YAML module
import yaml

import sys

ALLOWED_ARGS = set(['template', 'data'])

def print_usage():
    print('Wrong arguments: Usage: configplate.py --template=<template.jinja> --data=<data.yaml>')

def load_args(args):
    out = {}
    for arg in args:
        if arg == __file__: #ignore the filename argument
            continue
        args_split = arg.split('=', 1)
        out.setdefault(args_split[0].lstrip('-'), []).append(args_split[1])
    if (set(out.keys()) != set(ALLOWED_ARGS)) or len(out['data']) != 1:
        print_usage()
        raise ValueError('Required argument not present.')
    return out

def make(template, data):
    #Load Jinja2 templates
    env = Environment(loader = FileSystemLoader('.'), trim_blocks=True, lstrip_blocks=True)
    template = env.get_template(template)

    #Render the template with data and print the output
    return template.render(data)

sys.argv=[
    __file__,
    '--template=config1.jinja',
    '--template=config2.jinja',
    '--data=data.yaml'
]

if __name__ == '__main__':
    if len(sys.argv) < 4:
        print_usage()
    else:
        args = load_args(sys.argv)
        # Load data from YAML into Python dictionary **once** and **safely**
        config_data = yaml.safe_load(open(args['data'][0]))
        for template_file in args['template']:
            res = make(template_file, config_data)
            print(res)

你得到的相应文件：

<add name="Cloud" connectionString="key1" />
<add name="Cloud" connectionString="key2" />

导入argparse后，您可以删除AllOWED_ARGS，print_usage和load_args，然后执行以下操作：

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--template', action='append', nargs='+', required=True)
    parser.add_argument('--data', required=True)
    args = parser.parse_args()

    # Load data from YAML into Python dictionary **once** and **safely**
    config_data = yaml.safe_load(open(args.data))
    for template_file in args.template:
        res = make(template_file, config_data)
        print(res)

（BTW查找to split的共轭，没有“分裂”（或“分裂”）的东西，即“已被拆分”的东西，）

使用Jinja2的多个配置文件的YAML序列化数据

1 个答案: