Merge pull request #689 from ArjixWasTaken/patch-32
Improved the logic of the extractor selection in animerushmaster
commit
01f44f66b9
|
@ -3,6 +3,7 @@ import re
|
|||
|
||||
from anime_downloader.extractors.base_extractor import BaseExtractor
|
||||
from anime_downloader.sites import helpers
|
||||
from requests.exceptions import HTTPError
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
@ -10,7 +11,13 @@ 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
|
||||
|
|
|
@ -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 @@ class AnimeRushEpisode(AnimeEpisode, sitename='animerush'):
|
|||
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'
|
||||
|
|
Loading…
Reference in New Issue