使用ConfigParser读取没有节名的文件

问题描述 投票:0回答:7

我正在使用

ConfigParser
来读取脚本的运行时配置。

我希望能够灵活地不提供部分名称(有些脚本足够简单;它们不需要“部分”)。

ConfigParser
将抛出
NoSectionError
异常,并且不会接受该文件。

如何让 ConfigParser 简单地检索没有节名称的配置文件的

(key, value)
元组?

例如:

key1=val1
key2:val2

我宁愿不写入配置文件。

python parsing configuration-files
7个回答
62
投票

您只需一行代码即可完成此操作。

(但 tomllib 可能是更好的选择。请参阅我答案的底部。)

在 python 3 中,在配置文件数据前面添加一个假的节头,并将其传递给

read_string()

from configparser import ConfigParser

parser = ConfigParser()
with open("foo.conf") as stream:
    parser.read_string("[top]\n" + stream.read())  # This line does the trick.

您还可以使用

itertools.chain()
来模拟
read_file()
的节标题。这可能比上述方法更节省内存,如果您在受限的运行时环境中有大型配置文件,这可能会有所帮助。

from configparser import ConfigParser
from itertools import chain

parser = ConfigParser()
with open("foo.conf") as lines:
    lines = chain(("[top]",), lines)  # This line does the trick.
    parser.read_file(lines)

在 python 2 中,在配置文件数据前面添加一个假节头,将结果包装在

StringIO
对象中,并将其传递给
readfp()

from ConfigParser import ConfigParser
from StringIO import StringIO

parser = ConfigParser()
with open("foo.conf") as stream:
    stream = StringIO("[top]\n" + stream.read())  # This line does the trick.
    parser.readfp(stream)

使用任何这些方法,您的配置设置都将在

parser.items('top')
中可用。

您也可以在 python 3 中使用 StringIO,也许是为了与新旧 python 解释器兼容,但请注意,它现在位于

io
包中,并且
readfp()
现已弃用。

或者,您可以考虑使用 TOML 解析器而不是 ConfigParser。 Python 3.11 在标准库中添加了解析器:tomllib


58
投票

Alex Martelli 提供了一个解决方案,使用

ConfigParser
来解析
.properties
文件(显然是无节配置文件)。

他的解决方案是一个类似文件的包装器,它将自动插入一个虚拟节标题以满足

ConfigParser
的要求。


53
投票

jterrace的回答的启发,我想出了这个解决方案:

  1. 将整个文件读入字符串
  2. 带有默认部分名称的前缀
  3. 使用 StringIO 来模仿类似文件的对象
ini_str = '[root]\n' + open(ini_path, 'r').read()
ini_fp = StringIO.StringIO(ini_str)
config = ConfigParser.RawConfigParser()
config.readfp(ini_fp)


为未来的谷歌用户编辑:从Python 3.4+开始,

readfp
已被弃用,并且不再需要
StringIO
。相反,我们可以直接使用
read_string

with open('config_file') as f:
    file_content = '[dummy_section]\n' + f.read()

config_parser = ConfigParser.RawConfigParser()
config_parser.read_string(file_content)

20
投票

您可以使用 ConfigObj 库来简单地做到这一点:http://www.voidspace.org.uk/python/configobj.html

更新:在此处查找最新代码。

如果您使用的是 Debian/Ubuntu,您可以使用包管理器安装此模块:

apt-get install python-configobj

使用示例:

from configobj import ConfigObj

config = ConfigObj('myConfigFile.ini')
config.get('key1') # You will get val1
config.get('key2') # You will get val2

10
投票

我认为,最简单的方法是使用 python 的 CSV 解析器。这是演示此方法的读/写函数以及测试驱动程序。如果值不允许是多行的话,这应该可以工作。 :)

import csv
import operator

