def change_list_status(self, control_id):
     tmdb.change_fav_status(
         media_id=self.info.get_property("id"),
         media_type=self.TYPE_ALT,
         status=str(not bool(self.states["favorite"])).lower(),
     )
     self.update_states()

 def context_menu(self, control_id):
     item_id = self.FocusedItem(control_id).getProperty("id")
     media_type = self.FocusedItem(control_id).getVideoInfoTag().getMediaType()
     listitems = [addon.LANG(32169)] if media_type == "tvshow" else [addon.LANG(32113)]
     if self.logged_in:
         listitems += [addon.LANG(14076)]
         if not self.type == "tv":
             listitems += [addon.LANG(32107)]
         if self.mode == "list":
             listitems += [addon.LANG(32035)]
     selection = xbmcgui.Dialog().contextmenu(list=listitems)
     if selection == 0:
         rating = utils.input_userrating()
         if tmdb.set_rating(media_type="tv" if media_type == "tvshow" else "movie",
                            media_id=item_id,
                            rating=rating,
                            dbid=self.FocusedItem(control_id).getVideoInfoTag().getDbId()):
             xbmc.sleep(2000)
             self.update(force_update=True)
             self.setCurrentListPosition(self.position)
     elif selection == 1:
         tmdb.change_fav_status(media_id=item_id,
                                media_type=self.type,
                                status="true")
     elif selection == 2:
         self.list_dialog(item_id)
     elif selection == 3:
         tmdb.change_list_status(list_id=self.list_id,
                                 movie_id=item_id,
                                 status=False)
         self.update(force_update=True)
         self.setCurrentListPosition(self.position)

