Python下载zip文件并立即打开它

时间:2015-03-26 18:37:14

标签: python gzip

我从一个以zip(.gz)文件形式返回的网站下载赔率。 目前我的代码是:

f = urllib2.urlopen(url)

with open("code2.zip", "wb") as code:
    code.write(f.read())
code.close()

z = gzip.open("code2.zip", "r")
file_content = z.read()
z.close()
root = etree.fromstring(file_content)

有没有办法不保存文件而是将其作为字符串存储在内存中并在gzip.open()中读取?或者我可以提高效率的其他任何方式?

1 个答案:

答案 0 :(得分:2)

您可以利用StringIO.StringIO包并将内容保留在内存中:

in_mem_file = StringIO.StringIO(f.read())
file_content = gzip.open(in_mem_file).read()

来自StringIO Docs

  

这个模块实现了一个类似文件的类StringIO,它读取和   写一个字符串缓冲区(也称为内存文件)。