add megashares.com plugin (#262)

2 files changed, 148 insertions, 7 deletions

diff --git a/module/plugins/hoster/MediafireCom.py b/module/plugins/hoster/MediafireCom.py
index 8a39d364e..0bc0e504f 100644
--- a/module/plugins/hoster/MediafireCom.py
+++ b/module/plugins/hoster/MediafireCom.py
@@ -35,8 +35,8 @@ def getInfo(urls):
 
             found = re.search(MediafireCom.FILE_SIZE_PATTERN, html)
             if found is not None:
-                size, units = found.groups()
-                size = float(size) * 1024 ** {'kB': 1, 'MB': 2, 'GB': 3}[units]
+                size, units = found.group(1), found.group(2).replace('k','K')
+                size = float(size) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[units]
 
             found = re.search(MediafireCom.FILE_NAME_PATTERN, html)
             if found is not None:
@@ -53,9 +53,10 @@ class MediafireCom(Hoster):
     __name__ = "MediafireCom"
     __type__ = "hoster"
     __pattern__ = r"http://(?:\w*\.)*mediafire\.com/.*"
-    __version__ = "0.4"
+    __version__ = "0.5"
     __description__ = """Mediafire.com plugin - free only"""
     __author_name__ = ("zoidberg")
+    __author_mail__ = ("zoidberg@mujmail.cz")
 
     PAGE1_FUNCTION_PATTERN = r"function %s\(qk,pk1\)\{if[^']*'loadingicon'\);[^;]*; (.*?)eval"
     PAGE1_KEY_PATTERN = ";break;}\s*(\w+='';\w+=unescape.*?)eval\("
@@ -71,7 +72,7 @@ class MediafireCom(Hoster):
     FINAL_LINK_PATTERN = r'parent.document.getElementById\(\'(\w{32})\'\).*(http://[^"]+)" \+(\w+)\+ "([^"]+)">'
 
     FILE_NAME_PATTERN = r'<META NAME="description" CONTENT="([^"]+)"/>'
-    FILE_SIZE_PATTERN = r'<div style="font-size:14px;padding-top:12px;color:#777;">\(([0-9.]+) (kB|MB|GB)\)</div>'
+    FILE_SIZE_PATTERN = r'<div style="font-size:14px;padding-top:12px;color:#777;">\(([0-9.]+) (kB|KB|MB|GB)\)</div>'
     FILE_OFFLINE_PATTERN = r'class="error_msg_title"> Invalid or Deleted File. </div>'
 
     def setup(self):
@@ -79,16 +80,20 @@ class MediafireCom(Hoster):
 
     def process(self, pyfile):
         self.html = self.load(pyfile.url)
+        self.getFileInfo(pyfile)
+        self.handleFree()
 
+    def getFileInfo(self, pyfile):
+        if re.search(self.FILE_OFFLINE_PATTERN, self.html): self.offline()
+        
         found = re.search(self.FILE_NAME_PATTERN, self.html)
         if not found: self.fail("Parse error (file name)")
         pyfile.name = found.group(1)
 
         found = re.search(self.FILE_SIZE_PATTERN, self.html)
         if found:
-            pyfile.size = float(found.group(1)) * 1024 ** {'kB': 1, 'MB': 2, 'GB': 3}[found.group(2)]
-
-        self.handleFree()
+            size, units = found.group(1), found.group(2).replace('k','K')
+            pyfile.size = float(size) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[units]
 
     def handleFree(self):
         found = re.search(self.RECAPTCHA_PATTERN, self.html)
diff --git a/module/plugins/hoster/MegasharesCom.py b/module/plugins/hoster/MegasharesCom.py
new file mode 100644
index 000000000..0a326c268
--- /dev/null
+++ b/module/plugins/hoster/MegasharesCom.py
@@ -0,0 +1,136 @@
+# -*- coding: utf-8 -*-
+"""
+    This program is free software; you can redistribute it and/or modify
+    it under the terms of the GNU General Public License as published by
+    the Free Software Foundation; either version 3 of the License,
+    or (at your option) any later version.
+
+    This program is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
+    See the GNU General Public License for more details.
+
+    You should have received a copy of the GNU General Public License
+    along with this program; if not, see <http://www.gnu.org/licenses/>.
+
+    @author: zoidberg
+"""
+
+import re
+from time import time
+from module.plugins.Hoster import Hoster
+from module.network.RequestFactory import getURL
+
+def getInfo(urls):
+    result = []
+
+    for url in urls:
+        html = getURL(url, decode=True)
+        if re.search(MegasharesCom.FILE_OFFLINE_PATTERN, html):
+            # File offline
+            result.append((url, 0, 1, url))
+        else:
+            # Get file info
+            name, size = url, 0
+
+            found = re.search(MegasharesCom.FILE_SIZE_PATTERN, html)
+            if found is not None:
+                size, units = found.groups()
+                size = float(size) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[units]
+
+            found = re.search(MegasharesCom.FILE_NAME_PATTERN, html)
+            if found is not None:
+                name = found.group(1)
+
+            if found or size > 0:
+                result.append((name, size, 2, url))
+    yield result
+
+
+class MegasharesCom(Hoster):
+    __name__ = "MegasharesCom"
+    __type__ = "hoster"
+    __pattern__ = r"http://(\w+\.)?megashares.com/.*"
+    __version__ = "0.1"
+    __description__ = """megashares.com plugin - free only"""
+    __author_name__ = ("zoidberg")
+    __author_mail__ = ("zoidberg@mujmail.cz")
+
+    FILE_NAME_PATTERN = '<h1 class="black xxl"[^>]*title="([^"]+)">'
+    FILE_SIZE_PATTERN = '<strong><span class="black">Filesize:</span></strong> ([0-9.]+) (KB|MB|GB)<br />'
+    DOWNLOAD_URL_PATTERN = '<div id="show_download_button_2" style="display:none">\s*<a href="([^"]+)">'
+    PASSPORT_LEFT_PATTERN = 'Your Download Passport is: <[^>]*>(\w+).*\s*You have\s*<[^>]*>\s*([0-9.]+) (KB|MB|GB)'
+    PASSPORT_RENEW_PATTERN = 'Your download passport will renew in\s*<strong>(\d+)</strong>:<strong>(\d+)</strong>:<strong>(\d+)</strong>'
+    REACTIVATE_NUM_PATTERN = r'<input[^>]*id="random_num" value="(\d+)" />'
+    REACTIVATE_PASSPORT_PATTERN = r'<input[^>]*id="passport_num" value="(\w+)" />'
+    REQUEST_URI_PATTERN = r'var request_uri = "([^"]+)";'
+
+    FILE_OFFLINE_PATTERN = r'<dd class="red">Invalid Link Request - file does not exist.</dd>'
+
+    def setup(self):
+        self.multiDL = False
+
+    def process(self, pyfile):
+        self.html = self.load(pyfile.url, decode=True)
+        self.getFileInfo(pyfile)
+        self.handleFree(pyfile)
+
+    def getFileInfo(self, pyfile):
+        if re.search(self.FILE_OFFLINE_PATTERN, self.html): self.offline()
+
+        found = re.search(self.FILE_NAME_PATTERN, self.html)
+        if found is None: self.fail("Parse error (file name)")
+        pyfile.name = found.group(1)
+
+        found = re.search(self.FILE_SIZE_PATTERN, self.html)
+        if found is None: self.fail("Parse error (file size)")
+        pyfile.size = float(found.group(1)) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[found.group(2)]
+
+    def handleFree(self, pyfile):
+        if pyfile.size > 576716800: self.fail("This file is too large for free download")
+
+        # Reactivate passport if needed
+        found = re.search(self.REACTIVATE_PASSPORT_PATTERN, self.html)
+        if found:
+            passport_num = found.group(1)
+            request_uri = re.search(self.REQUEST_URI_PATTERN, self.html).group(1)
+
+            for i in range(5):
+                random_num = re.search(self.REACTIVATE_NUM_PATTERN, self.html).group(1)
+
+                verifyinput = self.decryptCaptcha("http://megashares.com/index.php?secgfx=gfx&random_num=%s" % random_num)
+                self.logInfo("Reactivating passport %s: %s %s" % (passport_num, random_num, verifyinput))
+
+                url = "http://d01.megashares.com%s&rs=check_passport_renewal" % request_uri + \
+                    "&rsargs[]=%s&rsargs[]=%s&rsargs[]=%s" % (verifyinput, random_num, passport_num) + \
+                    "&rsargs[]=replace_sec_pprenewal&rsrnd=%s" % str(int(time()*1000))
+                self.logDebug(url)
+                response = self.load(url)
+
+                if 'Thank you for reactivating your passport.' in response:
+                    self.correctCaptcha()
+                    self.retry(0)
+                else:
+                    self.invalidCaptcha()
+            else: self.fail("Failed to reactivate passport")
+
+        # Check traffic left on passport
+        found = re.search(self.PASSPORT_LEFT_PATTERN, self.html)
+        if not found: self.fail('Passport not found')
+        self.logInfo("Download passport: %s" % found.group(1))
+        data_left = float(found.group(2)) * 1024 ** {'KB': 1, 'MB': 2, 'GB': 3}[found.group(3)]
+        self.logInfo("Data left: %s %s (%d MB needed)" % (found.group(2), found.group(3), pyfile.size / 1048576))
+        if pyfile.size > data_left:
+            found = re.search(self.PASSPORT_RENEW_PATTERN, self.html)
+            if not found: self.fail('Passport renew time not found')
+            renew = found.group(1) + 60 * (found.group(2) + 60 * found.group(3))
+            self.retry(renew, 3, "Unable to get passport")
+
+        # Find download link
+        found = re.search(self.DOWNLOAD_URL_PATTERN, self.html)
+        if not found: self.fail('Download link not found')
+        download_url = found.group(1)
+
+        # Download
+        self.logDebug("Download URL: %s" % download_url)
+        self.download(download_url)
\ No newline at end of file