Python配置文件常用的方法

在日常开发小脚本时，经常要使用配置文件，以下是我在日常开发中总结的自己的常用的使用配置文件的方法：

config.conf

1. 核心用法

def get_config(self, section, key):
config = ConfigParser.ConfigParser()
path = os.path.split(os.path.realpath(__file__))[0] + '/config.conf'
config.read(path)
return config.get(section, key)

2. 使用示例

这个应该是我最常用的方案，python自带了ConfigParser，能够解析以下格式文件：

# ./config.conf

[api_filter_db]
dbhost = xxx.xxx.xxx.xxx
dbport = 3306
dbname = yyy
dbuser = root
dbpassword = zzz
dbcharset = utf8

[mshow_db]
dbhost = xxx.xxx.xxx.xxx
dbport = 3306
dbname = yyy
dbuser = root
dbpassword = zzz
dbcharset = utf8

# ./get_config.py
class Utils(object):
def get_config(self, section, key):
config = ConfigParser.ConfigParser()
path = os.path.split(os.path.realpath(__file__))[0] + '/config.conf'
config.read(path)
return config.get(section, key)

#可以引用get_config函数
class DoSomething(Utils):
conn = MySQLdb.connect(
host=self.get_config('api_filter_db', 'dbhost'),
port=int(self.get_config('api_filter_db', 'dbport')),
user=self.get_config('api_filter_db', 'dbuser'),
passwd=self.get_config('api_filter_db', 'dbpassword'),
db=self.get_config('api_filter_db', 'dbname'),
charset=self.get_config('api_filter_db', 'dbcharset')
)
cursor = conn.cursor()

YAML

YAML也是一种比较常用的配置文件方案，平时用ansible会用的比较多，要解析yaml文件，需要安装一个第三方的包：yaml

# pip install yaml

1. 核心用法

# ./get_config.py
class Utils(object):
def get_config_yaml(self, section):
path = os.path.split(os.path.realpath(__file__))[0] + '/config.yaml'
config = yaml.load(file(config)) # {'mail': ['xx@zz.cn', 'yy@zz.cn'], 'phone': [1111111, 2222222, 3333333]}
return config.get(section)

2. 使用示例

YAML能够解析以下格式文件：

mail:
- xx@zz.cn
- yy@zz.cn

phone:
- 1111111
- 2222222
- 3333333

yaml解析后生成字典，内部数据组成一个数组：

{'mail': ['xx@zz.cn', 'yy@zz.cn'], 'phone': [1111111, 2222222, 3333333]}

函数根据提供的参数返回其中的数据组值。