将重复键值对附加到YAML中的嵌套字典

Question

我试图通过python脚本将重复键：值对附加到YAML文件中的嵌套字典。以下是为实现此目的而编写的代码片段：

import click
import ruamel.yaml

def organization():
    org_num = int(input("Please enter the number of organizations to be created: "))
    org_val = 0
    while org_val!= org_num:
        print ("")
        print("Please enter values to create Organizations")
        print ("")
        for org in range(org_num):
            organization.org_name = str(raw_input("Enter the Organization Name: "))
            organization.org_description = str(raw_input("Enter the Description of Organization: "))
            print ("")
            if click.confirm("Organization Name: "+ organization.org_name + "\nDescription: "+ organization.org_description + "\nIs this Correct?", default=True):
                if org_val == 0:
                    org_val = org_val + 1
                    yaml = ruamel.yaml.YAML()
                    org_data = dict(
                        organizations=dict(
                            name=organization.org_name,
                            description=organization.org_description,
                        )
                    )
                    with open('input.yml', 'a') as outfile:
                        yaml.indent(mapping=2, sequence=4, offset=2)
                        yaml.dump(org_data, outfile)

               else:
                   org_val = org_val + 1
                   yaml = ruamel.yaml.YAML()
                   org_data = dict(
                            name=organization.org_name,
                            description=organization.org_description,
                            )
                   with open('input.yml', 'r') as yamlfile:
                       cur_yaml = yaml.load(yamlfile)
                       cur_yaml['organizations'].update(org_data)

                   if cur_yaml:
                       with open('input.yml','w') as yamlfile:
                           yaml.indent(mapping=2, sequence=4, offset=2)
                           yaml.dump(cur_yaml, yamlfile)
    return organization.org_name, organization.org_description

organization()

在python脚本结束时，我的input.yml文件应如下所示：

version: x.x.x
is_enterprise: 'true'
license: secrets/license.txt
organizations:
  -  description: xyz
     name: abc
  -  description: pqr
     name: def

但是，每次脚本运行时，不会将值附加到组织，而是覆盖它。

我也尝试使用append而不是更新，但我收到以下错误：

AttributeError：'CommentedMap'对象没有属性'append'

我该怎么做才能解决这个问题？

此外，由于我不熟悉开发，任何关于使代码更好的建议都会非常有用。

Answer 1

如果我理解正确，你想要的是cur_yaml['organizations'] += [org_data]。

请注意，如果多次运行脚本，则会多次使用相同的条目。

Answer 2

使用update不会起作用，因为关键organizations的值是一个序列，并加载为list类型CommentedSeq。所以append-ing将是正确的做法。

这不起作用有点不清楚，因为你没有提供你开始的输入，也没有使用appending在AttributeError上获得CommentedMap时使用的代码。

如果您有一个组织并添加另一个组织，那么这是有效的：

import sys
import ruamel.yaml

yaml_str = """\
version: x.x.x
is_enterprise: 'true'
license: secrets/license.txt
organizations:
  -  description: xyz
     name: abc
"""

org_data = dict(
   description='pqr',
   name='def',
)

yaml = ruamel.yaml.YAML()
yaml.indent(mapping=4, sequence=4, offset=2)

cur_yaml = yaml.load(yaml_str)
cur_yaml['organizations'].append(org_data)
yaml.dump(cur_yaml, sys.stdout)

这给出了：

version: x.x.x
is_enterprise: 'true'
license: secrets/license.txt
organizations:
  - description: xyz
    name: abc
  - description: pqr
    name: def

如果您还没有组织，请确保您的输入YAML如下所示：

version: x.x.x
is_enterprise: 'true'
license: secrets/license.txt
organizations: []

在旧版本的Python上，无法保证您添加的数据中的键的顺序。要在旧版本上强制执行该订单，请执行以下操作：

org_data = ruamel.yaml.comments.CommentedMap((('description', 'pqr'), ('name', 'def')))

要么

org_data = ruamel.yaml.comments.CommentedMap()
org_data['description'] = 'pqr'
org_data['name'] = 'def'

Answer 3

现在发现问题及其正常工作。由于名称和描述是组织的列表对象，我已将[]添加到下面的代码中，它开始工作。

org_data = dict(
    organizations=[dict(
        name=tower_organization.org_name,
        description=tower_organization.org_description,
    )
    ]
)

除了上面的内容之外，我认为追加是行不通的，因为连字符' - '作为标识的一部分从第一个对象中丢失了。修复上面的代码后，append工作正常。

谢谢大家的答案。