Use cStringIO where available.
diff --git a/Lib/tarfile.py b/Lib/tarfile.py
index b85f117..0dc7a42 100644
--- a/Lib/tarfile.py
+++ b/Lib/tarfile.py
@@ -1936,12 +1936,15 @@
def write(self, filename, arcname=None, compress_type=None):
self.tarfile.add(filename, arcname)
def writestr(self, zinfo, bytes):
- import StringIO
+ try:
+ from cStringIO import StringIO
+ except ImportError:
+ from StringIO import StringIO
import calendar
zinfo.name = zinfo.filename
zinfo.size = zinfo.file_size
zinfo.mtime = calendar.timegm(zinfo.date_time)
- self.tarfile.addfile(zinfo, StringIO.StringIO(bytes))
+ self.tarfile.addfile(zinfo, StringIO(bytes))
def close(self):
self.tarfile.close()
#class TarFileCompat