通过使用python从yaml文件中读取数据来创建sql文件

Question

我是python的新手。 我有包含如下数据的 yaml 文件 (config.yaml)。

Config:
    - name: identity
      value: 101
    - name: size
      value: 201
    - name: logloc
      value: /tmp/log
    - name: host
      value: xyz.dev.cloud

我正在寻找 python 脚本来读取上面的 yaml 文件，并为每个名称和值创建带有更新语句的 sql 文件，如下所示。

UPDATE SAMPLE SET PVALUE=<value> WHERE PNAME=<name>

提前致谢。

Answer 1

我相信，这会做你想做的事：

import yaml

with open('/tmp/config.yaml') as f:
    data = yaml.load(f, Loader=yaml.SafeLoader)

template = "UPDATE SAMPLE SET PVALUE={} WHERE PNAME={}"
statements = [template.format(nv['value'], nv['name']) for nv in data['Config']]

with open("/tmp/update_samples.sql", 'w') as f:
    for statement in statements:
        f.write(statement + '\n')

/tmp/update_samples.sql内容：

UPDATE SAMPLE SET PVALUE=101 WHERE PNAME=identity
UPDATE SAMPLE SET PVALUE=201 WHERE PNAME=size
UPDATE SAMPLE SET PVALUE=/tmp/log WHERE PNAME=logloc
UPDATE SAMPLE SET PVALUE=xyz.dev.cloud WHERE PNAME=host