时间戳

首页 > 解决方案 > 如何使用 PyYAML 从 Bash 中查找 YAML 值

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

问题描述

我正在测试在 RHEL7 环境中使用 PyYAML v3.12 来解析适度复杂的 YAML 配置文件内容的可行性，方法是为其提供一个密钥并取回密钥对值。查询看起来像这样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"

配置.yml：

---
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 库，但是我在如何正确地做到这一点上遇到了障碍 -任何输入都会挽救我的理智。

标签： pythonbashshellrhelpyyaml

解决方案

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

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 的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

推荐阅读