def get_episodes(content):
    shows = ItemList(content_type="episodes")
    url = ""
    if content == "shows":
        url = 'calendars/shows/%s/14' % datetime.date.today()
    elif content == "premieres":
        url = 'calendars/shows/premieres/%s/14' % datetime.date.today()
    results = get_data(url=url,
                       params={"extended": "full"},
                       cache_days=0.3)
    count = 1
    if not results:
        return None
    for day in results.iteritems():
        for episode in day[1]:
            ep = episode["episode"]
            tv = episode["show"]
            title = ep["title"] if ep["title"] else ""
            label = u"{0} - {1}x{2}. {3}".format(tv["title"],
                                                 ep["season"],
                                                 ep["number"],
                                                 title)
            show = VideoItem(label=label,
                             path=PLUGIN_BASE + 'extendedtvinfo&&tvdb_id=%s' % tv["ids"]["tvdb"])
            show.set_infos({'title': title,
                            'aired': ep["first_aired"],
                            'season': ep["season"],
                            'episode': ep["number"],
                            'tvshowtitle': tv["title"],
                            'mediatype': "episode",
                            'year': tv.get("year"),
                            'duration': tv["runtime"] * 60 if tv["runtime"] else "",
                            'studio': tv["network"],
                            'plot': tv["overview"],
                            'country': tv["country"],
                            'status': tv["status"],
                            'trailer': tv["trailer"],
                            'imdbnumber': ep["ids"]["imdb"],
                            'rating': tv["rating"],
                            'genre': " / ".join(tv["genres"]),
                            'mpaa': tv["certification"]})
            show.set_properties({'tvdb_id': ep["ids"]["tvdb"],
                                 'id': ep["ids"]["tvdb"],
                                 'imdb_id': ep["ids"]["imdb"],
                                 'homepage': tv["homepage"]})
            if tv["ids"].get("tmdb"):
                art_info = tmdb.get_tvshow(tv["ids"]["tmdb"], light=True)
                show.set_artwork(tmdb.get_image_urls(poster=art_info.get("poster_path"),
                                                     fanart=art_info.get("backdrop_path")))
            shows.append(show)
            count += 1
            if count > 20:
                break
    return shows

 def __init__(self, *args, **kwargs):
     super(DialogEpisodeInfo, self).__init__(*args, **kwargs)
     self.tvshow_id = kwargs.get('tvshow_id')
     tv_info = tmdb.get_tvshow(self.tvshow_id, light=True)
     data = tmdb.extended_episode_info(tvshow_id=self.tvshow_id,
                                       season=kwargs.get('season'),
                                       episode=kwargs.get('episode'))
     if not data:
         return None
     self.info, self.lists, self.states = data
     self.info.set_info("tvshowtitle", tv_info["name"])
     image_info = imagetools.blur(self.info.get_art("thumb"))
     self.info.update_properties(image_info)

 def add_to_list_button(self, control_id):
     busy.show_busy()
     account_lists = tmdb.get_account_lists()
     listitems = ["%s (%i)" % (i["name"], i["item_count"]) for i in account_lists]
     listitems.insert(0, addon.LANG(32139))
     listitems.append(addon.LANG(32138))
     busy.hide_busy()
     index = xbmcgui.Dialog().select(heading=addon.LANG(32136),
                                     list=listitems)
     if index == 0:
         listname = xbmcgui.Dialog().input(heading=addon.LANG(32137),
                                           type=xbmcgui.INPUT_ALPHANUM)
         if not listname:
             return None
         list_id = tmdb.create_list(listname)
         xbmc.sleep(1000)
         tmdb.change_list_status(list_id=list_id,
                                 movie_id=self.info.get_property("id"),
                                 status=True)
     elif index == len(listitems) - 1:
         if tmdb.remove_list_dialog(tmdb.handle_lists(account_lists)):
             self.update_states()
     elif index > 0:
         tmdb.change_list_status(account_lists[index - 1]["id"], self.info.get_property("id"), True)
         self.update_states()

 def update_states(self):
     xbmc.sleep(2000)  # delay because MovieDB takes some time to update
     _, __, self.states = tmdb.extended_episode_info(tvshow_id=self.tvshow_id,
                                                     season=self.info.get_info("season"),
                                                     episode=self.info.get_info("episode"),
                                                     cache_time=0)
     super(DialogEpisodeInfo, self).update_states()

 def set_rating_dialog(self, control_id):
     rating = utils.input_userrating()
     if tmdb.set_rating(media_type=self.TYPE_ALT,
                        media_id=self.get_identifier(),
                        rating=rating,
                        dbid=self.info.get("dbid")):
         self.update_states()

 def __init__(self, *args, **kwargs):
     super(DialogActorInfo, self).__init__(*args, **kwargs)
     data = tmdb.extended_actor_info(actor_id=kwargs.get('id'))
     if not data:
         return None
     self.info, self.lists = data
     self.info.update_properties(imagetools.blur(self.info.get_art("thumb")))

 def open_account_menu(self, control_id):
     if self.type == "tv":
         listitems = [addon.LANG(32145)]
         if self.logged_in:
             listitems.append(addon.LANG(32144))
     else:
         listitems = [addon.LANG(32135)]
         if self.logged_in:
             listitems.append(addon.LANG(32134))
     wm.show_busy()
     if self.logged_in:
         account_lists = tmdb.get_account_lists()
         listitems += ["%s (%i)" % (i["name"], i["item_count"]) for i in account_lists]
     wm.hide_busy()
     index = xbmcgui.Dialog().select(heading=addon.LANG(32136),
                                     list=listitems)
     if index == -1:
         pass
     elif index == 0:
         self.sort = "created_at"
         self.sort_label = addon.LANG(32157)
         self.filters = []
         self.reset("rating")
     elif index == 1:
         self.sort = "created_at"
         self.sort_label = addon.LANG(32157)
         self.filters = []
         self.reset("favorites")
     else:
         self.close()
         dialog = wm.open_video_list(filters=[],
                                     mode="list",
                                     list_id=account_lists[index - 2]["id"],
                                     filter_label=account_lists[index - 2]["name"])
         dialog.doModal()

 def update_states(self):
     xbmc.sleep(2000)  # delay because MovieDB takes some time to update
     info = tmdb.get_episode(tvshow_id=self.tvshow_id,
                             season=self.info.get_info("season"),
                             episode=self.info.get_info("episode"),
                             cache_days=0)
     self.states = info.get("account_states")
     super(DialogEpisodeInfo, self).update_states()

 def movie_context_menu(self, control_id):
     movie_id = self.FocusedItem(control_id).getProperty("id")
     dbid = self.FocusedItem(control_id).getVideoInfoTag().getDbId()
     options = [addon.LANG(32113)]
     if self.logged_in:
         options.append(addon.LANG(32083))
     index = xbmcgui.Dialog().contextmenu(list=options)
     if index == 0:
         rating = utils.input_userrating()
         if rating == -1:
             return None
         tmdb.set_rating(media_type="movie",
                         media_id=movie_id,
                         rating=rating,
                         dbid=dbid)
         xbmc.sleep(2000)
         tmdb.get_movie(movie_id=movie_id,
                        cache_days=0)
     elif index == 1:
         account_lists = tmdb.get_account_lists()
         if not account_lists:
             return False
         listitems = ["%s (%i)" % (i["name"], i["item_count"]) for i in account_lists]
         i = xbmcgui.Dialog().select(addon.LANG(32136), listitems)
         if i > -1:
             tmdb.change_list_status(list_id=account_lists[i]["id"],
                                     movie_id=movie_id,
                                     status=True)

 def set_rating_dialog(self, control_id):
     preselect = int(self.states["rated"]["value"]) if (self.states and self.states.get("rated")) else -1
     rating = utils.input_userrating(preselect=preselect)
     if rating == -1:
         return None
     if tmdb.set_rating(
         media_type=self.TYPE_ALT, media_id=self.get_identifier(), rating=rating, dbid=self.info.get_info("dbid")
     ):
         self.setProperty("rated", str(rating) if rating > 0 else "")
         self.update_states()

 def __init__(self, *args, **kwargs):
     super(DialogTVShowInfo, self).__init__(*args, **kwargs)
     data = tmdb.extended_tvshow_info(tvshow_id=kwargs.get('tmdb_id', False),
                                      dbid=self.dbid)
     if not data:
         return None
     self.info, self.lists, self.states = data
     if not self.info.get_info("dbid"):
         self.info.set_art("poster", utils.get_file(self.info.get_art("poster")))
     self.info.update_properties(imagetools.blur(self.info.get_art("poster")))

 def __init__(self, *args, **kwargs):
     super(DialogEpisodeInfo, self).__init__(*args, **kwargs)
     self.tvshow_id = kwargs.get('tvshow_id')
     data = tmdb.extended_episode_info(tvshow_id=self.tvshow_id,
                                       season=kwargs.get('season'),
                                       episode=kwargs.get('episode'))
     if not data:
         return None
     self.info, self.lists, self.states = data
     self.info.update_properties(imagetools.blur(self.info.get("thumb")))

 def set_actor_filter(self, control_id):
     result = xbmcgui.Dialog().input(heading=addon.LANG(16017),
                                     type=xbmcgui.INPUT_ALPHANUM)
     if not result or result == -1:
         return None
     response = tmdb.get_person_info(result)
     if not response:
         return None
     self.add_filter(key="with_people",
                     value=response["id"],
                     label=response["name"])

 def open_credit_dialog(self, credit_id):
     info = tmdb.get_credit_info(credit_id)
     listitems = []
     if "seasons" in info["media"]:
         listitems += tmdb.handle_seasons(info["media"]["seasons"])
     if "episodes" in info["media"]:
         listitems += tmdb.handle_episodes(info["media"]["episodes"])
     if not listitems:
         listitems += [{"label": addon.LANG(19055)}]
     listitem, index = selectdialog.open_selectdialog(header=addon.LANG(32151),
                                                      listitems=listitems)
     if listitem["mediatype"] == "episode":
         wm.open_episode_info(prev_window=self,
                              season=listitems[index]["season"],
                              episode=listitems[index]["episode"],
                              tvshow_id=info["media"]["id"])
     elif listitem["mediatype"] == "season":
         wm.open_season_info(prev_window=self,
                             season=listitems[index]["season"],
                             tvshow_id=info["media"]["id"])

