计算机正在玩弄我,我知道!
我正在用Python创建一个zip文件夹。单个文件在内存中生成,然后将整个文件压缩并保存到文件中。我被允许在zip中添加9个文件。我被允许在zip中添加11个文件。但是10,不,不是10个文件。 zip文件已保存到我的电脑,但我不允许打开它; Windows表示压缩的压缩文件夹无效。
我使用下面的代码,这是我从另一个stackoverflow问题得到的。它附加10个文件并保存压缩文件夹。当我点击文件夹时,我无法提取它。但是,删除其中一个追加(),它没关系。或者,添加另一个附加,它可以工作!
我在这里缺少什么?我怎样才能每次都做这个工作?
imz = InMemoryZip()
imz.append("1a.txt", "a").append("2a.txt", "a").append("3a.txt", "a").append("4a.txt", "a").append("5a.txt", "a").append("6a.txt", "a").append("7a.txt", "a").append("8a.txt", "a").append("9a.txt", "a").append("10a.txt", "a")
imz.writetofile("C:/path/test.zip")
import zipfile
import StringIO
class InMemoryZip(object):
def __init__(self):
# Create the in-memory file-like object
self.in_memory_zip = StringIO.StringIO()
def append(self, filename_in_zip, file_contents):
'''Appends a file with name filename_in_zip and contents of
file_contents to the in-memory zip.'''
# Get a handle to the in-memory zip in append mode
zf = zipfile.ZipFile(self.in_memory_zip, "a", zipfile.ZIP_DEFLATED, False)
# Write the file to the in-memory zip
zf.writestr(filename_in_zip, file_contents)
# Mark the files as having been created on Windows so that
# Unix permissions are not inferred as 0000
for zfile in zf.filelist:
zfile.create_system = 0
return self
def read(self):
'''Returns a string with the contents of the in-memory zip.'''
self.in_memory_zip.seek(0)
return self.in_memory_zip.read()
def writetofile(self, filename):
'''Writes the in-memory zip to a file.'''
f = file(filename, "w")
f.write(self.read())
f.close()
答案 0 :(得分:1)
在创建要保存到文件系统的文件时,应使用'wb'模式。这将确保文件以二进制形式写入。
否则,每次在zip文件中遇到换行符(\ n)时,python都会将其替换为与windows行结尾(\ r \ n)匹配。 10个文件出问题的原因是10恰好是\ n。
的代码所以你的写函数应该是这样的:
def writetofile(self, filename):
'''Writes the in-memory zip to a file.'''
f = file(filename, 'wb')
f.write(self.read())
f.close()
这应该可以解决您的问题并为您的示例中的文件工作。虽然,在您的情况下,您可能会发现将zip文件直接写入文件系统更容易,例如此代码包含上述一些注释:
import StringIO
import zipfile
class ZipCreator:
buffer = None
def __init__(self, fileName=None):
if fileName:
self.zipFile = zipfile.ZipFile(fileName, 'w', zipfile.ZIP_DEFLATED, False)
return
self.buffer = StringIO.StringIO()
self.zipFile = zipfile.ZipFile(self.buffer, 'w', zipfile.ZIP_DEFLATED, False)
def addToZipFromFileSystem(self, filePath, filenameInZip):
self.zipFile.write(filePath, filenameInZip)
def addToZipFromMemory(self, filenameInZip, fileContents):
self.zipFile.writestr(filenameInZip, fileContents)
for zipFile in self.zipFile.filelist:
zipFile.create_system = 0
def write(self, fileName):
if not self.buffer: # If the buffer was not initialized the file is written by the ZipFile
self.zipFile.close()
return
f = file(fileName, 'wb')
f.write(self.buffer.getvalue())
f.close()
# Use File Handle
zipCreator = ZipCreator('C:/path/test.zip')
# Use Memory Buffer
# zipCreator = ZipCreator()
for i in range(1, 10):
zipCreator.addToZipFromMemory('test/%sa.txt' % i, 'a')
zipCreator.write('C:/path/test.zip')
理想情况下,您可能会使用单独的类作为内存中的zip和从一开始就绑定到文件系统的zip。在添加难以重新创建的文件夹以及我仍在尝试追踪的文件夹时,我也看到了内存中拉链的一些问题。