python - 如何使用 PyYAML 从 Bash 中查找 YAML 值

Question

我正在测试在 RHEL7 环境中使用 PyYAML v3.12 来解析中等复杂的 YAML 配置文件的内容的可行性，方法是向它提供一个 key 并取回 key 对值。查询看起来像这样 python my_yaml_search.py key_to_search并让它打印回 value ， 例如:
所需的 bash 命令:python search_yaml.py $servername所需的响应(仅值，不是键值):myServer14到目前为止，我已经创建了以下 .py:

import sys
import yaml
key = sys.argv[1]

with open("config.yml") as f:
    try:
        data = yaml.safe_load(f)
        for k, v in data.items():
            if data[k].has_key(key):
                print data[k][v]
    
    except yaml.YAMLError as exc:
        print "Error: key not found in YAML"

配置文件:

---
server:
    servername: myServer14
    filename: testfile.zip
    location: http://test-location/1.com
    repo:
        server_name_fqdn: server.name.fqdn.com
        port: 1234

到目前为止，正在运行 python search_yaml.py $servername产生 list index out of range ; python search_yaml.py servername什么也不产生。
我是 Python/PyYAML 的新手，所以我假设我可能会错误地将变量传递给程序，而 sys 可能不是我需要的 Python 库，但是我在如何正确地做到这一点上遇到了困难 -任何输入都会挽救我的理智。

Answer 1

如果您知道要遍历的所有键，则可以执行以下操作:

import sys
import yaml

key = sys.argv[1]

with open("config.yml") as f:
    data = yaml.safe_load(f)
    n = key.count('.')
    parts = key.split('.')
    res = None
    i = 0
    while i <= n:
        try:
            if not res:
                res = data[parts[i]]
            else:
                res = res[parts[i]]
        except (yaml.YAMLError, KeyError) as exc:
            print ("Error: key not found in YAML")
            res = None
        i = i + 1
    if res:
        print(res)

测试

~# python search_yaml.py server.repo.port
~# 1234

~# python search_yaml.py server.servername
~# myServer14

这可能有错误，我编写代码只是为了看看它是否可以在没有第三方工具的情况下轻松完成。
YAML 的 CLI 应用程序
您可能对 yq 感兴趣程序。实际上有两个同名的程序，一个是用 Go 实现的，另一个是基于 Python 的(可能比上面的代码更复杂):-)
Go-based yq .
您可以安装提供的静态编译 yq来自 GitHub 发布的二进制文件或使用 yum 安装来自商业 GetPageSpeed 存储库，以便以后轻松更新:

sudo yum -y install https://extras.getpagespeed.com/release-latest.rpm
sudo yum -y install yq

然后你可以简单地:

~# yq read config.yml server.servername
~# myServer14