def handle_tvshows(results):
    shows = ItemList(content_type="tvshows")
    for i in results:
        item = i["show"] if "show" in i else i
        airs = item.get("airs", {})
        show = VideoItem(label=item["title"],
                         path='%sextendedtvinfo&&tvdb_id=%s' % (PLUGIN_BASE, item['ids']["tvdb"]))
        show.set_infos({'mediatype': "tvshow",
                        'title': item["title"],
                        'duration': item["runtime"] * 60 if item["runtime"] else "",
                        'year': item["year"],
                        'premiered': item["first_aired"][:10],
                        'country': item["country"],
                        'rating': round(item["rating"], 1),
                        'votes': item["votes"],
                        'imdbnumber': item['ids']["imdb"],
                        'mpaa': item["certification"],
                        'trailer': item["trailer"],
                        'status': item.get("status"),
                        'studio': item["network"],
                        'genre': " / ".join(item["genres"]),
                        'plot': item["overview"]})
        show.set_properties({'id': item['ids']["tmdb"],
                             'tvdb_id': item['ids']["tvdb"],
                             'imdb_id': item['ids']["imdb"],
                             'trakt_id': item['ids']["trakt"],
                             'language': item["language"],
                             'aired_episodes': item["aired_episodes"],
                             'homepage': item["homepage"],
                             'airday': airs.get("day"),
                             'airshorttime': airs.get("time"),
                             'watchers': item.get("watchers")})
        art_info = tmdb.get_tvshow(item["ids"]["tmdb"], light=True)
        show.set_artwork(tmdb.get_image_urls(poster=art_info.get("poster_path"),
                                             fanart=art_info.get("backdrop_path")))
        shows.append(show)
    shows = local_db.merge_with_local(media_type="tvshow",
                                      items=shows,
                                      library_first=False)
    shows.set_sorts(["mpaa", "duration"])
    return shows

 def __init__(self, *args, **kwargs):
     super(DialogSeasonInfo, self).__init__(*args, **kwargs)
     self.tvshow_id = kwargs.get('id')
     data = tmdb.extended_season_info(tvshow_id=self.tvshow_id,
                                      season_number=kwargs.get('season'))
     if not data:
         return None
     self.info, self.lists = data
     if not self.info.get_info("dbid"):  # need to add comparing for seasons
         poster = utils.get_file(url=self.info.get_art("poster"))
         self.info.set_art("poster", poster)
     self.info.update_properties(imagetools.blur(self.info.get_art("poster")))

 def tvshow_context_menu(self, control_id):
     tvshow_id = self.FocusedItem(control_id).getProperty("id")
     dbid = self.FocusedItem(control_id).getVideoInfoTag().getDbId()
     credit_id = self.FocusedItem(control_id).getProperty("credit_id")
     options = [addon.LANG(32169)]
     if credit_id:
         options.append(addon.LANG(32147))
     index = xbmcgui.Dialog().contextmenu(list=options)
     if index == 0:
         rating = utils.input_userrating()
         if rating == -1:
             return None
         tmdb.set_rating(media_type="tvshow",
                         media_id=tvshow_id,
                         rating=rating,
                         dbid=dbid)
         xbmc.sleep(2000)
         tmdb.get_tvshow(tvshow_id=tvshow_id,
                         cache_days=0)
     if index == 1:
         self.open_credit_dialog(credit_id=credit_id)

 def set_genre_filter(self, control_id):
     params = {"language": addon.setting("LanguageID")}
     response = tmdb.get_data(url="genre/%s/list" % (self.type),
                              params=params,
                              cache_days=10)
     ids = [item["id"] for item in response["genres"]]
     labels = [item["name"] for item in response["genres"]]
     index = xbmcgui.Dialog().select(heading=addon.LANG(32151),
                                     list=labels)
     if index == -1:
         return None
     self.add_filter(key="with_genres",
                     value=str(ids[index]),
                     typelabel=addon.LANG(135),
                     label=labels[index])