一、操作 YAML 文件

本文发布地址: https://www.jianshu.com/p/522ab1e4f2f1

yaml

和 json 类似，YAML(http://www.yaml.org)同样有键和值，但主要用于处理日期和时间这样的数据类型。这个是属于第三方模块，需要自行安装(https://pyyaml.org/wiki/PyYAML)。

安装

下载源码包PyYAML-3.12.tar.gz并解压缩。进入到目录PyYAML-3.12并运行

$ python3 setup.py install

使用

import yaml

yaml.load() 将YAML 字符串转为Python的数据类型，而yaml.domp()则正好相反。

类似于 true 、false、on 和 off 的值可以转换为 Python 的布尔值。

整数和字符串转换为 Python等价的。其他语法创建为列表和字典：

官方文档： https://pyyaml.org/wiki/PyYAMLDocumentation

示例 yaml 文件

# scanhosts.yaml

#####
#扫描主机配置信息
#####
hostsinfo :
# 主机段，EXP:['192.168.6'] or ['192.168.1',192.168.2]
    nets : ['192.168.1']
# 端口段
    ports : "22"
# SSH private key 文件列表
    ssh_key_file : ['/Users/yanshunjun/.ssh/id_rsa']
# 执行命令获取系统相关信息
    syscmd_list : ["cat /etc/redhat-release",
                   "hostname",
                    "cat /sys/class/net/[^vtlsb]*/address",
                    "dmidecode -s system-serial-number",
                    "dmidecode -s system-manufacturer",
                    "dmidecode -s system-product-name"]
#####
# SSH登录密码列表
#####
    ssh_pass : ['upsa']
# 黑名单查询
    black_list : ['192.168.1.108','192.168.1.1']
# Email收件人列表, 将扫描结果等信息,发送到下面的目标邮箱
    email_list : ["dockerhub@163.com"]

操作

import yaml
In [20]: ya = yaml.safe_load(open("./scanhosts.yaml"))

In [21]: ya['hostsinfo']['ports']
Out[21]: 22

下面的例子是接上例，写入

hosts_info_dic["hostsinfo"]['nets'][0]='172.16.153'

# print(hosts_info_dic)
yaml.safe_dump( hosts_info_dic,open('new_hostsinfo.yml','w'))