Merge pull request #689 from ArjixWasTaken/patch-32

Improved the logic of the extractor selection in animerush

anime_downloader/extractors/yourupload.py

@@ -3,14 +3,21 @@
 
 from anime_downloader.extractors.base_extractor import BaseExtractor
 from anime_downloader.sites import helpers
+from requests.exceptions import HTTPError
 
 logger = logging.getLogger(__name__)
 
 
 class Yourupload(BaseExtractor):
     def _get_data(self):
         regex = r"file: '([^']*)"
-        file = re.search(regex, helpers.get(self.url).text).group(1)
+        try:
+            response = helpers.get(self.url)
+        except HTTPError:
+            logger.error('File not found.')
+            return {'stream_url': ''}
+
+        file = re.search(regex, response.text).group(1)
         return {
             'stream_url': file,
             'referer': self.url

anime_downloader/sites/animerush.py

@@ -1,7 +1,9 @@
 from anime_downloader.sites.anime import Anime, AnimeEpisode, SearchResult
 from anime_downloader.sites import helpers
-from anime_downloader.extractors import get_extractor
+from anime_downloader.extractors.init import ALL_EXTRACTORS
+
 import logging
+import re
 
 logger = logging.getLogger(__name__)
 
@@ -41,12 +43,20 @@ def _get_sources(self):
         sources_list = []
         # Sources [0] is the url [1] is the name of the source
         # eg: [['https://mp4upload.com/embed-r07potgdvbkr-650x370.html', 'Mp4upload Video']]
+        domain_regex = r"\/\/(?:\w{3,6}\.)?(.*?)\."
         for i in sources:
-            # Not exactly ideal setup for more extractors
-            # If more advanced sources needs to get added look at watchmovie or darkanime
-            server = 'yourupload' if 'yourupload' in i[0] else 'mp4upload'
+            found = False
+            domain = re.findall(domain_regex, i[0])[0].lower()
+
+            for extractor in ALL_EXTRACTORS:
+                if re.match(extractor['regex'], domain):
+                    found = True
+
+            if not found:
+                continue
+
             sources_list.append({
-                'extractor': server,
+                'extractor': domain,
                 'url': i[0],
                 'server': i[1],
                 'version': 'subbed'