Python之读取配置文件
Python爬虫与算法
共 1115字，需浏览 3分钟
·
2021-11-13 09:55
在大公司工作了一段时间，其中的感悟之一就是好的代码能产生更大的价值，便于后来者维护。而写好代码，其中一个trick便是学会设计代码，用配置文件去控制流程，使整个项目简洁明了。
本文将介绍如何使用Python来读取配置文件，如ini、yaml文件等。
ini配置文件
我们以如下enviro.ini配置文件为例，使用Python的configparser模块来解析参数。
[mysql]
host=127.0.0.1
port=33060
username=root
password=123456
db=school
解析上述配置文件的示例代码如下：
# -*- coding: utf-8 -*-
from configparser import ConfigParser
conf = ConfigParser()
conf.read("enviro.ini")
host = conf.get("mysql", "host")
port = conf.getint("mysql", "port")
print(host, type(host))
print(port, type(port))
输出结果如下：
127.0.0.1 <class 'str'>
33060 <class 'int'>
yaml配置文件
yaml格式也常用于配置文件，我们以如下enviro.yaml为例，使用Python的yaml模块来解析参数。
mysql:
- host: 127.0.0.1
- port: 33060
- username: root
- password: 123456
- db: school
解析上述配置文件的示例代码如下：
# -*- coding: utf-8 -*-
import yaml
with open("enviro.yaml", encoding="utf-8") as yaml_file:
data = yaml.safe_load(yaml_file)
print(data["mysql"])
解析结果如下：
[{'host': '127.0.0.1'}, {'port': 33060}, {'username': 'root'}, {'password': 123456}, {'db': 'school'}]
总结
最近忙于工作，好久没有写文章了，先写点简单的上上手，后续有机会再分享这阵子工作所获得的经验~
评论