Python：读取每个键多行的配置文件

Question

我正在编写一个小型数据库测试套件，它会读取带有查询和预期结果的配置文件，例如：

query         = "SELECT * from cities WHERE name='Unknown';"
count         = 0
level         = 1
name          = "Check for cities whose name should be null"
suggested_fix = "UPDATE cities SET name=NULL WHERE name='Unknown';"

这很好用；我使用 Python 的string.partition('=') 划分每一行。

我的问题是很长的 SQL 查询。目前，我只是将这些查询粘贴为单行，这是丑陋且不可维护的。

我想找到一种优雅的 Pythonic 方式来读取表达式的右侧，即使跨越多行。

注意事项：

• 我的 SQL 查询可能包含 =
• 我不喜欢在右侧强制使用"s，因为很多现有文件都没有它。

编辑：

ConfigParser 很棒，但它迫使我在多行条目的每行开头添加一个空格或制表符。这可能会很痛苦。

提前致谢，

亚当

Answer 1

Python 标准库模块 ConfigParser 默认支持此功能。配置文件必须是标准格式：

[Long Section]
short: this is a normal line
long: this value continues
    in the next line

上面的配置文件可以用以下代码读取：

import ConfigParser
config = ConfigParser.ConfigParser()
config.read('longsections.cfg')
long = config.get('Long Section', 'long')

Answer 2

这几乎正是使我们切换到YAML 的用例（Wikipedia、python implementation、documentation；您可能想查看JSON 作为替代方案）。 YAML 比configparser 或json 有一些优势：

• 人类可读性（对于较大的文件比 JSON 更好）；
• 可以序列化任意python对象（这使得它和pickle一样不安全，但是python实现中有一个safe_load函数可以缓解这个问题）。这对于像 datetime 对象这样简单的东西已经很有用了。

为了完整起见，主要缺点（IMO）：

• Python 实现比 JSON 实现慢一个数量级；
• 跨平台的可移植性不如 JSON。

例如

import yaml

sql = """
query         : "SELECT * from cities
WHERE name='Unknown';"
count         : 0
level         : 1
name          : "Check for cities whose name should be null"
suggested_fix : "UPDATE cities SET name=NULL WHERE name='Unknown';"
"""

sql_dict = yaml.safe_load(sql)

print(sql_dict['query'])

打印

SELECT * from cities WHERE name='Unknown';

Answer 3

我建议你使用正则表达式...代码可能看起来像这样给你一个开始：

import re

test="""query = "select * from cities;"
count = 0
multine_query = "select *
from cities
     where name='unknown';"
"""

re_config = re.compile(r'^(\w+)\s*=\s*((?:".[^"]*")|(?:\d+))$', re.M)
for key, value in re_config.findall(test):
    if value.startswith('"'):
        value = value[1:-1]
    else:
        value = int(value)
    print key, '=', repr(value)

这个例子的输出是：

~> python test.py 
query = 'select * from cities;'
count = 0
multine_query = "select *\nfrom cities\n     where name='unknown';"

希望有帮助！

问候， 克里斯托夫