summaryrefslogtreecommitdiffstats
path: root/module/plugins/hoster/BitshareCom.py
diff options
context:
space:
mode:
Diffstat (limited to 'module/plugins/hoster/BitshareCom.py')
-rw-r--r--module/plugins/hoster/BitshareCom.py103
1 files changed, 63 insertions, 40 deletions
diff --git a/module/plugins/hoster/BitshareCom.py b/module/plugins/hoster/BitshareCom.py
index f44a89e23..496071fa3 100644
--- a/module/plugins/hoster/BitshareCom.py
+++ b/module/plugins/hoster/BitshareCom.py
@@ -10,6 +10,7 @@ from module.plugins.Hoster import Hoster
from module.plugins.ReCaptcha import ReCaptcha
from module.network.RequestFactory import getURL
+from wx.lib.analogclock.helpers import Hand
def unicode2str(unitext):
return unicodedata.normalize('NFKD', unitext).encode('ascii', 'ignore')
@@ -19,20 +20,22 @@ def getInfo(urls):
for url in urls:
- # Get html
+ # Get file info html
+ # TODO: Force responses in english language
html = getURL(url)
- if re.search(BitshareCom.OFFLINE_PATTERN, html):
+ if re.search(BitshareCom.FILE_OFFLINE_PATTERN, html):
result.append((url, 0, 1, url))
# Name
name1 = re.search(BitshareCom.__pattern__, url).group('name')
m = re.search(BitshareCom.FILE_INFO_PATTERN, html)
name2 = m.group('name')
- name = unicode2str(max(name1, name2))
+ name = unicode2str(max(name1, name2)) # Unicode BUG workaround
# Size
value = float(m.group('size'))
- pow = {'KB' : 1, 'MB' : 2, 'GB' : 3}[m.group('units')]
+ units = m.group('units')
+ pow = {'KB' : 1, 'MB' : 2, 'GB' : 3}[units]
size = int(value*1024**pow)
# Return info
@@ -44,12 +47,14 @@ class BitshareCom(Hoster):
__name__ = "BitshareCom"
__type__ = "hoster"
__pattern__ = r"http://(www\.)?bitshare\.com/(files/(?P<id1>[a-zA-Z0-9]+)(/(?P<name>.*?)\.html)?|\?f=(?P<id2>[a-zA-Z0-9]+))"
- __version__ = "0.2"
+ __version__ = "0.3"
__description__ = """Bitshare.Com File Download Hoster"""
- __author_name__ = ("paul", "king")
+ __author_name__ = ("paul", "king", "fragonib")
- OFFLINE_PATTERN = r'''(>We are sorry, but the requested file was not found in our database|>Error - File not available<|The file was deleted either by the uploader, inactivity or due to copyright claim)'''
+ FILE_OFFLINE_PATTERN = r'''(>We are sorry, but the requested file was not found in our database|>Error - File not available<|The file was deleted either by the uploader, inactivity or due to copyright claim)'''
FILE_INFO_PATTERN = r'<h1>.*\s(?P<name>.+?)\s-\s(?P<size>\d+)\s(?P<units>..)yte</h1>'
+ FILE_AJAXID_PATTERN = r'var ajaxdl = "(.*?)";'
+ CAPTCHA_KEY_PATTERN = r"http://api\.recaptcha\.net/challenge\?k=(.*?) "
def setup(self):
self.multiDL = False
@@ -57,74 +62,92 @@ class BitshareCom(Hoster):
def process(self, pyfile):
self.pyfile = pyfile
+
+ # Force responses language
self.req.cj.setCookie("bitshare.com", "language_selection", "EN")
# File id
m = re.match(self.__pattern__, self.pyfile.url)
self.file_id = max(m.group('id1'), m.group('id2'))
-
- # File url
- self.log.debug("%s: File_id is %s" % (self.__name__, self.file_id))
+ self.log.debug("%s: File id is [%s]" % (self.__name__, self.file_id))
# Load main page
self.html = self.load(self.pyfile.url, ref=False, utf8=True, cookies=True)
# Check offline
- if re.search(self.OFFLINE_PATTERN, self.html) is not None:
+ if re.search(self.FILE_OFFLINE_PATTERN, self.html) is not None:
self.offline()
# File name
name1 = re.search(BitshareCom.__pattern__, self.pyfile.url).group('name')
name2 = re.search(BitshareCom.FILE_INFO_PATTERN, self.html).group('name')
- self.pyfile.name = unicode2str(max(name1, name2))
+ self.pyfile.name = unicode2str(max(name1, name2)) # Unicode BUG workaround
- self.ajaxid = re.search("var ajaxdl = \"(.*?)\";",self.html).group(1)
-
- self.log.debug("%s: AjaxId %s" % (self.__name__, self.ajaxid))
+ # Ajax file id
+ self.ajaxid = re.search(BitshareCom.FILE_AJAXID_PATTERN, self.html).group(1)
+ self.log.debug("%s: File ajax id is [%s]" % (self.__name__, self.ajaxid))
+ # Handle free downloading
self.handleFree()
def handleFree(self):
- action = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
+ # Get download info
+ self.log.debug("%s: Getting download info" % (self.__name__))
+ response = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
post={"request" : "generateID", "ajaxid" : self.ajaxid})
- self.log.debug("%s: Result of generateID %s" % (self.__name__, action))
- parts = action.split(":")
-
- if parts[0] == "ERROR":
- self.fail(parts[1])
-
+ self.handleErrors(response, ':')
+ parts = response.split(":")
filetype = parts[0]
wait = int(parts[1])
captcha = int(parts[2])
+ self.log.debug("%s: Download info [type: '%s', waiting: %d, captcha: %d]" %
+ (self.__name__, filetype, wait, captcha))
+ # Waiting
if wait > 0:
- self.log.info("%s: Waiting %d seconds." % (self.__name__, wait))
+ self.log.debug("%s: Waiting %d seconds." % (self.__name__, wait))
self.setWait(wait, True)
self.wait()
+ # Resolve captcha
if captcha == 1:
- id = re.search(r"http://api\.recaptcha\.net/challenge\?k=(.*?) ", self.html).group(1)
- self.log.debug("%s: ReCaptcha key %s" % (self.__name__, id))
- for i in range(3): # Try upto 3 times
+ self.log.debug("%s: File is captcha protected" % (self.__name__))
+ id = re.search(BitshareCom.CAPTCHA_KEY_PATTERN, self.html).group(1)
+ # Try up to 3 times
+ for i in range(3):
+ self.log.debug("%s: Resolving ReCaptcha with key [%s], round %d" % (self.__name__, id, i+1))
recaptcha = ReCaptcha(self)
challenge, code = recaptcha.challenge(id)
- action = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
+ response = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
post={"request" : "validateCaptcha", "ajaxid" : self.ajaxid, "recaptcha_challenge_field" : challenge, "recaptcha_response_field" : code})
- parts = action.split(":")
- if parts[0] != "SUCCESS":
- self.invalidCaptcha()
- else:
+ if self.handleCaptchaErrors(response):
break
- action = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
- post={"request" : "getDownloadURL", "ajaxid" : self.ajaxid})
- parts = action.split("#")
-
- if parts[0] == "ERROR":
- self.fail(parts[1])
+ # Get download URL
+ self.log.debug("%s: Getting download url" % (self.__name__))
+ response = self.load("http://bitshare.com/files-ajax/" + self.file_id + "/request.html",
+ post={"request" : "getDownloadURL", "ajaxid" : self.ajaxid})
+ self.handleErrors(response, '#')
+ url = response.split("#")[-1]
- # this may either download our file or forward us to an error page
- self.log.debug("%s: Download url %s" % (self.__name__, parts[1]))
- dl = self.download(parts[1])
+ # Request download URL
+ # This may either download our file or forward us to an error page
+ self.log.debug("%s: Downloading file with url [%s]" % (self.__name__, url))
+ dl = self.download(url)
+
+ def handleErrors(self, response, separator):
+ self.log.debug("%s: Checking response [%s]" % (self.__name__, response))
+ if "ERROR" in response:
+ msg = response.split(separator)[-1]
+ self.fail(msg)
+
+ def handleCaptchaErrors(self, response):
+ self.log.debug("%s: Result of captcha resolving [%s]" % (self.__name__, response))
+ if "SUCCESS" in response:
+ return True
+
+ self.log.debug("%s: Wrong captcha" % (self.__name__))
+ self.invalidCaptcha()
+