Netload Premium Test

2 files changed, 15 insertions, 11 deletions

diff --git a/Plugins/NetloadIn.py b/Plugins/NetloadIn.py
index 39ea9213e..8bff067de 100644
--- a/Plugins/NetloadIn.py
+++ b/Plugins/NetloadIn.py
@@ -46,6 +46,10 @@ class NetloadIn(Plugin):
                 raise Exception, "The file was not found on the server."
 
             pyfile.status.filename = self.get_file_name()
+            
+            if self.config['premium']:
+                pyfile.status.url = self.parent.url
+                return True
 
             self.download_html2()
 
@@ -135,4 +139,8 @@ class NetloadIn(Plugin):
             return True
 
     def proceed(self, url, location):
+
+        if self.config['premium']:
+            self.req.add_auth(self.config['username'], self.config['password'])
+
         self.req.download(url, location, cookies=True)
diff --git a/Plugins/RapidshareCom.py b/Plugins/RapidshareCom.py
index a81affd90..731554f87 100644
--- a/Plugins/RapidshareCom.py
+++ b/Plugins/RapidshareCom.py
@@ -45,12 +45,16 @@ class RapidshareCom(Plugin):
 
             self.download_html()
 
-            pyfile.status.filename = self.get_file_name()
-
             pyfile.status.exists = self.file_exists()
-
+            
             if not pyfile.status.exists:
                 raise Exception, "The file was not found on the server."
+            
+            pyfile.status.filename = self.get_file_name()
+            
+            if self.config['premium']:
+                pyfile.status.url = self.parent.url
+                return True
 
             self.download_serverhtml()
             pyfile.status.waituntil = self.time_plus_wait
@@ -78,10 +82,6 @@ class RapidshareCom(Plugin):
     def download_serverhtml(self):
         """downloads html with the important informations
         """
-
-        if self.config['premium']:
-            return False
-
         file_server_url = re.search(r"<form action=\"(.*?)\"", self.html[0]).group(1)
         self.html[1] = self.req.load(file_server_url, None, {"dl.start": "Free"})
         self.html_old = time()
@@ -89,10 +89,6 @@ class RapidshareCom(Plugin):
 
     def get_wait_time(self):
 
-        if self.config['premium']:
-            self.time_plus_wait = 0
-            return True
-
         if re.search(r".*is already downloading.*", self.html[1]) != None:
             self.time_plus_wait = time() + 10 * 60
         try: