1 files changed, 68 insertions, 32 deletions
diff --git a/module/plugins/internal/UnZip.py b/module/plugins/internal/UnZip.py
index 501962442..5ec56cbdf 100644
--- a/module/plugins/internal/UnZip.py
+++ b/module/plugins/internal/UnZip.py
@@ -1,50 +1,86 @@
 # -*- coding: utf-8 -*-
 
-"""
-    This program is free software; you can redistribute it and/or modify
-    it under the terms of the GNU General Public License as published by
-    the Free Software Foundation; either version 3 of the License,
-    or (at your option) any later version.
+from __future__ import with_statement
 
-    This program is distributed in the hope that it will be useful,
-    but WITHOUT ANY WARRANTY; without even the implied warranty of
-    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
-    See the GNU General Public License for more details.
+import sys
+import zipfile
 
-    You should have received a copy of the GNU General Public License
-    along with this program; if not, see <http://www.gnu.org/licenses/>.
+from module.plugins.internal.Extractor import Extractor, ArchiveError, CRCError, PasswordError
 
-    @author: RaNaN
-"""
 
-import zipfile
-import sys
+class UnZip(Extractor):
+    __name__    = "UnZip"
+    __version__ = "1.01"
+
+    __description__ = """Zip extractor plugin"""
+    __license__     = "GPLv3"
+    __authors__     = [("Walter Purcaro", "vuolter@gmail.com")]
 
-from module.plugins.internal.AbstractExtractor import AbtractExtractor
 
+    EXTENSIONS = ["zip", "zip64"]
 
-class UnZip(AbtractExtractor):
-    __name__ = "UnZip"
-    __version__ = "0.1"
 
-    @staticmethod
-    def checkDeps():
+    @classmethod
+    def checkDeps(cls):
         return sys.version_info[:2] >= (2, 6)
 
-    @staticmethod
-    def getTargets(files_ids):
-        result = []
 
-        for file, id in files_ids:
-            if file.endswith(".zip"):
-                result.append((file, id))
+    @classmethod
+    def isArchive(cls, file):
+        return zipfile.is_zipfile(file)
+
+
+    def verify(self):
+        try:
+            with zipfile.ZipFile(self.file, 'r', allowZip64=True) as z:
+                z.setpassword(self.password)
+                badcrc = z.testzip()
+
+        except (BadZipfile, LargeZipFile), e:
+            raise ArchiveError(e)
+
+        except RuntimeError, e:
+            if 'encrypted' in e:
+                raise PasswordError
+            else:
+                raise ArchiveError(e)
+
+        else:
+            if badcrc:
+                raise CRCError
+
+        if not self.list():
+            raise ArchiveError("Empty archive")
+
+
+    def list(self):
+        try:
+            with zipfile.ZipFile(self.file, 'r', allowZip64=True) as z:
+                z.setpassword(self.password)
+                return z.namelist()
+        except Exception:
+            return list()
+
+
+    def extract(self, progress=lambda x: None):
+        try:
+            with zipfile.ZipFile(self.file, 'r', allowZip64=True) as z:
+                progress(0)
+                z.extractall(self.out, pwd=self.password)
+                progress(100)
+
+        except (BadZipfile, LargeZipFile), e:
+            raise ArchiveError(e)
+
+        except RuntimeError, e:
+            if e is "Bad password for file":
+                raise PasswordError
+            else:
+                raise ArchiveError(e)
 
-        return result
+        finally:
+            self.files = self.list()
 
-    def extract(self, progress, password=None):
-        z = zipfile.ZipFile(self.file)
-        self.files = z.namelist()
-        z.extractall(self.out)
 
     def getDeleteFiles(self):
         return [self.file]