tools v5.4

Calibre_Plugins/ignobleepub_plugin/zipfix.py

@@ -2,7 +2,7 @@
 
 import sys
 import zlib
-import zipfile
+import zipfilerugged
 import os
 import os.path
 import getopt
@@ -15,7 +15,7 @@ _FILENAME_OFFSET = 30
 _MAX_SIZE = 64 * 1024
 _MIMETYPE = 'application/epub+zip'
 
-class ZipInfo(zipfile.ZipInfo):
+class ZipInfo(zipfilerugged.ZipInfo):
     def __init__(self, *args, **kwargs):
         if 'compress_type' in kwargs:
             compress_type = kwargs.pop('compress_type')
@@ -27,11 +27,11 @@ class fixZip:
         self.ztype = 'zip'
         if zinput.lower().find('.epub') >= 0 :
             self.ztype = 'epub'
-        self.inzip = zipfile.ZipFile(zinput,'r')
-        self.outzip = zipfile.ZipFile(zoutput,'w')
+        self.inzip = zipfilerugged.ZipFile(zinput,'r')
+        self.outzip = zipfilerugged.ZipFile(zoutput,'w')
         # open the input zip for reading only as a raw file
-	self.bzf = file(zinput,'rb')
-        
+        self.bzf = file(zinput,'rb')
+
     def getlocalname(self, zi):
         local_header_offset = zi.header_offset
         self.bzf.seek(local_header_offset + _FILENAME_LEN_OFFSET)
@@ -76,17 +76,17 @@ class fixZip:
         data = None
 
         # if not compressed we are good to go
-        if zi.compress_type == zipfile.ZIP_STORED:
+        if zi.compress_type == zipfilerugged.ZIP_STORED:
             data = self.bzf.read(zi.file_size)
 
         # if compressed we must decompress it using zlib
-        if zi.compress_type == zipfile.ZIP_DEFLATED:
+        if zi.compress_type == zipfilerugged.ZIP_DEFLATED:
             cmpdata = self.bzf.read(zi.compress_size)
             data = self.uncompress(cmpdata)
 
         return data
 
-        
+
 
     def fix(self):
         # get the zipinfo for each member of the input archive
@@ -95,7 +95,7 @@ class fixZip:
 
         # if epub write mimetype file first, with no compression
         if self.ztype == 'epub':
-            nzinfo = ZipInfo('mimetype', compress_type=zipfile.ZIP_STORED)
+            nzinfo = ZipInfo('mimetype', compress_type=zipfilerugged.ZIP_STORED)
             self.outzip.writestr(nzinfo, _MIMETYPE)
 
         # write the rest of the files
@@ -103,9 +103,9 @@ class fixZip:
             if zinfo.filename != "mimetype" or self.ztype == '.zip':
                 data = None
                 nzinfo = zinfo
-                try: 
+                try:
                     data = self.inzip.read(zinfo.filename)
-                except zipfile.BadZipfile or zipfile.error:
+                except zipfilerugged.BadZipfile or zipfilerugged.error:
                     local_name = self.getlocalname(zinfo)
                     data = self.getfiledata(zinfo)
                     nzinfo.filename = local_name
@@ -126,7 +126,7 @@ def usage():
      inputzip is the source zipfile to fix
      outputzip is the fixed zip archive
     """
-    
+
 
 def repairBook(infile, outfile):
     if not os.path.exists(infile):
@@ -152,5 +152,3 @@ def main(argv=sys.argv):
 
 if __name__ == '__main__' :
     sys.exit(main())
-
-