configparser从zip加载配置文件

时间:2015-07-13 19:43:25

标签: python zipfile configparser

我正在创建一个从压缩文件加载和运行python脚本的程序。除了那些python脚本,我有一个配置文件,我以前使用configparser从程序的未压缩版本加载信息。

是否可以使用configparser直接读取zip文件中的配置文件?或者我是否必须将其解压缩到临时文件夹并从那里加载?

我试过直接给出路径:

>>> sysconf = configparser.ConfigParser()
>>> sysconf.read_file("compressed.zip/config_data.conf")

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.4/configparser.py", line 691, in read_file
    self._read(f, source)
  File "/usr/local/lib/python3.4/configparser.py", line 1058, in _read
    raise MissingSectionHeaderError(fpname, lineno, line)
configparser.MissingSectionHeaderError: File contains no section headers.
file: '<???>', line: 1

没用。没有惊喜。

然后我尝试使用zipfile

 >>> zf = zipfile.ZipFile("compressed.zip")
 >>> data = zf.read("config_data.conf")
 >>> sysconf = configparser.ConfigParser()
 >>> sysconf.read_file(data)

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.4/configparser.py", line 691, in read_file
    self._read(f, source)
  File "/usr/local/lib/python3.4/configparser.py", line 1009, in _read
    if line.strip().startswith(prefix):
AttributeError: 'int' object has no attribute 'strip'

并发现它也无效。

所以我使用了创建一个临时文件夹,解压缩它,并在那里读取conf文件。如果可能的话,我真的想避免这种情况,因为conf文件是唯一的限制因素。我可以(而且)从zip文件加载python模块就好了。

我可以获取文件的原始文本,如果有办法将其直接传递给configparser,但搜索文档我是空手而归。

更新: 我尝试使用stringIO作为文件对象,它似乎有点工作。 configparser不会拒绝它,但它也不喜欢它。

>>> zf = zipfile.ZipFile("compressed.zip")
>>> data = zf.read(config_data.conf)
>>> confdata = io.StringIO(str(data))
>>> sysconf = configparser.ConfigParser()
>>> sysconf.readfp(confdata)
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.4/configparser.py", line 736, in readfp
    self.read_file(fp, source=filename)
  File "/usr/local/lib/python3.4/configparser.py", line 691, in read_file
    self._read(f, source)
  File "/usr/local/lib/python3.4/configparser.py", line 1058, in _read
    raise MissingSectionHeaderError(fpname, lineno, line)
configparser.MissingSectionHeaderError: File contains no section headers.
file: '<???>', line: 1
(continues to spit out the entire contents of the file)

如果我使用read_file,它不会出错,但也不会加载任何内容。

>>> zf = zipfile.ZipFile("compressed.zip")
>>> data = zf.read(config_data.conf)
>>> confdata = io.StringIO(str(data))
>>> sysconf = configparser.ConfigParser()
>>> sysconf.read_file(confdata)
>>> sysconf.items("General") #(this is the main section in the file)
Traceback (most recent call last):
  File "/usr/local/lib/python3.4/configparser.py", line 824, in items
    d.update(self._sections[section])
KeyError: 'General'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/local/lib/python3.4/configparser.py", line 827, in items
    raise NoSectionError(section)
configparser.NoSectionError: No section: 'General'

3 个答案:

答案 0 :(得分:1)

  
    

可以获取文件的原始文本,如果有办法将其直接传递给configparser

  

尝试configparser.ConfigParser.read_string

当与适当的ZIP文件结合使用时,此代码适用于我:

import zipfile
import configparser

zf = zipfile.ZipFile("compressed.zip")
zf_config = zf.open("config_data.conf", "rU")
zf_config_data = zf_config.read().decode('ascii')

config = configparser.ConfigParser()
config.read_string(zf_config_data)
assert config['today']['lunch']=='cheeseburger'

经过反思,以下内容可能更合适:

import zipfile
import configparser
import io

zf = zipfile.ZipFile("compressed.zip")
zf_config = zf.open("config_data.conf", "rU")
zf_config = io.TextIOWrapper(zf_config)

config = configparser.ConfigParser()
config.read_file(zf_config)
assert config['today']['lunch']=='cheeseburger'

答案 1 :(得分:0)

ZipFile不仅支持read,还支持open,它返回类似文件的对象。所以,你可以这样做:

zf = zipfile.ZipFile("compressed.zip")
config_file = zf.open("config_data.conf")
sysconfig = configparser.ConfigParser()
sysconfig.readfp(config_file)

答案 2 :(得分:0)

正如评论中所写,@ matthewatabet答案不适用于Python 3.4(和较新版本)。这是因为ZipFile.open现在返回“字节状”对象,而不是“文件状”对象。您可以使用:

codecs.getreader("utf-8")(config_file)

使用UTF-8编码将config_file个字节状对象转换为文件状对象。现在的代码是:

import zipfile, configparser, codecs

with zipfile.ZipFile("compressed.zip") as zf:
    config_file = zf.open("config_data.conf")
    sysconfig = configparser.ConfigParser()
    sysconfig.read_file(codecs.getreader("utf-8")(config_file))

这似乎比创建string更令人满意,但是我不知道它是否更有效...