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