第二十二章 Python3 操作特殊类型文件

一、操作 YAML 文件

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 : ["[email protected]"]

操作

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'))

---

---

二、用配置文件解析器 configparser 操作 ini 风格的配置文件

好多计算机程序都会采用 ini 风格的文件作为程序的配置文件。

比如 mysql 的配置文件

这种风格的配置文件有以下特点

1. 把配置信息进行分类，每类配置信息采用一对中括号和包含在其内的内容来声明，如 [mysqlclient] 这成为一个部分 section 。

2. 具体的配置信息，使用 key = value 的形式表示，但是所有的 键值对都必须写在对应的 section 中。

3. 每个配置文件经过 python 的内置模块 configparser 处理后都有一个名字为 DEFAULT的默认 sections。

4. 每个键值对之间用 = 或者 : 分割

5. 文件内的注释使用 # 或者 ;

示例文件 my.cnf

[mysqld]
datadir=/var/lib/mysql
socket=/var/lib/mysql/mysql.sock
port = 3306

[mysqld_safe]
log-error=/var/log/mariadb/mariadb.log
pid-file=/var/run/mariadb/mariadb.pid

[client]
#
# include all files from the config directory
#

基本操作

confgparser 是 python 的标准库，也就是内置模块，无需安装

读操作

import configparser

cf = configparser.ConfigParser()
cf.read('my.cnf')

# 获取所有的 section
sect = cf.sections()
print(sect)

# 判断一个 section 在不在配置文件中，存在返回 True ，反之 False
print('mysqld' in sect)

# 循环所有的 section
for sec in cf:
    print(sec)

获得某一个 section 的全部键值对

可以像操作字典一样操作

kv = cf['mysqld']
for k, v in kv.items():
    print(k, v)

获取某一个 section 中一个 key 的值

val = cf['mysqld']['port']
val2 = cf['mysqld'].get('port')

print(val , val2)

支持的数据类型

配置解析器不会猜测配置文件中值的数据类型，始终将它们作为字符串存储在内部。这意味着如果您需要其他数据类型，您应该自己转换：

port = cf['mysqld']['port']

port = int(port)

由于这样的转换很常见，所以模块还内置了一些方法来直接获取到对应类型的值

prot = cf['mysqld'].getint('port')

getfloat() 可以获取到浮点类型

还提供了 getboolean()来获取布尔值，此方法不区分大小写，并从'yes'/ 'no'，'on'/'off'， 'true'/'false'和'1'/ '0' 识别布尔值。

写操作

接上例修改

cf['mysqld']['port'] = '3366'
with open('example.ini', 'w') as configfile:
    cf.write(configfile)

---

---

三、操作 EXCEL 文件

来不及了，快上车

---

四、操作 CSV 文件

来不及了，快上车