python - 将 Python 与 YAML 结合使用

Question

我有一个 YAML 文件，我想在游戏中存储不同的“玩家”。

YAML 文件如下所示:

Person:
  Name:
  Age:
  Nationality:
  Footed:
  Position:

创建播放器后，YAML 文件应如下所示:

Person:
   Name: Rich
   Age: 23 
   Nationality: British
   Footed: Right
   Position: Forward

到目前为止我的代码是这样的:

import yaml

name = input('What is your name? ')
age = int(input('What is your age? '))
nationality = input('What is your nationality? ')
footed = input('What foot? ')
position = input('What is your position? ')

with open('test.yml', 'a') as outfile:
    outfile.write(yaml.dump({'Name' : name, 'Age' : age, 'Nationality' : nationality,
    'Footed' : footed, 'Position' : position}))

但是当我运行它并给用户输入时，yaml 文件最终看起来像:

{Age: 2, Footed: r, Name: r, Nationality: r, Position: r}

如何将其添加到 YAML 文件而不是追加，以及如何垂直而不是水平构建它？最后，如果我想添加 10/20/n 数量的玩家，我希望 YAML 文件将它们全部存储在彼此下面，以便我可以单独调用每个玩家

Answer 1

当您像这样编写 YAML 时，首先将数据结构写入内存中的文件，然后以字符串形式检索内存文件内容，然后将其写入文件。这是低效且缓慢的。

您还应该读取您拥有的 YAML 文件，更新数据结构并将其转储:

import yaml

file_name = 'test.yml'

with open(file_name) as infile:
    data = yaml.load(infile)

person = data['Person']
person['name'] = input('What is your name? ')
person['age'] = int(input('What is your age? '))
person['nationality'] = input('What is your nationality? ')
person['footed'] = input('What foot? ')
person['position'] = input('What is your position? ')

with open(file_name, 'w') as outfile:
    yaml.dump(data, stream=outfile, default_flow_style=False, indent=3)

default_flow_style 参数可确保您的键值对列在彼此下方。

使用 PyYAML，内文件中的任何注释都将丢失，并且映射中键的顺序可能会被打乱。如果这是一个问题，我建议您使用 ruamel.yaml 包(免责声明:我是该包的作者)，并将代码更改为:

import ruamel.yaml

file_name = 'test.yml'

with open(file_name) as infile:
    data = ruamel.yaml.round_trip_load(infile)

person = data['Person']
person['name'] = input('What is your name? ')
person['age'] = int(input('What is your age? '))
person['nationality'] = input('What is your nationality? ')
person['footed'] = input('What foot? ')
person['position'] = input('What is your position? ')

with open(file_name, 'w') as outfile:
    yaml.round_trip_dump(data, stream=outfile, indent=3)

如果你想存储多个玩家。确保您的顶级数据结构是一个序列，或者是从某个唯一值(例如人名)映射的。在这种情况下，使用不同的输入文件作为模板，并通过读入更新输出文件，并将其附加到列表 cq 中。更新字典并写出文件。只要名称是唯一的，使用顶级映射/字典就可以更轻松地做到这一点。