Extend SimpleHoster in multi-hoster plugins

1 files changed, 44 insertions, 41 deletions

diff --git a/module/plugins/hoster/UnrestrictLi.py b/module/plugins/hoster/UnrestrictLi.py
index 786e25a9a..d0a03d7ec 100644
--- a/module/plugins/hoster/UnrestrictLi.py
+++ b/module/plugins/hoster/UnrestrictLi.py
@@ -5,7 +5,7 @@ import re
 from datetime import datetime, timedelta
 
 from module.common.json_layer import json_loads
-from module.plugins.Hoster import Hoster
+from module.plugins.internal.SimpleHoster import SimpleHoster, create_getInfo
 
 
 def secondsToMidnight(gmt=0):
@@ -26,7 +26,7 @@ def secondsToMidnight(gmt=0):
     return int(res)
 
 
-class UnrestrictLi(Hoster):
+class UnrestrictLi(SimpleHoster):
     __name__    = "UnrestrictLi"
     __type__    = "hoster"
     __version__ = "0.14"
@@ -39,53 +39,53 @@ class UnrestrictLi(Hoster):
 
 
     def setup(self):
-        self.chunkLimit = 16
+        self.chunkLimit     = 16
         self.resumeDownload = True
 
 
-    def process(self, pyfile):
-        if re.match(self.__pattern__, pyfile.url):
-            new_url = pyfile.url
-        elif not self.account:
-            self.logError(_("Please enter your %s account or deactivate this plugin") % "Unrestrict.li")
-            self.fail(_("No Unrestrict.li account provided"))
+    def handleMulti(self):
+        for _i in xrange(5):
+            page = self.load('https://unrestrict.li/unrestrict.php',
+                             post={'link': self.pyfile.url, 'domain': 'long'})
+            self.logDebug("JSON data: " + page)
+            if page != '':
+                break
         else:
-            self.logDebug("Old URL: %s" % pyfile.url)
-            for _i in xrange(5):
-                page = self.load('https://unrestrict.li/unrestrict.php',
-                                 post={'link': pyfile.url, 'domain': 'long'})
-                self.logDebug("JSON data: " + page)
-                if page != '':
-                    break
-            else:
-                self.logInfo(_("Unable to get API data, waiting 1 minute and retry"))
-                self.retry(5, 60, "Unable to get API data")
-
-            if 'Expired session' in page or ("You are not allowed to "
-                                             "download from this host" in page and self.premium):
-                self.account.relogin(self.user)
-                self.retry()
-            elif "File offline" in page:
-                self.offline()
-            elif "You are not allowed to download from this host" in page:
-                self.fail(_("You are not allowed to download from this host"))
-            elif "You have reached your daily limit for this host" in page:
-                self.logWarning(_("Reached daily limit for this host"))
-                self.retry(5, secondsToMidnight(gmt=2), "Daily limit for this host reached")
-            elif "ERROR_HOSTER_TEMPORARILY_UNAVAILABLE" in page:
-                self.logInfo(_("Hoster temporarily unavailable, waiting 1 minute and retry"))
-                self.retry(5, 60, "Hoster is temporarily unavailable")
-            page = json_loads(page)
-            new_url = page.keys()[0]
-            self.api_data = page[new_url]
-
-        if new_url != pyfile.url:
-            self.logDebug("New URL: " + new_url)
+            self.logInfo(_("Unable to get API data, waiting 1 minute and retry"))
+            self.retry(5, 60, "Unable to get API data")
+
+        if 'Expired session' in page or ("You are not allowed to "
+                                         "download from this host" in page and self.premium):
+            self.account.relogin(self.user)
+            self.retry()
+
+        elif "File offline" in page:
+            self.offline()
+
+        elif "You are not allowed to download from this host" in page:
+            self.fail(_("You are not allowed to download from this host"))
+
+        elif "You have reached your daily limit for this host" in page:
+            self.logWarning(_("Reached daily limit for this host"))
+            self.retry(5, secondsToMidnight(gmt=2), "Daily limit for this host reached")
+
+        elif "ERROR_HOSTER_TEMPORARILY_UNAVAILABLE" in page:
+            self.logInfo(_("Hoster temporarily unavailable, waiting 1 minute and retry"))
+            self.retry(5, 60, "Hoster is temporarily unavailable")
+
+        page = json_loads(page)
+        self.link = page.keys()[0]
+        self.api_data = page[self.link]
+
+        if self.link != self.pyfile.url:
+            self.logDebug("New URL: " + self.link)
 
         if hasattr(self, 'api_data'):
             self.setNameSize()
 
-        self.download(new_url, disposition=True)
+
+    def checkFile(self):
+        super(UnrestrictLi, self).checkFile()
 
         if self.getConfig("history"):
             self.load("https://unrestrict.li/history/", get={'delete': "all"})
@@ -97,3 +97,6 @@ class UnrestrictLi(Hoster):
             self.pyfile.name = self.api_data['name']
         if 'size' in self.api_data:
             self.pyfile.size = self.api_data['size']
+
+
+getInfo = create_getInfo(UnrestrictLi)