1. 程式人生 > >Python中配置檔案的使用

Python中配置檔案的使用

配置檔案的讀取示例

a) 配置檔案中包含一個或多個 section, 每個 section 有自己的 option;

b) section 用 [sect_name] 表示,每個option是一個鍵值對,使用分隔符 = 或 : 隔開;

c) 在 option 分隔符兩端的空格會被忽略掉

d) 配置檔案使用 # 和 ; 註釋

下面給出一個配置檔案的例子。

[mongodb]
host = amaster

[kafka]
bootstrap_servers = amaster:9092,anode1:9092,anode2:9092

[log]
log_file_name = kuaidaili_spider.log

[zookeeper]
conn_str = amaster:2181,anode1:2181,anode2:2181

測試程式碼:

#-*- coding: utf-8 -*-
import ConfigParser


def func():
    cp = ConfigParser.SafeConfigParser()
    cp.read('kuaidaili_spider.cfg')

    print 'mongodb host',cp.get('mongodb','host')
    print 'kafka bootstrap_servers',cp.get('kafka','bootstrap_servers')
    print 'log file name',cp.get('log','log_file_name')
    print 'zookeeper conn string',cp.get('zookeeper','conn_str')

if __name__ == '__main__':
    func()

輸出結果:

mongodb host amaster
kafka bootstrap_servers amaster:9092,anode1:9092,anode2:9092
log file name kuaidaili_spider.log
zookeeper conn string amaster:2181,anode1:2181,anode2:2181

Unicode 編碼的配置

配置檔案如果包含 Unicode 編碼的資料,需要使用 codecs 模組以合適的編碼開啟配置檔案。

myapp.conf

[msg]
hello = 你好
config_parser_unicode.py


import ConfigParser
import codecs

cp = ConfigParser.SafeConfigParser()
with codecs.open('myapp.conf', 'r', encoding='utf-8') as f:
    cp.readfp(f)

print cp.get('msg', 'hello')

allow_no_value

通常情況下, option 是一個鍵值對。但是,當 SafeConfigParser 的引數 allow_no_value 設定成 True 時,它允許 option 不設定值而只是作為一個標識。

allow_no_value.conf

# option as Flag
[flag]
flag_opt

allow_no_value.py

import ConfigParser

cp = ConfigParser.SafeConfigParser(allow_no_value = True)
cp.read('myapp.conf')
print cp.get('flag', 'flag_opt');    # None

allow_no_value 預設設定成 False,此時如果配置檔案中存在沒有設定值的 option,在讀取配置檔案時將丟擲異常

ConfigParser.ParsingError。當 allow_no_value 設定成 True 時,如果一個 option 沒有設定值,has_option 方法會返回 True,get 方法會返回 None。

DEFAULT section

如果配置檔案中存在一個名為 DEFAULT 的 section,那麼其他 section 會擴充套件它的 option 並且可以覆蓋它的 option。

db.conf

[DEFAULT]
host = 127.0.0.1
port = 3306

[db_root]
user = root
pass = root

[db_huey]
host = 192.168.1.101
user = huey
pass = huey

default_section.py

print cp.get('db_root', 'host')    # 127.0.0.1
print cp.get('db_huey', 'host')    # 192.168.1.101

插值 Interpolation

SafeConfigParser 提供了插值的特性來結合資料。

url.conf

[DEFAULT]
url = %(protocol)s://%(server)s:%(port)s/

[http]
protocol = http
server = localhost
port = 8080

[ftp]
url = %(protocol)s://%(server)s/
protocol = ftp
server = 192.168.1.102

interpolation_demo.py

import ConfigParser

cp = ConfigParser.SafeConfigParser()
cp.read('url.conf')

print cp.get('http', 'url')    # http://localhost:8080/
print cp.get('ftp', 'url')     # ftp://192.168.1.102/