This repository was archived by the owner on Jul 4, 2026. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathtvdb.py
More file actions
126 lines (107 loc) · 4.75 KB
/
Copy pathtvdb.py
File metadata and controls
126 lines (107 loc) · 4.75 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
# -*- coding: utf-8 -*-
from collections.abc import Sequence
import tvdb_api
from .api_utils import d
api_key = 'RS\nvT<%<g#~bsS~3'
class TvdbResult(object):
def __init__(self, show, season, episode=None):
self.show = show
self.season = season
self.episode = episode
def banner(self, season_number):
# todo offer choice of cover if multiple?
# todo cache banner upload per season?
# get best banner, preferably for season
def best_banner(banners):
def get_rating(banner):
return banner.get('ratingsInfo', {}).get('average', 0)
sorted_banners = sorted(banners, key=get_rating)
return sorted_banners[-1]
try:
season_banners = self.show['_banners']['season']
best_banner = best_banner(
[banner for banner in season_banners['raw']
if banner['subKey'] == str(season_number)])
return (season_banners[best_banner['resolution']]
[best_banner['id']]['_bannerpath'])
except (IndexError, KeyError):
for key in ('poster', 'series', 'fanart'):
try:
series_banners = self.show['_banners'][key]
best_banner = best_banner(series_banners['raw'])
return (series_banners[best_banner['resolution']]
[best_banner['id']]['_bannerpath'])
except (IndexError, KeyError):
pass
raise Exception('Unable to find cover')
def summary(self):
return {
'title': self.show['seriesName'],
'network': self.show['network'],
'genres': self.show['genre'],
'seriessummary': self.show['overview'],
'cast': self.show['_actors'],
# 'seasons': len(self.show),
# 'status': self.show['status'],
'contentrating': self.show['rating'],
'show_imdb_id': self.show['imdbId'],
}
class TvdbSeason(TvdbResult):
def summary(self):
s = super(TvdbSeason, self).summary()
some_episode = next(iter(self.season.values()))
season_number = some_episode['airedSeason']
series_url = 'https://thetvdb.com/series/%s' % (self.show['slug'],)
s.update(**{'num_episodes': len(self.season),
'episodes': []})
for episode_number in self.season:
episode = self.season[episode_number]
episode_url = 'https://thetvdb.com/series/{}/episodes/{}'.format(
self.show['slug'], episode['id'])
s["episodes"].append({
'title': episode['episodeName'],
'url': episode_url,
'imdb_id': episode['imdbId'],
'rating': episode['siteRating']})
s['url'] = series_url
s['cover'] = self.banner(season_number)
s['season'] = season_number
return s
class TvdbEpisode(TvdbResult):
def show_summary(self):
return super(TvdbEpisode, self).summary()
def summary(self):
summary = self.show_summary()
summary.update(**{
'season': self.episode['airedSeason'],
'episode': self.episode['airedEpisodeNumber'],
'episode_title': self.episode['episodeName'],
'imdb_id': self.episode['imdbId'],
'directors': self.episode['directors'],
'air_date': self.episode['firstAired'],
# 'air_dow': self.show['airsDayOfWeek'],
# 'air_time': self.show['airsTime'],
'writers': self.episode['writers'],
'rating': self.episode['siteRating'],
'votes': self.episode['siteRatingCount'],
'episodesummary': self.episode['overview'],
'language': self.episode['language'],
'url': 'https://thetvdb.com/series/{}'.format(
self.show['slug']),
'cover': self.banner(self.episode['airedSeason'])})
return summary
class TVDB(object):
def __init__(self, interactive=True):
self.tvdb = tvdb_api.Tvdb(interactive=interactive, banners=True,
actors=True, apikey=d(api_key))
def search(self, tv_specifier, tvdb_id=None):
show = self.tvdb[tvdb_id or tv_specifier.title]
season = show[tv_specifier.season]
if tv_specifier.episode is not None:
if not isinstance(tv_specifier.episode, Sequence):
episode = season[tv_specifier.episode]
return [TvdbEpisode(show, season, episode)]
# multi-episode
return [TvdbEpisode(show, season, season[e])
for e in tv_specifier.episode]
return TvdbSeason(show, season)