Update Extractor (again)

1 files changed, 163 insertions, 0 deletions

diff --git a/module/plugins/internal/SevenZip.py b/module/plugins/internal/SevenZip.py
new file mode 100644
index 000000000..508cf9c8d
--- /dev/null
+++ b/module/plugins/internal/SevenZip.py
@@ -0,0 +1,163 @@
+# -*- coding: utf-8 -*-
+
+import os
+import re
+
+from subprocess import Popen, PIPE
+
+from module.plugins.internal.UnRar import UnRar, renice
+from module.utils import save_join
+
+
+class SevenZip(UnRar):
+    __name__    = "SevenZip"
+    __version__ = "0.02"
+
+    __description__ = """7-Zip extractor plugin"""
+    __license__     = "GPLv3"
+    __authors__     = [("Michael Nowak", ""),
+                       ("Walter Purcaro", "vuolter@gmail.com")]
+
+
+    CMD = "7z"
+
+    EXTENSIONS = [".7z", ".xz", ".zip", ".gz", ".gzip", ".tgz", ".bz2", ".bzip2",
+                  ".tbz2", ".tbz", ".tar", ".wim", ".swm", ".lzma", ".rar", ".cab",
+                  ".arj", ".z", ".taz", ".cpio", ".rpm", ".deb", ".lzh", ".lha",
+                  ".chm", ".chw", ".hxs", ".iso", ".msi", ".doc", ".xls", ".ppt",
+                  ".dmg", ".xar", ".hfs", ".exe", ".ntfs", ".fat", ".vhd", ".mbr",
+                  ".squashfs", ".cramfs", ".scap"]
+
+
+    #@NOTE: there are some more uncovered 7z formats
+    re_filelist = re.compile(r'([\d\:]+)\s+([\d\:]+)\s+([\w\.]+)\s+(\d+)\s+(\d+)\s+(.+)')
+    re_wrongpwd = re.compile(r'(Can not open encrypted archive|Wrong password)', re.I)
+    re_wrongcrc = re.compile(r'Encrypted\s+\=\s+\+', re.I)
+
+
+    @classmethod
+    def checkDeps(cls):
+        if os.name == "nt":
+            cls.CMD = os.path.join(pypath, "7z.exe")
+            p = Popen([cls.CMD], stdout=PIPE, stderr=PIPE)
+            p.communicate()
+        else:
+            p = Popen([cls.CMD], stdout=PIPE, stderr=PIPE)
+            p.communicate()
+
+        return True
+
+
+    def checkArchive(self):
+        p = self.call_cmd("l", "-slt", self.target)
+        out, err = p.communicate()
+
+        if p.returncode > 1:
+            raise ArchiveError("Process terminated")
+
+        # check if output or error macthes the 'wrong password'-Regexp
+        if self.re_wrongpwd.search(out):
+            return True
+
+        # check if output matches 'Encrypted = +'
+        if self.re_wrongcrc.search(out):
+            return True
+
+        # check if archive is empty
+        self.files = self.list()
+        if not self.files:
+            raise ArchiveError("Empty Archive")
+
+        return False
+
+
+    def checkPassword(self, password):
+        p = self.call_cmd("l", self.target, password=password)
+        p.communicate()
+        return p.returncode == 0
+
+
+    def extract(self, password=None):
+        command = "x" if self.fullpath else "e"
+
+        p = self.call_cmd(command, '-o' + self.out, self.target, password=password)
+
+        renice(p.pid, self.renice)
+
+        progressstring = ""
+        while True:
+            c = p.stdout.read(1)
+            # quit loop on eof
+            if not c:
+                break
+            # reading a percentage sign -> set progress and restart
+            if c == '%':
+                self.notifyProgress(int(progressstring))
+                progressstring = ""
+            # not reading a digit -> therefore restart
+            elif c not in digits:
+                progressstring = ""
+            # add digit to progressstring
+            else:
+                progressstring += c
+
+        # retrieve stderr
+        err = p.stderr.read()
+
+        if self.re_wrongpwd.search(err):
+            raise PasswordError
+
+        elif self.re_wrongcrc.search(err):
+            raise CRCError
+
+        elif err.strip():  #: raise error if anything is on stderr
+            raise ArchiveError(err.strip())
+
+        if p.returncode > 1:
+            raise ArchiveError("Process terminated")
+
+        if not self.files:
+            self.files = self.list(password)
+
+
+    def list(self, password=None):
+        command = "l" if self.fullpath else "l"
+
+        p = self.call_cmd(command, self.target, password=password)
+        out, err = p.communicate()
+        code     = p.returncode
+
+        if "Can not open" in err:
+            raise ArchiveError("Cannot open file")
+
+        if code != 0:
+            raise ArchiveError("Process terminated unsuccessful")
+
+        result = set()
+        for groups in self.re_filelist.findall(out):
+            f = groups[-1].strip()
+            result.add(save_join(self.out, f))
+
+        return list(result)
+
+
+    def call_cmd(self, command, *xargs, **kwargs):
+        args = []
+
+        #overwrite flag
+        if self.overwrite:
+            args.append("-y")
+
+        #set a password
+        if "password" in kwargs and kwargs["password"]:
+            args.append("-p%s" % kwargs["password"])
+        else:
+            args.append("-p-")
+
+        #@NOTE: return codes are not reliable, some kind of threading, cleanup whatever issue
+        call = [self.cmd, command] + args + list(xargs)
+
+        self.manager.logDebug(" ".join([decode(arg) for arg in call]))
+
+        p = Popen(call, stdout=PIPE, stderr=PIPE)
+        return p