From 4a077f1ab66b201eb6e819ca3a8d18d54cc54cbf Mon Sep 17 00:00:00 2001
From: Walter Purcaro <vuolter@gmail.com>
Date: Fri, 26 Sep 2014 12:56:29 +0200
Subject: [TusfilesNet] Added support for Premium users

https://github.com/pyload/pyload/pull/755
---
 module/plugins/accounts/TusfilesNet.py | 69 ++++++++++++++++++++++++++++++++++
 module/plugins/hoster/TusfilesNet.py   | 12 ++++--
 2 files changed, 77 insertions(+), 4 deletions(-)
 create mode 100644 module/plugins/accounts/TusfilesNet.py

(limited to 'module/plugins')

diff --git a/module/plugins/accounts/TusfilesNet.py b/module/plugins/accounts/TusfilesNet.py
new file mode 100644
index 000000000..d7cdbaebb
--- /dev/null
+++ b/module/plugins/accounts/TusfilesNet.py
@@ -0,0 +1,69 @@
+# -*- coding: utf-8 -*-
+
+import re
+
+from time import mktime, strptime, gmtime
+
+from module.plugins.Account import Account
+from module.plugins.internal.SimpleHoster import parseHtmlForm
+from module.utils import parseFileSize
+
+
+class TusfilesNet(Account):
+    __name__ = "TusfilesNet"
+    __type__ = "account"
+    __version__ = "0.01"
+
+    __description__ = """ Tusfile.net account plugin """
+    __author_name__ = "guidobelix"
+    __author_mail__ = "guidobelix@hotmail.it"
+
+    VALID_UNTIL_PATTERN = r'<span class="label label-default">([^<]+)</span>'
+    TRAFFIC_LEFT_PATTERN = r'<td><img src="//www.tusfiles.net/i/icon/meter.png" alt=""/></td>\n<td>&nbsp;(?P<S>[^<]+)</td>'
+
+
+    def loadAccountInfo(self, user, req):
+        html = req.load("http://www.tusfiles.net/?op=my_account", decode=True)
+
+        validuntil = None
+        trafficleft = None
+        premium = False
+
+        m = re.search(self.VALID_UNTIL_PATTERN, html)
+        if m:
+            expiredate = m.group(1)
+            self.logDebug("Expire date: " + expiredate)
+
+            try:
+                validuntil = mktime(strptime(expiredate, "%d %B %Y"))
+            except Exception, e:
+                self.logError(e)
+
+            if validuntil > mktime(gmtime()):
+                premium = True
+            else:
+                premium = False
+                validuntil = None
+
+        m = re.search(self.TRAFFIC_LEFT_PATTERN, html)
+        if m:
+            trafficleft = m.group(1)
+            if "Unlimited" in trafficleft:
+                trafficleft = -1
+            else:
+                trafficleft = parseFileSize(trafficleft) * 1024
+
+        return {'validuntil': validuntil, 'trafficleft': trafficleft, 'premium': premium}
+
+
+    def login(self, user, data, req):
+        html = req.load("http://www.tusfiles.net/login.html", decode=True)
+        action, inputs = parseHtmlForm('name="FL"', html)
+        inputs.update({'login': user,
+                       'password': data['password'],
+                       'redirect': "http://www.tusfiles.net/"})
+
+        html = req.load("http://www.tusfiles.net/", post=inputs, decode=True)
+
+        if 'Incorrect Login or Password' in html or '>Error<' in html:
+            self.wrongPassword()
diff --git a/module/plugins/hoster/TusfilesNet.py b/module/plugins/hoster/TusfilesNet.py
index 7fb2a375e..350241ea9 100644
--- a/module/plugins/hoster/TusfilesNet.py
+++ b/module/plugins/hoster/TusfilesNet.py
@@ -6,13 +6,13 @@ from module.plugins.hoster.XFileSharingPro import XFileSharingPro, create_getInf
 class TusfilesNet(XFileSharingPro):
     __name__ = "TusfilesNet"
     __type__ = "hoster"
-    __version__ = "0.03"
+    __version__ = "0.04"
 
     __pattern__ = r'https?://(?:www\.)?tusfiles\.net/(?P<ID>\w+)'
 
-    __description__ = """Tusfiles.net hoster plugin"""
-    __author_name__ = "Walter Purcaro"
-    __author_mail__ = "vuolter@gmail.com"
+    __description__ = """ Tusfiles.net hoster plugin """
+    __author_name__ = ("Walter Purcaro", "guidobelix")
+    __author_mail__ = ("vuolter@gmail.com", "guidobelix@hotmail.it")
 
     HOSTER_NAME = "tusfiles.net"
 
@@ -28,4 +28,8 @@ class TusfilesNet(XFileSharingPro):
         self.resumeDownload = True
 
 
+    def handlePremium(self):
+        return self.handleFree()
+
+
 getInfo = create_getInfo(TusfilesNet)
-- 
cgit v1.2.3