[Python-checkins] r62724 - in python/trunk: Lib/test/regrtest.py Lib/test/test_zipfile.py Lib/zipfile.py Misc/NEWS
martin.v.loewis
python-checkins at python.org
Mon May 5 19:16:58 CEST 2008
Author: martin.v.loewis
Date: Mon May 5 19:16:58 2008
New Revision: 62724
Log:
Issue #1734346: Support Unicode file names for zipfiles.
Modified:
python/trunk/Lib/test/regrtest.py
python/trunk/Lib/test/test_zipfile.py
python/trunk/Lib/zipfile.py
python/trunk/Misc/NEWS
Modified: python/trunk/Lib/test/regrtest.py
==============================================================================
--- python/trunk/Lib/test/regrtest.py (original)
+++ python/trunk/Lib/test/regrtest.py Mon May 5 19:16:58 2008
@@ -1,6 +1,6 @@
#! /usr/bin/env python
-"""Regression test.
+"""Regression test driver.
This will find all modules whose name is "test_*" in the test
directory, and run them. Various command line options provide
Modified: python/trunk/Lib/test/test_zipfile.py
==============================================================================
--- python/trunk/Lib/test/test_zipfile.py (original)
+++ python/trunk/Lib/test/test_zipfile.py Mon May 5 19:16:58 2008
@@ -553,6 +553,15 @@
class OtherTests(unittest.TestCase):
+ def testUnicodeFilenames(self):
+ zf = zipfile.ZipFile(TESTFN, "w")
+ zf.writestr(u"foo.txt", "Test for unicode filename")
+ zf.writestr(u"fo\xf6.txt", "Test for unicode filename")
+ assert isinstance(zf.infolist()[0].filename, unicode)
+ zf.close()
+ zf = zipfile.ZipFile(TESTFN, "w")
+
+
def testCreateNonExistentFileForAppend(self):
if os.path.exists(TESTFN):
os.unlink(TESTFN)
Modified: python/trunk/Lib/zipfile.py
==============================================================================
--- python/trunk/Lib/zipfile.py (original)
+++ python/trunk/Lib/zipfile.py Mon May 5 19:16:58 2008
@@ -255,12 +255,28 @@
self.extract_version = max(45, self.extract_version)
self.create_version = max(45, self.extract_version)
+ filename, flag_bits = self._encodeFilenameFlags()
header = struct.pack(structFileHeader, stringFileHeader,
- self.extract_version, self.reserved, self.flag_bits,
+ self.extract_version, self.reserved, flag_bits,
self.compress_type, dostime, dosdate, CRC,
compress_size, file_size,
- len(self.filename), len(extra))
- return header + self.filename + extra
+ len(filename), len(extra))
+ return header + filename + extra
+
+ def _encodeFilenameFlags(self):
+ if isinstance(self.filename, unicode):
+ try:
+ return self.filename.encode('ascii'), self.flag_bits
+ except UnicodeEncodeError:
+ return self.filename.encode('utf-8'), self.flag_bits | 0x800
+ else:
+ return self.filename, self.flag_bits
+
+ def _decodeFilename(self):
+ if self.flag_bits & 0x800:
+ return self.filename.decode('utf-8')
+ else:
+ return self.filename
def _decodeExtra(self):
# Try to decode the extra field.
@@ -693,6 +709,7 @@
x._decodeExtra()
x.header_offset = x.header_offset + concat
+ x.filename = x._decodeFilename()
self.filelist.append(x)
self.NameToInfo[x.filename] = x
if self.debug > 2:
@@ -1054,12 +1071,13 @@
create_version = zinfo.create_version
try:
+ filename, flag_bits = zinfo._encodeFilenameFlags()
centdir = struct.pack(structCentralDir,
stringCentralDir, create_version,
zinfo.create_system, extract_version, zinfo.reserved,
- zinfo.flag_bits, zinfo.compress_type, dostime, dosdate,
+ flag_bits, zinfo.compress_type, dostime, dosdate,
zinfo.CRC, compress_size, file_size,
- len(zinfo.filename), len(extra_data), len(zinfo.comment),
+ len(filename), len(extra_data), len(zinfo.comment),
0, zinfo.internal_attr, zinfo.external_attr,
header_offset)
except DeprecationWarning:
@@ -1073,7 +1091,7 @@
header_offset)
raise
self.fp.write(centdir)
- self.fp.write(zinfo.filename)
+ self.fp.write(filename)
self.fp.write(extra_data)
self.fp.write(zinfo.comment)
Modified: python/trunk/Misc/NEWS
==============================================================================
--- python/trunk/Misc/NEWS (original)
+++ python/trunk/Misc/NEWS Mon May 5 19:16:58 2008
@@ -49,6 +49,8 @@
Library
-------
+- Issue #1734346: Support Unicode file names for zipfiles.
+
- Issue #2581: distutils: Vista UAC/elevation support for bdist_wininst
- Issue #2635: Fix bug in 'fix_sentence_endings' textwrap.fill option,
More information about the Python-checkins
mailing list