如何检查yaml文件Python中是否存在密钥

Question

我正在尝试检查yaml文件中是否存在用户名

我的文件格式如下：

random_uuid:
username: ""
random_uuid:
username: ""
random_uuid:
username: ""

我仍然不确定如何在谷歌上搜索后做到这一点，因为它必须查看所有的uuid

Answer 1

您可以使用yaml(pyyaml(模块，并将加载视为python dict。

import yaml
document = """
123:
a: ''
b: ''
c: ''
456:
a: ''
b: ''
c: ''
"""
yaml.dump(yaml.safe_load(document))
doc = yaml.safe_load(document)
for key in (123, 456, 789):
try:
print(f'{key} - {doc[key]} found')
except KeyError:
print(f'{key} - not found')

Answer 2

如果这是您的YAML文件的结构，只需迭代所有值，直到找到您要查找的内容。假设文件在test.yaml:中

#!/usr/bin/env python
import yaml
FILE_PATH = "test.yaml"
LOOKING_FOR_THIS_USER = "Shay"
with open(FILE_PATH, "r") as stream:
try:
loaded_yaml = yaml.safe_load(stream)
for k, v in loaded_yaml.items():
print(f"{k=}: {v=}")
if v['username'] == LOOKING_FOR_THIS_USER:
print(f"Found it! {k} is the user we looked for.")
except yaml.YAMLError as exc:
print(exc)