為什么要寫配置文件

在開發過程中，我們常常會用到一些固定參數或者是常量。對于這些較為固定且常用到的部分，往往會將其寫到一個固定文件中，避免在不同的模塊代碼中重復出現從而保持核心代碼整潔。這個固定文件我們可以直接寫成一個.py文件，例如settings.py或config.py，這樣的好處就是能夠在同一工程下直接通過import來導入當中的部分；但如果我們需要在其他非 Python 的平臺進行配置文件共享時，寫成單個.py就不是一個很好的選擇。這時我們就應該選擇通用的配置文件類型來作為存儲這些固定的部分。目前常用且流行的配置文件格式類型主要有ini、json、toml、yaml、xml等，這些類型的配置文件我們都可以通過標準庫或第三方庫來進行解析。

ini

ini即 Initialize 初始化之意，早期是在 Windows 上配置文件的存儲格式。ini文件的寫法通俗易懂，往往比較簡單，通常由節（Section）、鍵（key）和值（value）組成，就像以下形式：

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

Python 本身內置的configparser標準庫，我們直接就可以用來對ini文件進行解析。如我們將上述內容保存在一個名為db.ini的文件中，然后使用read()方法來進行解析和讀取，最后通過items()方法來獲取指定節點下的所有鍵值對。

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

需要注意的是，configparser默認將值以字符串的形式呈現，所以這也就是為什么我們在db.ini文件中沒有加引號而是直接將字面量寫在上面的原因。獲取到鍵值對后，我其實直接就將其轉換成字典，然后通過解包的方式進行穿參，保持代碼簡潔：

#!pipinstallpymysql
importpymysql
fromconfigparserimportConfigParser

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

con=pymysql.connect(**db_cfg)

json

json格式可以說是我們常見的一種文件形式了，也是目前在互聯網較為流行的一種數據交換格式。除此之外，json有時也是配置文件的一種。比如npm（JavaScript 包管理工具類似 Python 的pip）、以及微軟出品的目前被廣泛使用的 VSCode 編輯器，都使用json編寫配置參數。和configparser一樣，Python 也內置了json標準庫，可以通過load()和loads()方法來導入文件式和字符串的json內容。

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

我們將上述內容保存為db.json后進行讀取和解析，json庫讀取 json 文件相對簡單容易，而且很容易解析成 Python 的字典對象。

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

使用json文件配置的缺點就是語法標準嚴格限制，為人所詬病之一的就是無法在當中寫注釋，除非采取json類型的其他超集作為替代方案（VSCode 中能寫注釋的json參數配置文件便是代替方案的一種）；同時存在嵌套過深的問題，容易導致出錯，不宜用來寫過長或復雜的參數配置信息。

toml

toml格式（或tml格式）是 Github 聯合創始人 Tom Preston-Werner 所提出的一種配置文件格式。根據維基百科的資料，toml最開始提出時是在 2013年7月份，距今已有七年時間；它在某些方面也與后面要談到的yaml文件有些類似，但如果當你知道 yaml 的規范有幾十頁（沒有錯，真的就是幾十頁……）的時候，可能你真的就不太愿意去寫那么復雜的配置文件，toml格式則倒是個不錯的選擇。toml格式大致如下：

01-toml樣式從這里可以看出toml有點類似于前面所講的ini文件。但是它比ini擴展了更多的內容。在樣例圖片中我們可以看到，除了基本的字符串以外，例如時間戳、布爾值、數組等都進一步支持，而且樣式和 Python 的原生寫法十分類似。當然這里不會過多介紹toml格式的一些規范說明，有人已經對官方的規范文檔進行了翻譯，有興趣的朋友可以直接查閱。這么契合 Python 方式的配置文件類型已經有開發者造出了相應的「輪子」，目前在 Github 上 Stars 數最多的是則是uiri/toml的版本，不過該版本僅通過了 v0.5 版本toml規范，但在使用上還是蠻簡潔的，我們可以通過pip命令進行安裝

pipinstalltoml

該庫的解析方式很簡單，也有點類似于json庫的解析用法，即通過load()或loads()來進行解析；同理轉換并導出也是同樣類似的用法。比如我們現在將以下內容寫入到config.toml中：

[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

yaml格式（或yml格式）是目前較為流行的一種配置文件，它早在 2001 由一個名為 Clark Evans 的人提出；同時它也是目前被廣泛使用的配置文件類型，典型的就是 Docker 容器里的docker-compose.yml配置文件，如果經常使用 Docker 進行部署的人對此不會陌生。yaml文件的設計從 Python、XML 等地方獲取靈感，所以在使用時能很清楚地看到這些部分的影子。在上一節toml內容里我曾提到，yaml的規范內容可以說是冗長和復雜，足足有80頁之多（斗尊強者，恐怖如斯……）。

02-yaml規范頁數所以感興趣的朋友可以再自行了解相關用法。YAML 官方早已經提供了相應的 Python 庫進行支持，即PyYAML；當然也同樣需要我們事先進行安裝：

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庫的解析結果基本一致。

結尾

本文列舉了一些主流且常見的配置文件類型及其 Python 的讀取方法，可能有的讀者會發現當中沒有xml格式類型的內容。對于xml配置文件可能與 Java 系語言打交道的朋友遇見得會多一些，但xml文件的可讀性實在是讓人望而生畏；對xml文件不了解的朋友可以使用 Chrome 瀏覽器隨便進入一個網站然后按下 F12 進入開發者后查看那密密麻麻的 html 元素便是.xml的縮影。除了這些主流的配置文件類型之外，像一些.cfg、.properties等都可以作為配置文件，甚至和開頭提到的那樣，你單獨用一個.py文件來書寫各類配置信息作為配置文件進行導入都是沒問題，只是在跨語言共享時可能會有些障礙。因此本文就不過多介紹，感興趣的朋友可以進一步自行了解。在本文里列舉的配置文件類型其復雜性由上到下依次增加：ini < json ≈ toml < yaml，它們之間各有優劣，可以根據自己實際的需求和團隊協作要求來具體選擇。

編輯：jq

聲明：本文內容及配圖由入駐作者撰寫或者入駐合作網站授權轉載。文章觀點僅代表作者本人，不代表電子發燒友網立場。文章及其配圖僅供工程師學習之用，如有內容侵權或者其他違規問題，請聯系本站處理。舉報投訴