def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for a MediaBrowser-style series.xml
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
tv_node = Element("Series")
indexer_lang = show_obj.lang or sickrage.app.config.indexer_default_language
lINDEXER_API_PARMS = IndexerApi(show_obj.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if show_obj.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(show_obj.indexer).indexer(**lINDEXER_API_PARMS)
myShow = t[int(show_obj.indexer_id)]
if not myShow:
return False
# check for title and id
if not (getattr(myShow, 'seriesname', None)
and getattr(myShow, 'id', None)):
sickrage.app.log.info("Incomplete info for show with id " +
str(show_obj.indexer_id) + " on " +
IndexerApi(show_obj.indexer).name +
", skipping it")
return False
if getattr(myShow, 'id', None):
indexer_id = SubElement(tv_node, "id")
indexer_id.text = str(myShow['id'])
if getattr(myShow, 'seriesname', None):
SeriesName = SubElement(tv_node, "SeriesName")
SeriesName.text = myShow['seriesname']
if getattr(myShow, 'status', None):
Status = SubElement(tv_node, "Status")
Status.text = myShow['status']
if getattr(myShow, 'network', None):
Network = SubElement(tv_node, "Network")
Network.text = myShow['network']
if getattr(myShow, 'airstime', None):
Airs_Time = SubElement(tv_node, "Airs_Time")
Airs_Time.text = myShow['airstime']
if getattr(myShow, 'airsdayofweek', None):
Airs_DayOfWeek = SubElement(tv_node, "Airs_DayOfWeek")
Airs_DayOfWeek.text = myShow['airsdayofweek']
FirstAired = SubElement(tv_node, "FirstAired")
if getattr(myShow, 'firstaired', None):
FirstAired.text = myShow['firstaired']
if getattr(myShow, 'contentrating', None):
ContentRating = SubElement(tv_node, "ContentRating")
ContentRating.text = myShow['contentrating']
MPAARating = SubElement(tv_node, "MPAARating")
MPAARating.text = myShow['contentrating']
certification = SubElement(tv_node, "certification")
certification.text = myShow['contentrating']
MetadataType = SubElement(tv_node, "Type")
MetadataType.text = "Series"
if getattr(myShow, 'overview', None):
Overview = SubElement(tv_node, "Overview")
Overview.text = myShow['overview']
if getattr(myShow, 'firstaired', None):
PremiereDate = SubElement(tv_node, "PremiereDate")
PremiereDate.text = myShow['firstaired']
if getattr(myShow, 'rating', None):
Rating = SubElement(tv_node, "Rating")
Rating.text = myShow['rating']
if getattr(myShow, 'firstaired', None):
try:
year_text = str(
datetime.datetime.strptime(myShow['firstaired'],
dateFormat).year)
if year_text:
ProductionYear = SubElement(tv_node, "ProductionYear")
ProductionYear.text = year_text
except Exception:
pass
if getattr(myShow, 'runtime', None):
RunningTime = SubElement(tv_node, "RunningTime")
RunningTime.text = myShow['runtime']
Runtime = SubElement(tv_node, "Runtime")
Runtime.text = myShow['runtime']
if getattr(myShow, 'imdbid', None):
imdb_id = SubElement(tv_node, "IMDB_ID")
imdb_id.text = myShow['imdbid']
imdb_id = SubElement(tv_node, "IMDB")
imdb_id.text = myShow['imdbid']
imdb_id = SubElement(tv_node, "IMDbId")
imdb_id.text = myShow['imdbid']
if getattr(myShow, 'zap2itid', None):
Zap2ItId = SubElement(tv_node, "Zap2ItId")
Zap2ItId.text = myShow['zap2itid']
if getattr(myShow, 'genre', None) and isinstance(myShow["genre"], str):
Genres = SubElement(tv_node, "Genres")
for genre in myShow['genre'].split('|'):
if genre.strip():
cur_genre = SubElement(Genres, "Genre")
cur_genre.text = genre.strip()
Genre = SubElement(tv_node, "Genre")
Genre.text = "|".join(
[x.strip() for x in myShow["genre"].split('|') if x.strip()])
if getattr(myShow, 'network', None):
Studios = SubElement(tv_node, "Studios")
Studio = SubElement(Studios, "Studio")
Studio.text = myShow['network']
Persons = SubElement(tv_node, "Persons")
for actor in t.actors(int(show_obj.indexer_id)):
if not ('name' in actor and actor['name'].strip()):
continue
cur_actor = SubElement(Persons, "Person")
cur_actor_name = SubElement(cur_actor, "Name")
cur_actor_name.text = actor['name'].strip()
cur_actor_type = SubElement(cur_actor, "Type")
cur_actor_type.text = "Actor"
if 'role' in actor and actor['role'].strip():
cur_actor_role = SubElement(cur_actor, "Role")
cur_actor_role.text = actor['role'].strip()
indent_xml(tv_node)
data = ElementTree(tv_node)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a MediaBrowser style episode.xml
and returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
persons_dict = {'Director': [], 'GuestStar': [], 'Writer': []}
indexer_lang = ep_obj.show.lang or sickrage.app.config.indexer_default_language
lINDEXER_API_PARMS = IndexerApi(ep_obj.show.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if ep_obj.show.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(ep_obj.show.indexer).indexer(**lINDEXER_API_PARMS)
myShow = t[ep_obj.show.indexer_id]
if not myShow:
return False
rootNode = Element("Item")
# write an MediaBrowser XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
except (indexer_episodenotfound, indexer_seasonnotfound):
sickrage.app.log.info(
"Unable to find episode %dx%d on %s... has it been removed? Should I delete from db?"
% (curEpToWrite.season, curEpToWrite.episode,
IndexerApi(ep_obj.show.indexer).name))
return None
if curEpToWrite == ep_obj:
# root (or single) episode
# default to today's date for specials if firstaired is not set
if ep_obj.season == 0 and not getattr(myEp, 'firstaired',
None):
myEp['firstaired'] = str(datetime.date.min)
if not (getattr(myEp, 'episodename', None)
and getattr(myEp, 'firstaired', None)):
return None
episode = rootNode
if curEpToWrite.name:
EpisodeName = SubElement(episode, "EpisodeName")
EpisodeName.text = curEpToWrite.name
EpisodeNumber = SubElement(episode, "EpisodeNumber")
EpisodeNumber.text = str(ep_obj.episode)
if ep_obj.related_episodes:
EpisodeNumberEnd = SubElement(episode, "EpisodeNumberEnd")
EpisodeNumberEnd.text = str(curEpToWrite.episode)
SeasonNumber = SubElement(episode, "SeasonNumber")
SeasonNumber.text = str(curEpToWrite.season)
if not ep_obj.related_episodes and getattr(
myEp, 'absolute_number', None):
absolute_number = SubElement(episode, "absolute_number")
absolute_number.text = str(myEp['absolute_number'])
if curEpToWrite.airdate > datetime.date.min:
FirstAired = SubElement(episode, "FirstAired")
FirstAired.text = str(curEpToWrite.airdate)
MetadataType = SubElement(episode, "Type")
MetadataType.text = "Episode"
if curEpToWrite.description:
Overview = SubElement(episode, "Overview")
Overview.text = curEpToWrite.description
if not ep_obj.related_episodes:
if getattr(myEp, 'rating', None):
Rating = SubElement(episode, "Rating")
Rating.text = myEp['rating']
if getattr(myShow, 'imdb_id', None):
IMDB_ID = SubElement(episode, "IMDB_ID")
IMDB_ID.text = myShow['imdb_id']
IMDB = SubElement(episode, "IMDB")
IMDB.text = myShow['imdb_id']
IMDbId = SubElement(episode, "IMDbId")
IMDbId.text = myShow['imdb_id']
indexer_id = SubElement(episode, "id")
indexer_id.text = str(curEpToWrite.indexer_id)
# fill in Persons section with collected directors, guest starts and writers
Persons = SubElement(episode, "Persons")
for person_type, names in persons_dict.items():
# remove doubles
names = list(set(names))
for cur_name in names:
Person = SubElement(Persons, "Person")
cur_person_name = SubElement(Person, "Name")
cur_person_name.text = cur_name
cur_person_type = SubElement(Person, "Type")
cur_person_type.text = person_type
for actor in t.actors(int(ep_obj.show.indexer_id)):
if not ('name' in actor and actor['name'].strip()):
continue
cur_actor = SubElement(Persons, "Person")
cur_actor_name = SubElement(cur_actor, "Name")
cur_actor_name.text = actor['name'].strip()
cur_actor_type = SubElement(cur_actor, "Type")
cur_actor_type.text = "Actor"
if 'role' in actor and actor['role'].strip():
cur_actor_role = SubElement(cur_actor, "Role")
cur_actor_role.text = actor['role'].strip()
Language = SubElement(episode, "Language")
try:
Language.text = myEp['language']['overview']
except Exception:
Language.text = sickrage.app.config.indexer_default_language
thumb = SubElement(episode, "filename")
# TODO: See what this is needed for.. if its still needed
# just write this to the NFO regardless of whether it actually exists or not
# note: renaming files after nfo generation will break this, tough luck
thumb_text = self.get_episode_thumb_path(ep_obj)
if thumb_text:
thumb.text = thumb_text
else:
# append data from (if any) related episodes
EpisodeNumberEnd.text = str(curEpToWrite.episode)
if curEpToWrite.name:
if not EpisodeName.text:
EpisodeName.text = curEpToWrite.name
else:
EpisodeName.text = ', '.join(
[EpisodeName.text, curEpToWrite.name])
if curEpToWrite.description:
if not Overview.text:
Overview.text = curEpToWrite.description
else:
Overview.text = '\r'.join(
[Overview.text, curEpToWrite.description])
# collect all directors, guest stars and writers
if getattr(myEp, 'director', None):
persons_dict['Director'] += [
x.strip() for x in myEp['director'].split('|')
if x.strip()
]
if getattr(myEp, 'gueststars', None):
persons_dict['GuestStar'] += [
x.strip() for x in myEp['gueststars'].split('|')
if x.strip()
]
if getattr(myEp, 'writer', None):
persons_dict['Writer'] += [
x.strip() for x in myEp['writer'].split('|') if x.strip()
]
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a MediaBrowser style episode.xml
and returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
indexer_lang = ep_obj.show.lang or sickrage.app.config.indexer_default_language
try:
# There's gotta be a better way of doing this but we don't wanna
# change the language value elsewhere
lINDEXER_API_PARMS = IndexerApi(ep_obj.show.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if ep_obj.show.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(ep_obj.show.indexer).indexer(**lINDEXER_API_PARMS)
myShow = t[ep_obj.show.indexer_id]
except indexer_shownotfound as e:
raise ShowNotFoundException(str(e))
except indexer_error as e:
sickrage.app.log.error(
"Unable to connect to TVDB while creating meta files - skipping - {}".format(e))
return False
rootNode = Element("details")
movie = SubElement(rootNode, "movie")
movie.attrib["isExtra"] = "false"
movie.attrib["isSet"] = "false"
movie.attrib["isTV"] = "true"
# write an MediaBrowser XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
except (indexer_episodenotfound, indexer_seasonnotfound):
sickrage.app.log.info(
"Unable to find episode %dx%d on %s... has it been removed? Should I delete from db?" %
(curEpToWrite.season, curEpToWrite.episode, IndexerApi(ep_obj.show.indexer).name))
return None
if curEpToWrite == ep_obj:
# root (or single) episode
# default to today's date for specials if firstaired is not set
if curEpToWrite.season == 0 and not getattr(myEp, 'firstaired', None):
myEp['firstaired'] = str(datetime.date.min)
if not (getattr(myEp, 'episodename', None) and getattr(myEp, 'firstaired', None)):
return None
episode = movie
if curEpToWrite.name:
EpisodeName = SubElement(episode, "title")
EpisodeName.text = curEpToWrite.name
SeasonNumber = SubElement(episode, "season")
SeasonNumber.text = str(curEpToWrite.season)
EpisodeNumber = SubElement(episode, "episode")
EpisodeNumber.text = str(curEpToWrite.episode)
if getattr(myShow, "firstaired", None):
try:
year_text = str(datetime.datetime.strptime(myShow["firstaired"], dateFormat).year)
if year_text:
year = SubElement(episode, "year")
year.text = year_text
except:
pass
if getattr(myShow, "overview", None):
plot = SubElement(episode, "plot")
plot.text = myShow["overview"]
if curEpToWrite.description:
Overview = SubElement(episode, "episodeplot")
Overview.text = curEpToWrite.description
if getattr(myShow, 'contentrating', None):
mpaa = SubElement(episode, "mpaa")
mpaa.text = myShow["contentrating"]
if not ep_obj.related_episodes and getattr(myEp, "rating", None):
try:
rating = int((float(myEp['rating']) * 10))
except ValueError:
rating = 0
if rating:
Rating = SubElement(episode, "rating")
Rating.text = str(rating)
if getattr(myEp, 'director', None):
director = SubElement(episode, "director")
director.text = myEp['director']
if getattr(myEp, 'writer', None):
writer = SubElement(episode, "credits")
writer.text = myEp['writer']
cast = SubElement(episode, "cast")
if getattr(myEp, 'gueststars', None) and isinstance(myEp['gueststars'], str):
for actor in (x.strip() for x in myEp['gueststars'].split('|') if x.strip()):
cur_actor = SubElement(cast, "actor")
cur_actor.text = actor
for actor in t.actors(int(ep_obj.show.indexer_id)):
if 'name' in actor and actor['name'].strip():
cur_actor = SubElement(cast, "actor")
cur_actor.text = actor['name'].strip()
else:
# append data from (if any) related episodes
if curEpToWrite.name:
if not EpisodeName.text:
EpisodeName.text = curEpToWrite.name
else:
EpisodeName.text = EpisodeName.text + ", " + curEpToWrite.name
if curEpToWrite.description:
if not Overview.text:
Overview.text = curEpToWrite.description
else:
Overview.text = Overview.text + "\r" + curEpToWrite.description
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for a MediaBrowser-style series.xml
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
indexer_lang = show_obj.lang or sickrage.app.config.indexer_default_language
lINDEXER_API_PARMS = IndexerApi(show_obj.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if show_obj.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(show_obj.indexer).indexer(**lINDEXER_API_PARMS)
rootNode = Element("details")
tv_node = SubElement(rootNode, "movie")
tv_node.attrib["isExtra"] = "false"
tv_node.attrib["isSet"] = "false"
tv_node.attrib["isTV"] = "true"
try:
myShow = t[int(show_obj.indexer_id)]
except indexer_shownotfound:
sickrage.app.log.error(
"Unable to find show with id " + str(show_obj.indexer_id) + " on tvdb, skipping it")
raise
except indexer_error:
sickrage.app.log.error("TVDB is down, can't use its data to make the NFO")
raise
# check for title and id
if not (getattr(myShow, 'seriesname', None) and getattr(myShow, 'id', None)):
sickrage.app.log.info("Incomplete info for "
"show with id " + str(show_obj.indexer_id) + " on " + IndexerApi(show_obj.indexer).name + ", skipping it")
return False
SeriesName = SubElement(tv_node, "title")
SeriesName.text = myShow['seriesname']
if getattr(myShow, "genre", None):
Genres = SubElement(tv_node, "genres")
for genre in myShow['genre'].split('|'):
if genre and genre.strip():
cur_genre = SubElement(Genres, "Genre")
cur_genre.text = genre.strip()
if getattr(myShow, 'firstaired', None):
FirstAired = SubElement(tv_node, "premiered")
FirstAired.text = myShow['firstaired']
if getattr(myShow, "firstaired", None):
try:
year_text = str(datetime.datetime.strptime(myShow["firstaired"], dateFormat).year)
if year_text:
year = SubElement(tv_node, "year")
year.text = year_text
except Exception:
pass
if getattr(myShow, 'overview', None):
plot = SubElement(tv_node, "plot")
plot.text = myShow["overview"]
if getattr(myShow, 'rating', None):
try:
rating = int(float(myShow['rating']) * 10)
except ValueError:
rating = 0
if rating:
Rating = SubElement(tv_node, "rating")
Rating.text = str(rating)
if getattr(myShow, 'status', None):
Status = SubElement(tv_node, "status")
Status.text = myShow['status']
if getattr(myShow, "contentrating", None):
mpaa = SubElement(tv_node, "mpaa")
mpaa.text = myShow["contentrating"]
if getattr(myShow, 'imdb_id', None):
imdb_id = SubElement(tv_node, "id")
imdb_id.attrib["moviedb"] = "imdb"
imdb_id.text = myShow['imdb_id']
if getattr(myShow, 'id', None):
indexer_id = SubElement(tv_node, "indexer_id")
indexer_id.text = str(myShow['id'])
if getattr(myShow, 'runtime', None):
Runtime = SubElement(tv_node, "runtime")
Runtime.text = myShow['runtime']
cast = SubElement(tv_node, "cast")
for actor in t.actors(int(show_obj.indexer_id)):
if 'name' in actor and actor['name'].strip():
cur_actor = SubElement(cast, "actor")
cur_actor.text = actor['name'].strip()
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for an KODI-style episode.nfo and
returns the resulting data object.
show_obj: a TVEpisode instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
series_provider_language = ep_obj.show.lang or sickrage.app.config.general.series_provider_default_language
series_info = ep_obj.show.series_provider.get_series_info(ep_obj.show.series_id, language=series_provider_language)
if not series_info:
return False
if len(eps_to_write) > 1:
root_node = Element("kodimultiepisode")
else:
root_node = Element("episodedetails")
# write an NFO containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
series_episode_info = series_info[curEpToWrite.season][curEpToWrite.episode]
except (SeriesProviderEpisodeNotFound, SeriesProviderSeasonNotFound):
sickrage.app.log.info(
f"Unable to find episode {curEpToWrite.season:d}x{curEpToWrite.episode:d} on {ep_obj.show.series_provider.name}"
f"... has it been removed? Should I delete from db?")
return None
if not getattr(series_episode_info, 'firstAired', None):
series_episode_info["firstAired"] = str(datetime.date.min)
if not getattr(series_episode_info, 'name', None):
sickrage.app.log.debug("Not generating nfo because the ep has no title")
return None
sickrage.app.log.debug(f"Creating metadata for episode {ep_obj.season}x{ep_obj.episode}")
if len(eps_to_write) > 1:
episode = SubElement(root_node, "episodedetails")
else:
episode = root_node
if getattr(series_episode_info, 'name', None):
title = SubElement(episode, "title")
title.text = series_episode_info['name']
if getattr(series_info, 'name', None):
showtitle = SubElement(episode, "showtitle")
showtitle.text = series_info['name']
season = SubElement(episode, "season")
season.text = str(curEpToWrite.season)
episodenum = SubElement(episode, "episode")
episodenum.text = str(curEpToWrite.episode)
uniqueid = SubElement(episode, "uniqueid")
uniqueid.text = str(curEpToWrite.series_id)
if curEpToWrite.airdate > datetime.date.min:
aired = SubElement(episode, "aired")
aired.text = str(curEpToWrite.airdate)
if getattr(series_episode_info, 'overview', None):
plot = SubElement(episode, "plot")
plot.text = series_episode_info['overview']
if curEpToWrite.season and getattr(series_info, 'runtime', None):
runtime = SubElement(episode, "runtime")
runtime.text = series_info["runtime"]
if getattr(series_episode_info, 'airsbefore_season', None):
displayseason = SubElement(episode, "displayseason")
displayseason.text = series_episode_info['airsbefore_season']
if getattr(series_episode_info, 'airsbefore_episode', None):
displayepisode = SubElement(episode, "displayepisode")
displayepisode.text = series_episode_info['airsbefore_episode']
if getattr(series_episode_info, 'filename', None):
thumb = SubElement(episode, "thumb")
thumb.text = series_episode_info['filename'].strip()
# watched = SubElement(episode, "watched")
# watched.text = 'false'
if getattr(series_episode_info, 'rating', None):
rating = SubElement(episode, "rating")
rating.text = str(series_episode_info['rating'])
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(episode, "actor")
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
if person['imageUrl'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = person['imageUrl'].strip()
elif person['role'].strip() == 'Writer':
ep_credits = SubElement(episode, "credits")
ep_credits.text = series_episode_info['writer'].strip()
elif person['role'].strip() == 'Director':
director = SubElement(episode, "director")
director.text = series_episode_info['director'].strip()
elif person['role'].strip() == 'Guest Star':
cur_actor = SubElement(episode, "actor")
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
# Make it purdy
indent_xml(root_node)
data = ElementTree(root_node)
return data
def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for an KODI-style tvshow.nfo and
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
tv_node = Element("tvshow")
series_provider_language = show_obj.lang or sickrage.app.config.general.series_provider_default_language
series_info = show_obj.series_provider.get_series_info(show_obj.series_id, language=series_provider_language)
if not series_info:
return False
# check for title and id
if not (getattr(series_info, 'name', None) and getattr(series_info, 'id', None)):
sickrage.app.log.info("Incomplete info for show with id " + str(show_obj.series_id) + " on " + show_obj.series_provider.name + ", skipping it")
return False
title = SubElement(tv_node, "title")
title.text = series_info["name"]
if getattr(series_info, 'rating', None):
rating = SubElement(tv_node, "rating")
rating.text = str(series_info["rating"])
if getattr(series_info, 'firstAired', None):
try:
year_text = str(datetime.datetime.strptime(series_info["firstAired"], dateFormat).year)
if year_text:
year = SubElement(tv_node, "year")
year.text = year_text
except Exception:
pass
if getattr(series_info, 'overview', None):
plot = SubElement(tv_node, "plot")
plot.text = series_info["overview"]
# if getattr(series_info, 'id', None):
# episodeguide = SubElement(tv_node, "episodeguide")
# episodeguideurl = SubElement(episodeguide, "url")
# episodeguideurl.text = IndexerApi(show_obj.series_provider_id).config['base_url'] + str(
# series_info["id"]) + '/all/en.zip'
# if getattr(series_info, 'contentrating', None):
# mpaa = SubElement(tv_node, "mpaa")
# mpaa.text = series_info["contentrating"]
if getattr(series_info, 'id', None):
series_id = SubElement(tv_node, "id")
series_id.text = str(series_info["id"])
if getattr(series_info, 'genres', None):
genre = SubElement(tv_node, "genre")
genre.text = " / ".join(x['name'] for x in series_info["genres"])
if getattr(series_info, 'firstAired', None):
premiered = SubElement(tv_node, "premiered")
premiered.text = series_info["firstAired"]
if getattr(series_info, 'network', None):
studio = SubElement(tv_node, "studio")
studio.text = series_info["network"].strip()
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(tv_node, "actor")
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
if person['imageUrl'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = person['imageUrl'].strip()
# Make it purdy
indent_xml(tv_node)
data = ElementTree(tv_node)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a MediaBrowser style episode.xml
and returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
series_provider_language = ep_obj.show.lang or sickrage.app.config.general.series_provider_default_language
series_info = ep_obj.show.series_provider.get_series_info(ep_obj.show.series_id, language=series_provider_language)
if not series_info:
return False
rootNode = Element("details")
movie = SubElement(rootNode, "movie")
movie.attrib["isExtra"] = "false"
movie.attrib["isSet"] = "false"
movie.attrib["isTV"] = "true"
# write an MediaBrowser XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
series_episode_info = series_info[curEpToWrite.season][curEpToWrite.episode]
except (SeriesProviderEpisodeNotFound, SeriesProviderSeasonNotFound):
sickrage.app.log.info(
f"Unable to find episode {curEpToWrite.season:d}x{curEpToWrite.episode:d} on {ep_obj.show.series_provider.name}"
f"... has it been removed? Should I delete from db?")
return None
if curEpToWrite == ep_obj:
# root (or single) episode
# default to today's date for specials if firstaired is not set
if curEpToWrite.season == 0 and not getattr(series_episode_info, 'firstAired', None):
series_episode_info['firstAired'] = str(datetime.date.min)
if not (getattr(series_episode_info, 'name', None) and getattr(series_episode_info, 'firstAired', None)):
return None
episode = movie
if curEpToWrite.name:
EpisodeName = SubElement(episode, "title")
EpisodeName.text = curEpToWrite.name
SeasonNumber = SubElement(episode, "season")
SeasonNumber.text = str(curEpToWrite.season)
EpisodeNumber = SubElement(episode, "episode")
EpisodeNumber.text = str(curEpToWrite.episode)
if getattr(series_info, "firstAired", None):
try:
year_text = str(datetime.datetime.strptime(series_info["firstAired"], dateFormat).year)
if year_text:
year = SubElement(episode, "year")
year.text = year_text
except:
pass
if getattr(series_info, "overview", None):
plot = SubElement(episode, "plot")
plot.text = series_info["overview"]
if curEpToWrite.description:
Overview = SubElement(episode, "episodeplot")
Overview.text = curEpToWrite.description
if getattr(series_info, 'contentrating', None):
mpaa = SubElement(episode, "mpaa")
mpaa.text = series_info["contentrating"]
if not ep_obj.related_episodes and getattr(series_episode_info, "rating", None):
try:
rating = int((float(series_episode_info['rating']) * 10))
except ValueError:
rating = 0
if rating:
Rating = SubElement(episode, "rating")
Rating.text = str(rating)
cast = SubElement(episode, "cast")
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(cast, "actor")
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
if person['imageUrl'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = person['imageUrl'].strip()
elif person['role'].strip() == 'Writer':
writer = SubElement(episode, "credits")
writer.text = series_episode_info['writer'].strip()
elif person['role'].strip() == 'Director':
director = SubElement(episode, "director")
director.text = series_episode_info['director'].strip()
elif person['role'].strip() == 'Guest Star':
cur_actor = SubElement(cast, "actor")
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
else:
episode = movie
# append data from (if any) related episodes
if curEpToWrite.name:
EpisodeName = SubElement(episode, "title")
if not EpisodeName.text:
EpisodeName.text = curEpToWrite.name
else:
EpisodeName.text = EpisodeName.text + ", " + curEpToWrite.name
if curEpToWrite.description:
Overview = SubElement(episode, "episodeplot")
if not Overview.text:
Overview.text = curEpToWrite.description
else:
Overview.text = Overview.text + "\r" + curEpToWrite.description
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for a MediaBrowser-style series.xml
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
root_node = Element("details")
tv_node = SubElement(root_node, "movie")
tv_node.attrib["isExtra"] = "false"
tv_node.attrib["isSet"] = "false"
tv_node.attrib["isTV"] = "true"
series_provider_language = show_obj.lang or sickrage.app.config.general.series_provider_default_language
series_info = show_obj.series_provider.get_series_info(show_obj.series_id, language=series_provider_language)
if not series_info:
return False
# check for title and id
if not (getattr(series_info, 'name', None) and getattr(series_info, 'id', None)):
sickrage.app.log.info("Incomplete info for "
"show with id " + str(show_obj.series_id) + " on " + show_obj.series_provider.name + ", skipping it")
return False
SeriesName = SubElement(tv_node, "title")
SeriesName.text = series_info['name']
if getattr(series_info, "genre", None):
Genres = SubElement(tv_node, "genres")
for genre in series_info['genre']:
cur_genre = SubElement(Genres, "Genre")
cur_genre.text = genre['name'].strip()
if getattr(series_info, 'firstAired', None):
FirstAired = SubElement(tv_node, "premiered")
FirstAired.text = series_info['firstAired']
if getattr(series_info, "firstAired", None):
try:
year_text = str(datetime.datetime.strptime(series_info["firstAired"], dateFormat).year)
if year_text:
year = SubElement(tv_node, "year")
year.text = year_text
except Exception:
pass
if getattr(series_info, 'overview', None):
plot = SubElement(tv_node, "plot")
plot.text = series_info["overview"]
if getattr(series_info, 'rating', None):
try:
rating = int(float(series_info['rating']) * 10)
except ValueError:
rating = 0
if rating:
Rating = SubElement(tv_node, "rating")
Rating.text = str(rating)
if getattr(series_info, 'status', None):
Status = SubElement(tv_node, "status")
Status.text = series_info['status']
# if getattr(series_info, "contentrating", None):
# mpaa = SubElement(tv_node, "mpaa")
# mpaa.text = series_info["contentrating"]
if getattr(series_info, 'imdbId', None):
imdb_id = SubElement(tv_node, "id")
imdb_id.attrib["moviedb"] = "imdb"
imdb_id.text = series_info['imdbId']
if getattr(series_info, 'id', None):
series_id = SubElement(tv_node, "series_id")
series_id.text = str(series_info['id'])
if getattr(series_info, 'runtime', None):
Runtime = SubElement(tv_node, "runtime")
Runtime.text = str(series_info['runtime'])
cast = SubElement(tv_node, "cast")
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(cast, "actor")
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
if person['imageUrl'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = person['imageUrl'].strip()
indent_xml(root_node)
data = ElementTree(root_node)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a MediaBrowser style episode.xml
and returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
series_provider_language = ep_obj.show.lang or sickrage.app.config.general.series_provider_default_language
myShow = ep_obj.show.series_provider.search(
ep_obj.show.series_id, language=series_provider_language)
if not myShow:
return False
rootNode = Element("details")
movie = SubElement(rootNode, "movie")
movie.attrib["isExtra"] = "false"
movie.attrib["isSet"] = "false"
movie.attrib["isTV"] = "true"
# write an MediaBrowser XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
except (SeriesProviderEpisodeNotFound,
SeriesProviderSeasonNotFound):
sickrage.app.log.info(
f"Unable to find episode {curEpToWrite.season:d}x{curEpToWrite.episode:d} on {ep_obj.show.series_provider.name}"
f"... has it been removed? Should I delete from db?")
return None
if curEpToWrite == ep_obj:
# root (or single) episode
# default to today's date for specials if firstaired is not set
if curEpToWrite.season == 0 and not getattr(
myEp, 'firstaired', None):
myEp['firstaired'] = str(datetime.date.min)
if not (getattr(myEp, 'episodename', None)
and getattr(myEp, 'firstaired', None)):
return None
episode = movie
if curEpToWrite.name:
EpisodeName = SubElement(episode, "title")
EpisodeName.text = curEpToWrite.name
SeasonNumber = SubElement(episode, "season")
SeasonNumber.text = str(curEpToWrite.season)
EpisodeNumber = SubElement(episode, "episode")
EpisodeNumber.text = str(curEpToWrite.episode)
if getattr(myShow, "firstaired", None):
try:
year_text = str(
datetime.datetime.strptime(myShow["firstaired"],
dateFormat).year)
if year_text:
year = SubElement(episode, "year")
year.text = year_text
except:
pass
if getattr(myShow, "overview", None):
plot = SubElement(episode, "plot")
plot.text = myShow["overview"]
if curEpToWrite.description:
Overview = SubElement(episode, "episodeplot")
Overview.text = curEpToWrite.description
if getattr(myShow, 'contentrating', None):
mpaa = SubElement(episode, "mpaa")
mpaa.text = myShow["contentrating"]
if not ep_obj.related_episodes and getattr(
myEp, "rating", None):
try:
rating = int((float(myEp['rating']) * 10))
except ValueError:
rating = 0
if rating:
Rating = SubElement(episode, "rating")
Rating.text = str(rating)
if getattr(myEp, 'director', None):
director = SubElement(episode, "director")
director.text = myEp['director']
if getattr(myEp, 'writer', None):
writer = SubElement(episode, "credits")
writer.text = myEp['writer']
cast = SubElement(episode, "cast")
if getattr(myEp, 'gueststars', None) and isinstance(
myEp['gueststars'], str):
for actor in (x.strip()
for x in myEp['gueststars'].split('|')
if x.strip()):
cur_actor = SubElement(cast, "actor")
cur_actor.text = actor
for actor in ep_obj.show.series_provider.actors(
int(ep_obj.show.series_id)):
if 'name' in actor and actor['name'].strip():
cur_actor = SubElement(cast, "actor")
cur_actor.text = actor['name'].strip()
else:
# append data from (if any) related episodes
if curEpToWrite.name:
if not EpisodeName.text:
EpisodeName.text = curEpToWrite.name
else:
EpisodeName.text = EpisodeName.text + ", " + curEpToWrite.name
if curEpToWrite.description:
if not Overview.text:
Overview.text = curEpToWrite.description
else:
Overview.text = Overview.text + "\r" + curEpToWrite.description
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a WDTV style episode.xml
and returns the resulting data object.
ep_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
indexer_lang = ep_obj.show.lang or sickrage.app.config.indexer_default_language
try:
lINDEXER_API_PARMS = IndexerApi(
ep_obj.show.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if ep_obj.show.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(ep_obj.show.indexer).indexer(**lINDEXER_API_PARMS)
myShow = t[ep_obj.show.indexer_id]
except indexer_shownotfound as e:
raise ShowNotFoundException(str(e))
except indexer_error as e:
sickrage.app.log.error(
"Unable to connect to " +
IndexerApi(ep_obj.show.indexer).name +
" while creating meta files - skipping - {}".format(e))
return False
rootNode = Element("details")
# write an WDTV XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
except (indexer_episodenotfound, indexer_seasonnotfound):
sickrage.app.log.info(
"Unable to find episode %dx%d on %s... has it been removed? Should I delete from db?"
% (curEpToWrite.season, curEpToWrite.episode,
IndexerApi(ep_obj.show.indexer).name))
return None
if ep_obj.season == 0 and not getattr(myEp, 'firstaired', None):
myEp["firstaired"] = str(datetime.date.min)
if not (getattr(myEp, 'episodename', None)
and getattr(myEp, 'firstaired', None)):
return None
if len(eps_to_write) > 1:
episode = SubElement(rootNode, "details")
else:
episode = rootNode
# TODO: get right EpisodeID
episodeID = SubElement(episode, "id")
episodeID.text = str(curEpToWrite.indexer_id)
title = SubElement(episode, "title")
title.text = ep_obj.pretty_name()
if getattr(myShow, 'seriesname', None):
seriesName = SubElement(episode, "series_name")
seriesName.text = myShow["seriesname"]
if curEpToWrite.name:
episodeName = SubElement(episode, "episode_name")
episodeName.text = curEpToWrite.name
seasonNumber = SubElement(episode, "season_number")
seasonNumber.text = str(curEpToWrite.season)
episodeNum = SubElement(episode, "episode_number")
episodeNum.text = str(curEpToWrite.episode)
firstAired = SubElement(episode, "firstaired")
if curEpToWrite.airdate > datetime.date.min:
firstAired.text = str(curEpToWrite.airdate)
if getattr(myShow, 'firstaired', None):
try:
year_text = str(
datetime.datetime.strptime(myShow["firstaired"],
dateFormat).year)
if year_text:
year = SubElement(episode, "year")
year.text = year_text
except Exception:
pass
if curEpToWrite.season != 0 and getattr(myShow, 'runtime', None):
runtime = SubElement(episode, "runtime")
runtime.text = myShow["runtime"]
if getattr(myShow, 'genre', None):
genre = SubElement(episode, "genre")
genre.text = " / ".join([
x.strip() for x in myShow["genre"].split('|') if x.strip()
])
if getattr(myEp, 'director', None):
director = SubElement(episode, "director")
director.text = myEp['director']
for actor in t.actors(int(ep_obj.show.indexer_id)):
if not ('name' in actor and actor['name'].strip()):
continue
cur_actor = SubElement(episode, "actor")
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = actor['name']
if 'role' in actor and actor['role'].strip():
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = actor['role'].strip()
if curEpToWrite.description:
overview = SubElement(episode, "overview")
overview.text = curEpToWrite.description
# Make it purdy
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a WDTV style episode.xml
and returns the resulting data object.
ep_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
series_provider_language = ep_obj.show.lang or sickrage.app.config.general.series_provider_default_language
series_info = ep_obj.show.series_provider.get_series_info(
ep_obj.show.series_id, language=series_provider_language)
if not series_info:
return False
rootNode = Element("details")
# write an WDTV XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
series_episode_info = series_info[curEpToWrite.season][
curEpToWrite.episode]
except (SeriesProviderEpisodeNotFound,
SeriesProviderSeasonNotFound):
sickrage.app.log.info(
"Unable to find episode %dx%d on %s... has it been removed? Should I delete from db?"
% (curEpToWrite.season, curEpToWrite.episode,
ep_obj.show.series_provider.name))
return None
if ep_obj.season == 0 and not getattr(series_episode_info,
'firstAired', None):
series_episode_info["firstAired"] = str(datetime.date.min)
if not (getattr(series_episode_info, 'name', None)
and getattr(series_episode_info, 'firstAired', None)):
return None
if len(eps_to_write) > 1:
episode = SubElement(rootNode, "details")
else:
episode = rootNode
# TODO: get right EpisodeID
episodeID = SubElement(episode, "id")
episodeID.text = str(curEpToWrite.series_id)
title = SubElement(episode, "title")
title.text = ep_obj.pretty_name()
if getattr(series_info, 'name', None):
seriesName = SubElement(episode, "series_name")
seriesName.text = series_info["name"]
if curEpToWrite.name:
episodeName = SubElement(episode, "episode_name")
episodeName.text = curEpToWrite.name
seasonNumber = SubElement(episode, "season_number")
seasonNumber.text = str(curEpToWrite.season)
episodeNum = SubElement(episode, "episode_number")
episodeNum.text = str(curEpToWrite.episode)
firstAired = SubElement(episode, "firstAired")
if curEpToWrite.airdate > datetime.date.min:
firstAired.text = str(curEpToWrite.airdate)
if getattr(series_info, 'firstAired', None):
try:
year_text = str(
datetime.datetime.strptime(series_info["firstAired"],
dateFormat).year)
if year_text:
year = SubElement(episode, "year")
year.text = year_text
except Exception:
pass
if curEpToWrite.season != 0 and getattr(series_info, 'runtime',
None):
runtime = SubElement(episode, "runtime")
runtime.text = series_info["runtime"]
if getattr(series_info, 'genre', None):
genre = SubElement(episode, "genre")
genre.text = " / ".join(
[x['name'].strip() for x in series_info["genre"]])
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(episode, "actor")
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = person['name'].strip()
elif person['role'].strip() == 'Director':
director = SubElement(episode, "director")
director.text = series_episode_info['director'].strip()
if curEpToWrite.description:
overview = SubElement(episode, "overview")
overview.text = curEpToWrite.description
# Make it purdy
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for an KODI-style tvshow.nfo and
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
show_ID = show_obj.indexer_id
indexer_lang = show_obj.lang or sickrage.app.config.indexer_default_language
lINDEXER_API_PARMS = IndexerApi(show_obj.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if show_obj.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
t = IndexerApi(show_obj.indexer).indexer(**lINDEXER_API_PARMS)
tv_node = Element("tvshow")
try:
myShow = t[int(show_ID)]
except indexer_shownotfound:
sickrage.app.log.error("Unable to find show with id " +
str(show_ID) + " on " +
IndexerApi(show_obj.indexer).name +
", skipping it")
raise
except indexer_error:
sickrage.app.log.error(
"" + IndexerApi(show_obj.indexer).name +
" is down, can't use its data to add this show")
raise
# check for title and id
if not (getattr(myShow, 'seriesname', None)
and getattr(myShow, 'id', None)):
sickrage.app.log.info("Incomplete info for show with id " +
str(show_ID) + " on " +
IndexerApi(show_obj.indexer).name +
", skipping it")
return False
title = SubElement(tv_node, "title")
title.text = myShow["seriesname"]
if getattr(myShow, 'rating', None):
rating = SubElement(tv_node, "rating")
rating.text = myShow["rating"]
if getattr(myShow, 'firstaired', None):
try:
year_text = str(
datetime.datetime.strptime(myShow["firstaired"],
dateFormat).year)
if year_text:
year = SubElement(tv_node, "year")
year.text = year_text
except:
pass
if getattr(myShow, 'overview', None):
plot = SubElement(tv_node, "plot")
plot.text = myShow["overview"]
# if getattr(myShow, 'id', None):
# episodeguide = SubElement(tv_node, "episodeguide")
# episodeguideurl = SubElement(episodeguide, "url")
# episodeguideurl.text = IndexerApi(show_obj.indexer).config['base_url'] + str(
# myShow["id"]) + '/all/en.zip'
if getattr(myShow, 'contentrating', None):
mpaa = SubElement(tv_node, "mpaa")
mpaa.text = myShow["contentrating"]
if getattr(myShow, 'id', None):
indexer_id = SubElement(tv_node, "id")
indexer_id.text = str(myShow["id"])
if getattr(myShow, 'genre', None) and isinstance(myShow["genre"], str):
genre = SubElement(tv_node, "genre")
genre.text = " / ".join(x.strip()
for x in myShow["genre"].split('|')
if x.strip())
if getattr(myShow, 'firstaired', None):
premiered = SubElement(tv_node, "premiered")
premiered.text = myShow["firstaired"]
if getattr(myShow, 'network', None):
studio = SubElement(tv_node, "studio")
studio.text = myShow["network"].strip()
for actor in t.actors(int(show_ID)):
cur_actor = SubElement(tv_node, "actor")
if 'name' in actor and actor['name'].strip():
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = actor['name'].strip()
else:
continue
if 'role' in actor and actor['role'].strip():
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = actor['role'].strip()
if 'image' in actor and actor['image'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = actor['image'].strip()
# Make it purdy
indent_xml(tv_node)
data = ElementTree(tv_node)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for an KODI-style episode.nfo and
returns the resulting data object.
show_obj: a TVEpisode instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
indexer_lang = ep_obj.show.lang or sickrage.app.config.indexer_default_language
lINDEXER_API_PARMS = IndexerApi(ep_obj.show.indexer).api_params.copy()
lINDEXER_API_PARMS['language'] = indexer_lang
if ep_obj.show.dvdorder != 0:
lINDEXER_API_PARMS['dvdorder'] = True
try:
t = IndexerApi(ep_obj.show.indexer).indexer(**lINDEXER_API_PARMS)
myShow = t[ep_obj.show.indexer_id]
except indexer_shownotfound as e:
raise ShowNotFoundException(str(e))
except indexer_error as e:
sickrage.app.log.error(
"Unable to connect to {} while creating meta files - skipping - {}"
.format(IndexerApi(ep_obj.show.indexer).name, e))
return
if len(eps_to_write) > 1:
rootNode = Element("kodimultiepisode")
else:
rootNode = Element("episodedetails")
# write an NFO containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
myEp = myShow[curEpToWrite.season][curEpToWrite.episode]
except (indexer_episodenotfound, indexer_seasonnotfound):
sickrage.app.log.info(
"Unable to find episode %dx%d on %s... has it been removed? Should I delete from db?"
% (curEpToWrite.season, curEpToWrite.episode,
IndexerApi(ep_obj.show.indexer).name))
return None
if not getattr(myEp, 'firstaired', None):
myEp["firstaired"] = str(datetime.date.min)
if not getattr(myEp, 'episodename', None):
sickrage.app.log.debug(
"Not generating nfo because the ep has no title")
return None
sickrage.app.log.debug("Creating metadata for episode " +
str(ep_obj.season) + "x" +
str(ep_obj.episode))
if len(eps_to_write) > 1:
episode = SubElement(rootNode, "episodedetails")
else:
episode = rootNode
if getattr(myEp, 'episodename', None):
title = SubElement(episode, "title")
title.text = myEp['episodename']
if getattr(myShow, 'seriesname', None):
showtitle = SubElement(episode, "showtitle")
showtitle.text = myShow['seriesname']
season = SubElement(episode, "season")
season.text = str(curEpToWrite.season)
episodenum = SubElement(episode, "episode")
episodenum.text = str(curEpToWrite.episode)
uniqueid = SubElement(episode, "uniqueid")
uniqueid.text = str(curEpToWrite.indexer_id)
if curEpToWrite.airdate > datetime.date.min:
aired = SubElement(episode, "aired")
aired.text = str(curEpToWrite.airdate)
if getattr(myEp, 'overview', None):
plot = SubElement(episode, "plot")
plot.text = myEp['overview']
if curEpToWrite.season and getattr(myShow, 'runtime', None):
runtime = SubElement(episode, "runtime")
runtime.text = myShow["runtime"]
if getattr(myEp, 'airsbefore_season', None):
displayseason = SubElement(episode, "displayseason")
displayseason.text = myEp['airsbefore_season']
if getattr(myEp, 'airsbefore_episode', None):
displayepisode = SubElement(episode, "displayepisode")
displayepisode.text = myEp['airsbefore_episode']
if getattr(myEp, 'filename', None):
thumb = SubElement(episode, "thumb")
thumb.text = myEp['filename'].strip()
# watched = SubElement(episode, "watched")
# watched.text = 'false'
if getattr(myEp, 'writer', None):
ep_credits = SubElement(episode, "credits")
ep_credits.text = myEp['writer'].strip()
if getattr(myEp, 'director', None):
director = SubElement(episode, "director")
director.text = myEp['director'].strip()
if getattr(myEp, 'rating', None):
rating = SubElement(episode, "rating")
rating.text = myEp['rating']
if getattr(myEp, 'gueststars', None) and isinstance(
myEp['gueststars'], str):
for actor in (x.strip() for x in myEp['gueststars'].split('|')
if x.strip()):
cur_actor = SubElement(episode, "actor")
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = actor
for actor in t.actors(int(ep_obj.show.indexer_id)):
cur_actor = SubElement(episode, "actor")
if 'name' in actor and actor['name'].strip():
cur_actor_name = SubElement(cur_actor, "name")
cur_actor_name.text = actor['name'].strip()
else:
continue
if 'role' in actor and actor['role'].strip():
cur_actor_role = SubElement(cur_actor, "role")
cur_actor_role.text = actor['role'].strip()
if 'image' in actor and actor['image'].strip():
cur_actor_thumb = SubElement(cur_actor, "thumb")
cur_actor_thumb.text = actor['image'].strip()
# Make it purdy
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _ep_data(self, ep_obj):
"""
Creates an elementTree XML structure for a MediaBrowser style episode.xml
and returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
eps_to_write = [ep_obj] + ep_obj.related_episodes
persons_dict = {'Director': [], 'GuestStar': [], 'Writer': []}
series_provider_language = ep_obj.show.lang or sickrage.app.config.general.series_provider_default_language
series_info = ep_obj.show.series_provider.get_series_info(
ep_obj.show.series_id, language=series_provider_language)
if not series_info:
return False
rootNode = Element("Item")
# write an MediaBrowser XML containing info for all matching episodes
for curEpToWrite in eps_to_write:
try:
series_episode_info = series_info[curEpToWrite.season][
curEpToWrite.episode]
except (SeriesProviderEpisodeNotFound,
SeriesProviderSeasonNotFound):
sickrage.app.log.info(
f"Unable to find episode {curEpToWrite.season:d}x{curEpToWrite.episode:d} on {ep_obj.show.series_provider.name}... has it been removed? Should I delete from db?"
)
return None
if curEpToWrite == ep_obj:
# root (or single) episode
# default to today's date for specials if firstaired is not set
if ep_obj.season == 0 and not getattr(series_episode_info,
'firstAired', None):
series_episode_info['firstAired'] = str(datetime.date.min)
if not (getattr(series_episode_info, 'name', None)
and getattr(series_episode_info, 'firstAired', None)):
return None
episode = rootNode
if curEpToWrite.name:
EpisodeName = SubElement(episode, "EpisodeName")
EpisodeName.text = curEpToWrite.name
EpisodeNumber = SubElement(episode, "EpisodeNumber")
EpisodeNumber.text = str(ep_obj.episode)
if ep_obj.related_episodes:
EpisodeNumberEnd = SubElement(episode, "EpisodeNumberEnd")
EpisodeNumberEnd.text = str(curEpToWrite.episode)
SeasonNumber = SubElement(episode, "SeasonNumber")
SeasonNumber.text = str(curEpToWrite.season)
if not ep_obj.related_episodes and getattr(
series_episode_info, 'absolute_number', None):
absolute_number = SubElement(episode, "absolute_number")
absolute_number.text = str(
series_episode_info['absolute_number'])
if curEpToWrite.airdate > datetime.date.min:
FirstAired = SubElement(episode, "FirstAired")
FirstAired.text = str(curEpToWrite.airdate)
MetadataType = SubElement(episode, "Type")
MetadataType.text = "Episode"
if curEpToWrite.description:
Overview = SubElement(episode, "Overview")
Overview.text = curEpToWrite.description
if not ep_obj.related_episodes:
if getattr(series_episode_info, 'rating', None):
Rating = SubElement(episode, "Rating")
Rating.text = str(series_episode_info['rating'])
if getattr(series_info, 'imdb_id', None):
IMDB_ID = SubElement(episode, "IMDB_ID")
IMDB_ID.text = series_info['imdbId']
IMDB = SubElement(episode, "IMDB")
IMDB.text = series_info['imdbId']
IMDbId = SubElement(episode, "IMDbId")
IMDbId.text = series_info['imdbId']
series_id = SubElement(episode, "id")
series_id.text = str(curEpToWrite.series_id)
# fill in Persons section with collected directors, guest starts and writers
Persons = SubElement(episode, "Persons")
for person_type, names in persons_dict.items():
# remove doubles
names = list(set(names))
for cur_name in names:
Person = SubElement(Persons, "Person")
cur_person_name = SubElement(Person, "Name")
cur_person_name.text = cur_name
cur_person_type = SubElement(Person, "Type")
cur_person_type.text = person_type
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(Persons, "Person")
cur_actor_role = SubElement(cur_actor, "Role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "Name")
cur_actor_name.text = person['name'].strip()
cur_actor_type = SubElement(cur_actor, "Type")
cur_actor_type.text = "Actor"
Language = SubElement(episode, "Language")
try:
Language.text = series_episode_info['language']['overview']
except Exception:
Language.text = sickrage.app.config.general.series_provider_default_language
thumb = SubElement(episode, "filename")
# TODO: See what this is needed for.. if its still needed
# just write this to the NFO regardless of whether it actually exists or not
# note: renaming files after nfo generation will break this, tough luck
thumb_text = self.get_episode_thumb_path(ep_obj)
if thumb_text:
thumb.text = thumb_text
else:
episode = rootNode
# append data from (if any) related episodes
EpisodeNumberEnd = SubElement(episode, "EpisodeNumberEnd")
EpisodeNumberEnd.text = str(curEpToWrite.episode)
if curEpToWrite.name:
EpisodeName = SubElement(episode, "EpisodeName")
if not EpisodeName.text:
EpisodeName.text = curEpToWrite.name
else:
EpisodeName.text = ', '.join(
[EpisodeName.text, curEpToWrite.name])
if curEpToWrite.description:
Overview = SubElement(episode, "Overview")
if not Overview.text:
Overview.text = curEpToWrite.description
else:
Overview.text = '\r'.join(
[Overview.text, curEpToWrite.description])
# collect all directors, guest stars and writers
persons_dict['Director'] += [
x['name'] for x in series_info['people']
if x['role'].strip() == 'Director'
]
persons_dict['GuestStar'] += [
x['name'] for x in series_info['people']
if x['role'].strip() == 'Guest Star'
]
persons_dict['Writer'] += [
x['name'] for x in series_info['people']
if x['role'].strip() == 'Writer'
]
indent_xml(rootNode)
data = ElementTree(rootNode)
return data
def _show_data(self, show_obj):
"""
Creates an elementTree XML structure for a MediaBrowser-style series.xml
returns the resulting data object.
show_obj: a TVShow instance to create the NFO for
"""
tv_node = Element("Series")
series_provider_language = show_obj.lang or sickrage.app.config.general.series_provider_default_language
series_info = show_obj.series_provider.get_series_info(
show_obj.series_id, language=series_provider_language)
if not series_info:
return False
# check for title and id
if not (getattr(series_info, 'name', None)
and getattr(series_info, 'id', None)):
sickrage.app.log.info("Incomplete info for show with id " +
str(show_obj.series_id) + " on " +
show_obj.series_provider.name +
", skipping it")
return False
if getattr(series_info, 'id', None):
series_id = SubElement(tv_node, "id")
series_id.text = str(series_info['id'])
if getattr(series_info, 'name', None):
SeriesName = SubElement(tv_node, "SeriesName")
SeriesName.text = series_info['name']
if getattr(series_info, 'status', None):
Status = SubElement(tv_node, "Status")
Status.text = series_info['status']
if getattr(series_info, 'network', None):
Network = SubElement(tv_node, "Network")
Network.text = series_info['network']
if getattr(series_info, 'airTime', None):
Airs_Time = SubElement(tv_node, "Airs_Time")
Airs_Time.text = series_info['airTime']
if getattr(series_info, 'airDay', None):
Airs_DayOfWeek = SubElement(tv_node, "Airs_DayOfWeek")
Airs_DayOfWeek.text = series_info['airDay']
FirstAired = SubElement(tv_node, "FirstAired")
if getattr(series_info, 'firstAired', None):
FirstAired.text = series_info['firstAired']
if getattr(series_info, 'contentrating', None):
ContentRating = SubElement(tv_node, "ContentRating")
ContentRating.text = series_info['contentrating']
MPAARating = SubElement(tv_node, "MPAARating")
MPAARating.text = series_info['contentrating']
certification = SubElement(tv_node, "certification")
certification.text = series_info['contentrating']
MetadataType = SubElement(tv_node, "Type")
MetadataType.text = "Series"
if getattr(series_info, 'overview', None):
Overview = SubElement(tv_node, "Overview")
Overview.text = series_info['overview']
if getattr(series_info, 'firstAired', None):
PremiereDate = SubElement(tv_node, "PremiereDate")
PremiereDate.text = series_info['firstAired']
if getattr(series_info, 'rating', None):
Rating = SubElement(tv_node, "Rating")
Rating.text = str(series_info['rating'])
if getattr(series_info, 'firstAired', None):
try:
year_text = str(
datetime.datetime.strptime(series_info['firstAired'],
dateFormat).year)
if year_text:
ProductionYear = SubElement(tv_node, "ProductionYear")
ProductionYear.text = year_text
except Exception:
pass
if getattr(series_info, 'runtime', None):
RunningTime = SubElement(tv_node, "RunningTime")
RunningTime.text = series_info['runtime']
Runtime = SubElement(tv_node, "Runtime")
Runtime.text = str(series_info['runtime'])
if getattr(series_info, 'imdbid', None):
imdb_id = SubElement(tv_node, "IMDB_ID")
imdb_id.text = series_info['imdbId']
imdb_id = SubElement(tv_node, "IMDB")
imdb_id.text = series_info['imdbId']
imdb_id = SubElement(tv_node, "IMDbId")
imdb_id.text = series_info['imdbId']
if getattr(series_info, 'zap2itid', None):
Zap2ItId = SubElement(tv_node, "Zap2ItId")
Zap2ItId.text = series_info['zap2itid']
if getattr(series_info, 'genre', None) and isinstance(
series_info["genre"], str):
Genres = SubElement(tv_node, "Genres")
for genre in series_info['genre']:
cur_genre = SubElement(Genres, "Genre")
cur_genre.text = genre['name'].strip()
Genre = SubElement(tv_node, "Genre")
Genre.text = "|".join([
x.strip() for x in series_info["genre"].split('|')
if x.strip()
])
if getattr(series_info, 'network', None):
Studios = SubElement(tv_node, "Studios")
Studio = SubElement(Studios, "Studio")
Studio.text = series_info['network']
Persons = SubElement(tv_node, "Persons")
for person in series_info['people']:
if 'name' not in person or not person['name'].strip():
continue
if person['role'].strip() == 'Actor':
cur_actor = SubElement(Persons, "Person")
cur_actor_role = SubElement(cur_actor, "Role")
cur_actor_role.text = person['role'].strip()
cur_actor_name = SubElement(cur_actor, "Name")
cur_actor_name.text = person['name'].strip()
cur_actor_type = SubElement(cur_actor, "Type")
cur_actor_type.text = "Actor"
indent_xml(tv_node)
data = ElementTree(tv_node)
return data
