1 files changed, 79 insertions, 0 deletions
diff --git a/pyload/plugin/account/HellshareCz.py b/pyload/plugin/account/HellshareCz.py
new file mode 100644
index 000000000..94467b375
--- /dev/null
+++ b/pyload/plugin/account/HellshareCz.py
@@ -0,0 +1,79 @@
+# -*- coding: utf-8 -*-
+
+import re
+import time
+
+from pyload.plugin.Account import Account
+
+
+class HellshareCz(Account):
+    __name__    = "HellshareCz"
+    __type__    = "account"
+    __version__ = "0.16"
+
+    __description__ = """Hellshare.cz account plugin"""
+    __license__     = "GPLv3"
+    __authors__     = [("zoidberg", "zoidberg@mujmail.cz")]
+
+
+    CREDIT_LEFT_PATTERN = r'<div class="credit-link">\s*<table>\s*<tr>\s*<th>(\d+|\d\d\.\d\d\.)</th>'
+
+
+    def loadAccountInfo(self, user, req):
+        self.relogin(user)
+        html = req.load("http://www.hellshare.com/")
+
+        m = re.search(self.CREDIT_LEFT_PATTERN, html)
+        if m is None:
+            trafficleft = None
+            validuntil = None
+            premium = False
+        else:
+            credit = m.group(1)
+            premium = True
+            try:
+                if "." in credit:
+                    #Time-based account
+                    vt = [int(x) for x in credit.split('.')[:2]]
+                    lt = time.localtime()
+                    year = lt.tm_year + int(vt[1] < lt.tm_mon or (vt[1] == lt.tm_mon and vt[0] < lt.tm_mday))
+                    validuntil = time.mktime(time.strptime("%s%d 23:59:59" % (credit, year), "%d.%m.%Y %H:%M:%S"))
+                    trafficleft = -1
+                else:
+                    #Traffic-based account
+                    trafficleft = self.parseTraffic(credit + "MB")
+                    validuntil = -1
+            except Exception, e:
+                self.logError(_("Unable to parse credit info"), e)
+                validuntil = -1
+                trafficleft = -1
+
+        return {"validuntil": validuntil, "trafficleft": trafficleft, "premium": premium}
+
+
+    def login(self, user, data, req):
+        html = req.load('http://www.hellshare.com/', decode=True)
+        if req.lastEffectiveURL != 'http://www.hellshare.com/':
+            #Switch to English
+            self.logDebug("Switch lang - URL: %s" % req.lastEffectiveURL)
+
+            json = req.load("%s?do=locRouter-show" % req.lastEffectiveURL)
+            hash = re.search(r"(\-\-[0-9a-f]+\-)", json).group(1)
+
+            self.logDebug("Switch lang - HASH: %s" % hash)
+
+            html = req.load('http://www.hellshare.com/%s/' % hash, decode=True)
+
+        if re.search(self.CREDIT_LEFT_PATTERN, html):
+            self.logDebug("Already logged in")
+            return
+
+        html = req.load('http://www.hellshare.com/login?do=loginForm-submit',
+                        post={"login": "Log in",
+                              "password": data['password'],
+                              "username": user,
+                              "perm_login": "on"},
+                        decode=True)
+
+        if "<p>You input a wrong user name or wrong password</p>" in html:
+            self.wrongPassword()