1 files changed, 42 insertions, 0 deletions
diff --git a/pyload/plugin/account/UploadheroCom.py b/pyload/plugin/account/UploadheroCom.py
new file mode 100644
index 000000000..c73fc30f5
--- /dev/null
+++ b/pyload/plugin/account/UploadheroCom.py
@@ -0,0 +1,42 @@
+# -*- coding: utf-8 -*-
+
+import re
+import datetime
+import time
+
+from pyload.plugin.Account import Account
+
+
+class UploadheroCom(Account):
+    __name__    = "UploadheroCom"
+    __type__    = "account"
+    __version__ = "0.21"
+
+    __description__ = """Uploadhero.co account plugin"""
+    __license__     = "GPLv3"
+    __authors__     = [("mcmyst", "mcmyst@hotmail.fr")]
+
+
+    def loadAccountInfo(self, user, req):
+        premium_pattern = re.compile('Il vous reste <span class="bleu">(\d+)</span> jours premium')
+
+        data = self.getAccountData(user)
+        html = req.load("http://uploadhero.co/my-account")
+
+        if premium_pattern.search(html):
+            end_date = datetime.date.today() + datetime.timedelta(days=int(premium_pattern.search(html).group(1)))
+            end_date = time.mktime(future.timetuple())
+            account_info = {"validuntil": end_date, "trafficleft": -1, "premium": True}
+        else:
+            account_info = {"validuntil": -1, "trafficleft": -1, "premium": False}
+
+        return account_info
+
+
+    def login(self, user, data, req):
+        html = req.load("http://uploadhero.co/lib/connexion.php",
+                        post={"pseudo_login": user, "password_login": data['password']},
+                        decode=True)
+
+        if "mot de passe invalide" in html:
+            self.wrongPassword()