Fix issue with running on python 2.6

itsjohncs · Nov 6, 2013 · c5bbbfc · c5bbbfc
1 parent 9d06f66
commit c5bbbfc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/superzippy/bootstrapper/zipsite.py b/superzippy/bootstrapper/zipsite.py
@@ -20,6 +20,8 @@
 import site as _site
 from site import *
 
+
+from contextlib import closing
 import zipfile
 import os
 import traceback
@@ -112,7 +114,7 @@ def exists(path):
         return os.path.exists(path)
 
     # otherwise check the zip file.
-    with zipfile.ZipFile(archive_path, mode = "r") as archive:
+    with closing(zipfile.ZipFile(archive_path, mode = "r")) as archive:
         try:
             archive.getinfo(file_path)
         except KeyError:
@@ -145,7 +147,7 @@ def addsitedir(sitedir, known_paths = None, prepend_mode = False):
     else:
         sys.path.append(sitedir)
 
-    with zipfile.ZipFile(archive_path, mode = "r") as archive:
+    with closing(zipfile.ZipFile(archive_path, mode = "r")) as archive:
         # Go trhough everything in the archive...
         for i in archive.infolist():
             # and grab all the .pth files.