def read_properties(filename):
    """ Reads a given properties file with each line of the format key=value.  Returns a dictionary containing the pairs.

    Keyword arguments:
        filename -- the name of the file to be read
    """
    result={ }
    with open(filename, "rb") as csvfile:
        reader = csv.reader(csvfile, delimiter='=', escapechar='\\', quoting=csv.QUOTE_NONE)
        for row in reader:
            if len(row) != 2:
                raise csv.Error("Too many fields on row with contents: "+str(row))
            result[row[0]] = row[1] 
    return result

def write_properties(filename,dictionary):
    """ Writes the provided dictionary in key-sorted order to a properties file with each line of the format key=value

    Keyword arguments:
        filename -- the name of the file to be written
        dictionary -- a dictionary containing the key/value pairs.
    """
    with open(filename, "wb") as csvfile:
        writer = csv.writer(csvfile, delimiter='=', escapechar='\\', quoting=csv.QUOTE_NONE)
        for key, value in sorted(dictionary.items(), key=operator.itemgetter(0)):
                writer.writerow([ key, value])

def main():
    data={
        "Hello": "5+5=10",
        "World": "Snausage",
        "Awesome": "Possum"
    }

    filename="test.properties"
    write_properties(filename,data)
    newdata=read_properties(filename)

    print "Read in: "
    print newdata
    print

    contents=""
    with open(filename, 'rb') as propfile:
        contents=propfile.read()
    print "File contents:"
    print contents

    print ["Failure!", "Success!"][data == newdata]
    return

if __name__ == '__main__': 
     main() 

6
投票

我自己也遇到了这个问题,我为 ConfigParser (Python 2 中的版本)编写了一个完整的包装器,它可以透明地读取和写入没有节的文件,基于 Alex Martelli 的方法(链接到已接受的答案)。它应该是 ConfigParser 任何用法的直接替代品。发布它以防任何需要的人找到此页面。

import ConfigParser
import StringIO

class SectionlessConfigParser(ConfigParser.RawConfigParser):
    """
    Extends ConfigParser to allow files without sections.

    This is done by wrapping read files and prepending them with a placeholder
    section, which defaults to '__config__'
    """

    def __init__(self, *args, **kwargs):
        default_section = kwargs.pop('default_section', None)
        ConfigParser.RawConfigParser.__init__(self, *args, **kwargs)

        self._default_section = None
        self.set_default_section(default_section or '__config__')

    def get_default_section(self):
        return self._default_section

    def set_default_section(self, section):
        self.add_section(section)

        # move all values from the previous default section to the new one
        try:
            default_section_items = self.items(self._default_section)
            self.remove_section(self._default_section)
        except ConfigParser.NoSectionError:
            pass
        else:
            for (key, value) in default_section_items:
                self.set(section, key, value)

        self._default_section = section

    def read(self, filenames):
        if isinstance(filenames, basestring):
            filenames = [filenames]

        read_ok = []
        for filename in filenames:
            try:
                with open(filename) as fp:
                    self.readfp(fp)
            except IOError:
                continue
            else:
                read_ok.append(filename)

        return read_ok

    def readfp(self, fp, *args, **kwargs):
        stream = StringIO()

        try:
            stream.name = fp.name
        except AttributeError:
            pass

        stream.write('[' + self._default_section + ']\n')
        stream.write(fp.read())
        stream.seek(0, 0)

        return ConfigParser.RawConfigParser.readfp(self, stream, *args,
                                                   **kwargs)

    def write(self, fp):
        # Write the items from the default section manually and then remove them
        # from the data. They'll be re-added later.
        try:
            default_section_items = self.items(self._default_section)
            self.remove_section(self._default_section)

            for (key, value) in default_section_items:
                fp.write("{0} = {1}\n".format(key, value))

            fp.write("\n")
        except ConfigParser.NoSectionError:
            pass

        ConfigParser.RawConfigParser.write(self, fp)

        self.add_section(self._default_section)
        for (key, value) in default_section_items:
            self.set(self._default_section, key, value)

6
投票

Blueicefield的答案提到了configobj,但原来的lib仅支持Python 2。现在它有Python 3+兼容端口:

https://github.com/DiffSK/configobj

API 没有改变,请参阅 doc

© www.soinside.com 2019 - 2024. All rights reserved.