diff options
Diffstat (limited to 'module/plugins/hoster/BitshareCom.py')
| -rw-r--r-- | module/plugins/hoster/BitshareCom.py | 103 | 
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() +         | 
