如何在 Python 中为 YAML 文件的特定键添加值？

Question

最近我想用我的 discord.py 机器人更多地使用 YAML 文件。 所以我目前正在考虑如何向 YAML 文件中的特定键添加值。 这是我当前的 YAML 文件：

connected_guilds:
    - 1st_guild_id

每当它连接到另一个公会时，它应该看起来像这样：

connected_guilds:
    - 1st_guild_id
    - 2nd_guild_id

等等。

这是我当前的代码：

        with open('guilds.yaml', 'r+') as guild_add:
            yaml.dump({'connected_guilds'.replace("'", ""): [guild_id]}, guild_add, indent=4)

所以基本上我认为我是否应该有一个 if 子句检查密钥是否已经存在，如果存在，请将其添加到下面，但在实现它时我仍然感到困惑。

如果有人可以提供帮助，我将不胜感激！

Answer 1

您正在打开，但未读取文件。 您创建一个新列表而不是附加到现有列表。

这是如何做到的：

with open('guilds.yaml', 'r+') as f:
  # load the content
  content = yaml.safe_load(f)
  # append the new id to the existing list
  content["connected_guilds"].append(guild_id)
  # reset the position in the file (it's at the end since we read the file)
  f.seek(0)
  # write the updated YAML to the file
  yaml.dump(content, f)
  # throw away any (old) content of the file after the current position,
  # which is at the end of the YAML we just wrote.
  # since we added more content, it's unlikely that there is more content here,
  # but not impossible!
  f.truncate()