17

我需要能够使用ConfigParser读取同一个键的多个值。示例配置文件:

[test]
foo = value1
foo = value2
xxx = yyy

在“标准”使用中ConfigParser,将有一个foo带有 value 的键value2。但我需要解析器读取这两个值。

在重复键上的条目之后,我创建了以下示例代码:

from collections import OrderedDict
from ConfigParser import RawConfigParser

class OrderedMultisetDict(OrderedDict):
    def __setitem__(self, key, value):

        try:
            item = self.__getitem__(key)
        except KeyError:
            super(OrderedMultisetDict, self).__setitem__(key, value)
            return

        print "item: ", item, value
        if isinstance(value, list):
            item.extend(value)
        else:
            item.append(value)
        super(OrderedMultisetDict, self).__setitem__(key, item)


config = RawConfigParser(dict_type = OrderedDict)
config.read(["test.cfg"])
print config.get("test",  "foo")
print config.get("test",  "xxx")

config2 = RawConfigParser(dict_type = OrderedMultisetDict)
config2.read(["test.cfg"])
print config2.get("test",  "foo")
print config.get("test",  "xxx")

第一部分(带有config)读取我们“通常”的配置文件,只留下(覆盖/删除另一个值)value2foo值,我得到以下预期输出:

value2
yyy

第二部分 ( config2) 使用我的方法将多个值附加到列表中,但输出是

['value1', 'value2', 'value1\nvalue2']
['yyy', 'yyy']

如何摆脱重复值?我期望输出如下:

['value1', 'value2']
yyy

或者

['value1', 'value2']
['yyy']

(我不介意每个值是否都在列表中......)。欢迎任何建议。

4

5 回答 5

17

经过一个小的修改,我能够实现你想要的:

class MultiOrderedDict(OrderedDict):
    def __setitem__(self, key, value):
        if isinstance(value, list) and key in self:
            self[key].extend(value)
        else:
            super(MultiOrderedDict, self).__setitem__(key, value)
            # super().__setitem__(key, value) in Python 3

config = ConfigParser.RawConfigParser(dict_type=MultiOrderedDict)
config.read(['a.txt'])
print config.get("test",  "foo")
print config.get("test",  "xxx")

输出:

['value1', 'value2']
['yyy']
于 2013-04-06T08:57:48.073 回答
5

接受的答案中断config.sections(),它总是返回一个空列表(用 Python 3.5.3 测试)。替换super(OrderedDict, self).__setitem__(key, value)super().__setitem__(key, value)修复了这个问题,但现在config.get(section, key)返回一个连接的字符串,不再是字符串列表。

我的解决方案是:

class ConfigParserMultiValues(collections.OrderedDict):

    def __setitem__(self, key, value):
        if key in self and isinstance(value, list):
            self[key].extend(value)
        else:
            super().__setitem__(key, value)

    @staticmethod
    def getlist(value):
        return value.split(os.linesep)

    config = configparser.ConfigParser(strict=False, empty_lines_in_values=False, dict_type=ConfigParserMultiValues, converters={"list": ConfigParserMultiValues.getlist})
    ...
    values = config.getlist("Section", "key") # => ["value1", "value2"]

配置 INI 文件接受重复的键:

[Section]
    key = value1
    key = value2
于 2017-08-15T09:11:34.003 回答
4

在 python 3.8 中,您还需要添加strict=False

class MultiOrderedDict(OrderedDict):
    def __setitem__(self, key, value):
        if isinstance(value, list) and key in self:
            self[key].extend(value)
        else:
            super().__setitem__(key, value)

config = ConfigParser.RawConfigParser(dict_type=MultiOrderedDict, strict=False)
config.read(['a.txt'])
print config.get("test",  "foo")
print config.get("test",  "xxx")
于 2020-04-22T13:21:20.663 回答
-1

更多示例 中的多个值test.cfg

[test]
foo = value1
foo = value2
 value3
xxx = yyy

<whitespace>value3附加value3到 foo 列表。

ConfigParser将列表转换为字符串。

/usr/lib/python2.7/ConfigParser.pyc in _read(self, fp, fpname)
    552             for name, val in options.items():
    553                 if isinstance(val, list):
--> 554                     options[name] = '\n'.join(val)
    555 

value在转换之前总是列表或字典(MultiOrderedDict)。

试试这个 - 用它,config.items工作:

from collections import OrderedDict
import ConfigParser

class MultiOrderedDict(OrderedDict):
    def __setitem__(self, key, value):
        if key in self:
            if isinstance(value, list):
                self[key].extend(value)
                return
            elif isinstance(value,str):
                return # ignore conversion list to string (line 554)
        super(MultiOrderedDict, self).__setitem__(key, value)

config = ConfigParser.RawConfigParser(dict_type=MultiOrderedDict)
config.read(['test.cfg'])
print config.get("test",  "foo")
print config.get("test",  "xxx")
print config.items("test")

输出:

['value1', 'value2', 'value3']
['yyy']
[('foo', ['value1', 'value2', 'value3']), ('xxx', ['yyy'])]

另一种实现MultiOrderedDict

class MultiOrderedDict(OrderedDict):
    def __setitem__(self, key, value):
        if key in self:
            if isinstance(value, list):
                self[key].extend(value)
                return
            elif isinstance(value,str):
                if len(self[key])>1:
                    return
        super(MultiOrderedDict, self).__setitem__(key, value)

输出:

['value1', 'value2', 'value3']
yyy
[('foo', ['value1', 'value2', 'value3']), ('xxx', 'yyy')]
于 2016-07-09T21:35:37.143 回答
-2

只需对@abarnert 的答案稍作改动,否则它会递归调用 __setitem__ ,并且由于某种原因不会停止。

ini 文件:

[section]
key1   = value1
key2[] = value21
key2[] = value22

Python:

class MultiOrderedDict(OrderedDict):
    LIST_SUFFIX = '[]'
    LIST_SUFFIX_LEN = len(LIST_SUFFIX)

    def __setitem__(self, key, value):
        if key.endswith(self.LIST_SUFFIX):
            values = super(OrderedDict, self).setdefault(key, [])
            if isinstance(value, list):
                values.extend(value)
            else:
                values.append(value)
        else:
            super(MultiOrderedDict, self).__setitem__(key, value)

    def __getitem__(self, key):
        value = super(MultiOrderedDict, self).__getitem__(key)
        if key.endswith(self.LIST_SUFFIX) and not isinstance(value, list):
            value = value.split('\n')
        return value

测试:

def test_ini(self):
    dir_path = os.path.dirname(os.path.realpath(__file__))
    config = RawConfigParser(dict_type=MultiOrderedDict, strict=False)
    config.readfp(codecs.open('{}/../config/sample.ini'.format(dir_path), encoding="utf_8_sig"))
    self.assertEquals(config.get("section1", "key1"), 'value1')
    self.assertEquals(config.get("section1", "key2[]"), ['value21', 'value22'])
于 2017-05-08T06:11:33.937 回答