为什么要写配置文件

ini

[localdb]
host     = 127.0.0.1
user     = root
password = 123456
port     = 3306
database = mysql

>>>fromconfigparserimportConfigParser
>>>cfg=ConfigParser()
>>>cfg.read("/Users/Bobot/db.ini")
['/Users/Bobot/db.ini']
>>>cfg.items("localdb")
[('host','127.0.0.1'),('user','root'),('password','123456'),('port','3306'),('database','mysql')]

#!pipinstallpymysql
importpymysql
fromconfigparserimportConfigParser

cfg=ConfigParser()
cfg.read("/Users/Bobot/db.ini")
db_cfg=dict(cfg.items("localdb"))

con=pymysql.connect(**db_cfg)

json

{
"localdb":{
"host":"127.0.0.1",
"user":"root",
"password":"123456",
"port":3306,
"database":"mysql"
}
}

>>>importjson
>>>frompprintimportpprint
>>>
>>>withopen('/Users/Bobot/db.json')asj:
...cfg=json.load(j)['localdb']
...
>>>pprint(cfg)
{'database':'mysql',
'host':'127.0.0.1',
'password':'123456',
'port':3306,
'user':'root'}

toml

pipinstalltoml

[mysql]
host     = "127.0.0.1"
user     = "root"
port     = 3306
database = "test"

  [mysql.parameters]
  pool_size = 5
  charset   = "utf8"

  [mysql.fields]
  pandas_cols = [ "id", "name", "age", "date"]

紧接着我们就可以通过toml库中的load()方法来进行读取：

>>>importtoml
>>>importos
>>>frompprintimportpprint
>>>cfg=toml.load(os.path.expanduser("~/Desktop/config.toml"))
>>>pprint(cfg)
{'mysql':{'database':'test',
'fields':{'pandas_cols':['id','name','age','date']},
'host':'127.0.0.1',
'parameters':{'charset':'utf8','pool_size':5},
'port':3306,
'user':'root'}}

可以看到toml文件被间接地转化成了字典类型，当然这也就是json版的写法（将单引号替换成双引号即可），方便我们后续调用或者传参。

yaml

pipinstallpyyaml

同json库和toml库一样，通过load()方法来进行加载。

需要注意的是，使用load()方法会存在一定的安全隐患，从思科 Talos 的这份报告中我们可以看到，如果加载了未知或不信任的yaml文件，那么有可能会存在被攻击的风险和网络安全隐患，因为它能够直接调用相应的 Python 函数来执行为攻击者所需要的命令，比如说在yaml文件中写入这么一段：

#使用Linux和macOS的朋友不要轻易尝试
!!python/object/apply:os.system["rm-rf/"]

因此最好是使用safe_load()来代替load()方法。

这和 Python 内置的string标准库中Template类的substitute()模板方法一样存在着同样的安全隐患，所以使用safe_substitute()来替代是一样的道理。

如我们现在将之前的一些配置信息写入config.yaml文件中：

mysql:
host:"127.0.0.1"
port:3306
user:"root"
password:"123456"
database:"test"

parameter:
pool_size:5
charset:"utf8"

fields:
pandas_cols:
-id
-name
-age
-date

然后我们通过safe_load()方法进行解析：

>>>importos
>>>frompprintimportpprint
>>>
>>>withopen(os.path.expanduser("~/config.yaml"),"r")asconfig:
...cfg=yaml.safe_load(config)
...
>>>pprint(cfg)
{'mysql':{'database':'test',
'fields':{'pandas_cols':['id','name','age','date']},
'host':'127.0.0.1',
'parameter':{'charset':'utf8','pool_size':5},
'password':'123456',
'port':3306,
'user':'root'}}

可以看到最后结果和前面的toml库的解析结果基本一致。

结尾