|
|
@ -6,18 +6,21 @@ __author__ = 'Prinz23' |
|
|
__version__ = '1.0' |
|
|
__version__ = '1.0' |
|
|
__api_version__ = '1.0.0' |
|
|
__api_version__ = '1.0.0' |
|
|
|
|
|
|
|
|
import logging |
|
|
|
|
|
import datetime |
|
|
import datetime |
|
|
|
|
|
import logging |
|
|
|
|
|
import re |
|
|
|
|
|
|
|
|
import requests |
|
|
import requests |
|
|
from requests.packages.urllib3.util.retry import Retry |
|
|
from urllib3.util.retry import Retry |
|
|
from requests.adapters import HTTPAdapter |
|
|
from requests.adapters import HTTPAdapter |
|
|
|
|
|
|
|
|
from six import iteritems |
|
|
from six import iteritems |
|
|
from sg_helpers import get_url, try_int |
|
|
from sg_helpers import get_url, try_int |
|
|
from lib.dateutil.parser import parser |
|
|
from lib.dateutil.parser import parser |
|
|
|
|
|
# noinspection PyProtectedMember |
|
|
from lib.dateutil.tz.tz import _datetime_to_timestamp |
|
|
from lib.dateutil.tz.tz import _datetime_to_timestamp |
|
|
from lib.exceptions_helper import ConnectionSkipException, ex |
|
|
from lib.exceptions_helper import ConnectionSkipException, ex |
|
|
from .tvmaze_exceptions import * |
|
|
# from .tvmaze_exceptions import * |
|
|
from lib.tvinfo_base import TVInfoBase, TVInfoImage, TVInfoImageSize, TVInfoImageType, Character, Crew, \ |
|
|
from lib.tvinfo_base import TVInfoBase, TVInfoImage, TVInfoImageSize, TVInfoImageType, Character, Crew, \ |
|
|
crew_type_names, Person, RoleTypes, TVInfoShow, TVInfoEpisode, TVInfoIDs, TVInfoSeason, PersonGenders, \ |
|
|
crew_type_names, Person, RoleTypes, TVInfoShow, TVInfoEpisode, TVInfoIDs, TVInfoSeason, PersonGenders, \ |
|
|
TVINFO_TVMAZE, TVINFO_TVDB, TVINFO_IMDB |
|
|
TVINFO_TVMAZE, TVINFO_TVDB, TVINFO_IMDB |
|
|
@ -25,7 +28,7 @@ from lib.pytvmaze import tvmaze |
|
|
|
|
|
|
|
|
# noinspection PyUnreachableCode |
|
|
# noinspection PyUnreachableCode |
|
|
if False: |
|
|
if False: |
|
|
from typing import Any, AnyStr, Dict, List, Optional, Union |
|
|
from typing import Any, AnyStr, Dict, List, Optional |
|
|
from six import integer_types |
|
|
from six import integer_types |
|
|
|
|
|
|
|
|
log = logging.getLogger('tvmaze.api') |
|
|
log = logging.getLogger('tvmaze.api') |
|
|
@ -38,8 +41,10 @@ def tvmaze_endpoint_standard_get(url): |
|
|
retries = Retry(total=5, |
|
|
retries = Retry(total=5, |
|
|
backoff_factor=0.1, |
|
|
backoff_factor=0.1, |
|
|
status_forcelist=[429]) |
|
|
status_forcelist=[429]) |
|
|
|
|
|
# noinspection HttpUrlsUsage |
|
|
s.mount('http://', HTTPAdapter(max_retries=retries)) |
|
|
s.mount('http://', HTTPAdapter(max_retries=retries)) |
|
|
s.mount('https://', HTTPAdapter(max_retries=retries)) |
|
|
s.mount('https://', HTTPAdapter(max_retries=retries)) |
|
|
|
|
|
# noinspection PyProtectedMember |
|
|
return get_url(url, json=True, session=s, hooks={'response': tvmaze._record_hook}, raise_skip_exception=True) |
|
|
return get_url(url, json=True, session=s, hooks={'response': tvmaze._record_hook}, raise_skip_exception=True) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@ -140,9 +145,9 @@ class TvMaze(TVInfoBase): |
|
|
'network': s.network and s.network.name, |
|
|
'network': s.network and s.network.name, |
|
|
'genres': s.genres, 'overview': s.summary, |
|
|
'genres': s.genres, 'overview': s.summary, |
|
|
'aliases': [a.name for a in s.akas], 'image': s.image and s.image.get('original'), |
|
|
'aliases': [a.name for a in s.akas], 'image': s.image and s.image.get('original'), |
|
|
'ids': TVInfoIDs(tvdb=s.externals.get('thetvdb'), rage=s.externals.get('tvrage'), tvmaze=s.id, |
|
|
'ids': TVInfoIDs( |
|
|
imdb=s.externals.get('imdb') and try_int(s.externals.get('imdb').replace('tt', ''), |
|
|
tvdb=s.externals.get('thetvdb'), rage=s.externals.get('tvrage'), tvmaze=s.id, |
|
|
None))} |
|
|
imdb=s.externals.get('imdb') and try_int(s.externals.get('imdb').replace('tt', ''), None))} |
|
|
results = [] |
|
|
results = [] |
|
|
if ids: |
|
|
if ids: |
|
|
for t, p in iteritems(ids): |
|
|
for t, p in iteritems(ids): |
|
|
@ -201,11 +206,11 @@ class TvMaze(TVInfoBase): |
|
|
return results |
|
|
return results |
|
|
|
|
|
|
|
|
def _set_episode(self, sid, ep_obj): |
|
|
def _set_episode(self, sid, ep_obj): |
|
|
for _k, _s in [('seasonnumber', 'season_number'), ('episodenumber', 'episode_number'), |
|
|
for _k, _s in ( |
|
|
('episodename', 'title'), ('overview', 'summary'), ('firstaired', 'airdate'), |
|
|
('seasonnumber', 'season_number'), ('episodenumber', 'episode_number'), |
|
|
('airtime', 'airtime'), ('runtime', 'runtime'), |
|
|
('episodename', 'title'), ('overview', 'summary'), ('firstaired', 'airdate'), |
|
|
('seriesid', 'maze_id'), ('id', 'maze_id'), ('is_special', 'special'), |
|
|
('airtime', 'airtime'), ('runtime', 'runtime'), |
|
|
('filename', 'image')]: |
|
|
('seriesid', 'maze_id'), ('id', 'maze_id'), ('is_special', 'special'), ('filename', 'image')): |
|
|
if 'filename' == _k: |
|
|
if 'filename' == _k: |
|
|
image = getattr(ep_obj, _s, {}) or {} |
|
|
image = getattr(ep_obj, _s, {}) or {} |
|
|
image = image.get('original') or image.get('medium') |
|
|
image = image.get('original') or image.get('medium') |
|
|
@ -221,7 +226,8 @@ class TvMaze(TVInfoBase): |
|
|
except (BaseException, Exception): |
|
|
except (BaseException, Exception): |
|
|
pass |
|
|
pass |
|
|
|
|
|
|
|
|
def _set_network(self, show_obj, network, is_stream): |
|
|
@staticmethod |
|
|
|
|
|
def _set_network(show_obj, network, is_stream): |
|
|
show_obj['network'] = network.name |
|
|
show_obj['network'] = network.name |
|
|
show_obj['network_timezone'] = network.timezone |
|
|
show_obj['network_timezone'] = network.timezone |
|
|
show_obj['network_country'] = network.country |
|
|
show_obj['network_country'] = network.country |
|
|
@ -229,17 +235,21 @@ class TvMaze(TVInfoBase): |
|
|
show_obj['network_id'] = network.maze_id |
|
|
show_obj['network_id'] = network.maze_id |
|
|
show_obj['network_is_stream'] = is_stream |
|
|
show_obj['network_is_stream'] = is_stream |
|
|
|
|
|
|
|
|
def _get_show_data(self, sid, language, get_ep_info=False, banners=False, posters=False, seasons=False, |
|
|
def _get_tvm_show(self, show_id, get_ep_info): |
|
|
seasonwides=False, fanart=False, actors=False, **kwargs): |
|
|
|
|
|
log.debug('Getting all series data for %s' % sid) |
|
|
|
|
|
try: |
|
|
try: |
|
|
self.show_not_found = False |
|
|
self.show_not_found = False |
|
|
show_data = tvm_obj.get_show(maze_id=sid, embed='cast%s' % ('', ',episodeswithspecials')[get_ep_info]) |
|
|
return tvm_obj.get_show(maze_id=show_id, embed='cast%s' % ('', ',episodeswithspecials')[get_ep_info]) |
|
|
except tvmaze.ShowNotFound: |
|
|
except tvmaze.ShowNotFound: |
|
|
self.show_not_found = True |
|
|
self.show_not_found = True |
|
|
return False |
|
|
except (BaseException, Exception): |
|
|
except (BaseException, Exception) as e: |
|
|
log.debug('Error getting data for tvmaze show id: %s' % show_id) |
|
|
log.debug('Error getting data for tvmaze show id: %s' % sid) |
|
|
|
|
|
|
|
|
def _get_show_data(self, sid, language, get_ep_info=False, banners=False, posters=False, seasons=False, |
|
|
|
|
|
seasonwides=False, fanart=False, actors=False, **kwargs): |
|
|
|
|
|
log.debug('Getting all series data for %s' % sid) |
|
|
|
|
|
|
|
|
|
|
|
show_data = self._get_tvm_show(sid, get_ep_info) |
|
|
|
|
|
if not show_data: |
|
|
return False |
|
|
return False |
|
|
|
|
|
|
|
|
show_obj = self.shows[sid].__dict__ |
|
|
show_obj = self.shows[sid].__dict__ |
|
|
@ -330,19 +340,20 @@ class TvMaze(TVInfoBase): |
|
|
except (BaseException, Exception): |
|
|
except (BaseException, Exception): |
|
|
print('error') |
|
|
print('error') |
|
|
pass |
|
|
pass |
|
|
existing_person.p_id, existing_person.name, existing_person.image, existing_person.gender, \ |
|
|
(existing_person.p_id, existing_person.name, existing_person.image, existing_person.gender, |
|
|
existing_person.birthdate, existing_person.deathdate, existing_person.country, \ |
|
|
existing_person.birthdate, existing_person.deathdate, existing_person.country, |
|
|
existing_person.country_code, existing_person.country_timezone, existing_person.thumb_url, \ |
|
|
existing_person.country_code, existing_person.country_timezone, existing_person.thumb_url, |
|
|
existing_person.url, existing_person.ids = \ |
|
|
existing_person.url, existing_person.ids) = \ |
|
|
ch.person.id, ch.person.name, ch.person.image and ch.person.image.get('original'), \ |
|
|
(ch.person.id, ch.person.name, |
|
|
PersonGenders.named.get(ch.person.gender and ch.person.gender.lower(), |
|
|
ch.person.image and ch.person.image.get('original'), |
|
|
PersonGenders.unknown),\ |
|
|
PersonGenders.named.get( |
|
|
person.birthdate, person.deathdate,\ |
|
|
ch.person.gender and ch.person.gender.lower(), PersonGenders.unknown), |
|
|
ch.person.country and ch.person.country.get('name'),\ |
|
|
person.birthdate, person.deathdate, |
|
|
ch.person.country and ch.person.country.get('code'),\ |
|
|
ch.person.country and ch.person.country.get('name'), |
|
|
ch.person.country and ch.person.country.get('timezone'),\ |
|
|
ch.person.country and ch.person.country.get('code'), |
|
|
ch.person.image and ch.person.image.get('medium'),\ |
|
|
ch.person.country and ch.person.country.get('timezone'), |
|
|
ch.person.url, {TVINFO_TVMAZE: ch.person.id} |
|
|
ch.person.image and ch.person.image.get('medium'), |
|
|
|
|
|
ch.person.url, {TVINFO_TVMAZE: ch.person.id}) |
|
|
else: |
|
|
else: |
|
|
existing_character.person.append(person) |
|
|
existing_character.person.append(person) |
|
|
else: |
|
|
else: |
|
|
@ -396,7 +407,7 @@ class TvMaze(TVInfoBase): |
|
|
show_obj['ids'] = TVInfoIDs(tvdb=show_data.externals.get('thetvdb'), |
|
|
show_obj['ids'] = TVInfoIDs(tvdb=show_data.externals.get('thetvdb'), |
|
|
rage=show_data.externals.get('tvrage'), |
|
|
rage=show_data.externals.get('tvrage'), |
|
|
imdb=show_data.externals.get('imdb') and |
|
|
imdb=show_data.externals.get('imdb') and |
|
|
try_int(show_data.externals.get('imdb').replace('tt', ''), None)) |
|
|
try_int(show_data.externals.get('imdb').replace('tt', ''), None)) |
|
|
|
|
|
|
|
|
if show_data.network: |
|
|
if show_data.network: |
|
|
self._set_network(show_obj, show_data.network, False) |
|
|
self._set_network(show_obj, show_data.network, False) |
|
|
@ -406,15 +417,8 @@ class TvMaze(TVInfoBase): |
|
|
if get_ep_info and not getattr(self.shows.get(sid), 'ep_loaded', False): |
|
|
if get_ep_info and not getattr(self.shows.get(sid), 'ep_loaded', False): |
|
|
log.debug('Getting all episodes of %s' % sid) |
|
|
log.debug('Getting all episodes of %s' % sid) |
|
|
if None is show_data: |
|
|
if None is show_data: |
|
|
try: |
|
|
show_data = self._get_tvm_show(sid, get_ep_info) |
|
|
self.show_not_found = False |
|
|
if not show_data: |
|
|
show_data = tvm_obj.get_show(maze_id=sid, embed='cast%s' % ('', ',episodeswithspecials')[ |
|
|
|
|
|
get_ep_info]) |
|
|
|
|
|
except tvmaze.ShowNotFound: |
|
|
|
|
|
self.show_not_found = True |
|
|
|
|
|
return False |
|
|
|
|
|
except (BaseException, Exception) as e: |
|
|
|
|
|
log.debug('Error getting data for tvmaze show id: %s' % sid) |
|
|
|
|
|
return False |
|
|
return False |
|
|
|
|
|
|
|
|
if show_data.episodes: |
|
|
if show_data.episodes: |
|
|
@ -543,3 +547,21 @@ class TvMaze(TVInfoBase): |
|
|
p = None |
|
|
p = None |
|
|
if p: |
|
|
if p: |
|
|
return self._convert_person(p) |
|
|
return self._convert_person(p) |
|
|
|
|
|
|
|
|
|
|
|
def get_premieres(self): |
|
|
|
|
|
# type: (...) -> List[tvmaze.Episode] |
|
|
|
|
|
return self.filtered_schedule(lambda e: all([1 == e.season_number, 1 == e.episode_number])) |
|
|
|
|
|
|
|
|
|
|
|
def get_returning(self): |
|
|
|
|
|
# type: (...) -> List[tvmaze.Episode] |
|
|
|
|
|
return self.filtered_schedule(lambda e: all([1 != e.season_number, 1 == e.episode_number])) |
|
|
|
|
|
|
|
|
|
|
|
@staticmethod |
|
|
|
|
|
def filtered_schedule(condition): |
|
|
|
|
|
try: |
|
|
|
|
|
return sorted([ |
|
|
|
|
|
e for e in tvmaze.get_full_schedule() |
|
|
|
|
|
if condition(e) and (None is e.show.language or re.search('(?i)eng|jap', e.show.language))], |
|
|
|
|
|
key=lambda x: x.show.premiered or x.airstamp) |
|
|
|
|
|
except(BaseException, Exception): |
|
|
|
|
|
return [] |
|
|
|
JackDandy
4 years ago