# -*- coding: utf-8 -*- from __future__ import absolute_import from builtins import map from builtins import str import json from .ui import database from resources.lib.ui.globals import g from .debrid import all_debrid, real_debrid, premiumize from . import pages from .ui.BrowserBase import BrowserBase from .indexers import simkl, trakt import ast import requests import datetime class KaitoBrowser(BrowserBase): def _parse_history_view(self, res): name = res return g.allocate_item(name, "search/" + name + "/1", True) def _parse_airing_dub_view(self, res): name = list(res.values())[0] mal_id = (list(res.keys())[0]).rsplit('/')[-2] url = 'watchlist_to_ep/{}//0'.format(mal_id) try: image = ast.literal_eval(database.get_show_mal(mal_id)['kodi_meta'])['poster'] except: image = 'DefaultVideo.png' info = {} info['title'] = name info['plot'] = '** = Dub production suspended until further notice.\n++ = Dub is being produced from home studios with an irregular release schedule.' info['mediatype'] = 'tvshow' return g.allocate_item(name, url, True, image, info) def _json_request(self, url, data=''): response = json.loads(self._get_request(url, data)) return response # TODO: Not sure i want this here.. def search_history(self,search_array): result = list(map(self._parse_history_view,search_array)) result.insert(0,g.allocate_item("New Search", "search", True)) result.insert(len(result),g.allocate_item("Clear Search History...", "clear_history", False)) return result def get_airing_dub(self): resp = requests.get('https://armkai.vercel.app/api/airingdub') if not resp.ok: return [] all_results = list(map(self._parse_airing_dub_view, resp.json())) return all_results def get_latest(self, real_debrid_enabled, premiumize_enabled): if real_debrid_enabled or premiumize_enabled: page = pages.nyaa.sources else: page = pages.gogoanime.sources latest = database.get(page().get_latest, 0.125) return latest def get_latest_dub(self, real_debrid_enabled, premiumize_enabled): if real_debrid_enabled or premiumize_enabled: page = pages.nyaa.sources else: page = pages.gogoanime.sources latest_dub = database.get(page().get_latest_dub, 0.125) return latest_dub def get_backup(self, anilist_id, source): show = database.get_show(anilist_id) mal_id = show['mal_id'] if not mal_id: mal_id = self.get_mal_id(anilist_id) database.add_mapping_id(anilist_id, 'mal_id', str(mal_id)) result = requests.get("https://kaito-b.firebaseio.com/%s/Pages/%s.json" % (mal_id, source)) return result.json() def get_mal_id(self, anilist_id): arm_resp = self._json_request("https://armkai.vercel.app/api/search?type=anilist&id={}".format(anilist_id)) mal_id = arm_resp["mal"] return mal_id def clean_show(self, show_id, meta_ids): database.add_meta_ids(show_id, meta_ids) database.remove_season(show_id) database.remove_episodes(show_id) name = ast.literal_eval(database.get_show(show_id)['kodi_meta']) name.pop('fanart', None) database.add_fanart(show_id, name) def search_trakt_shows(self, anilist_id): shows = trakt.TRAKTAPI().search_trakt_shows(anilist_id) return shows def get_trakt_episodes(self, show_id, season, page=1): return trakt.TRAKTAPI().get_trakt_episodes(show_id, season) def get_anime_trakt(self, anilist_id, db_correction=False, filter_lang=None): anime = trakt.TRAKTAPI().get_anime(anilist_id, db_correction) if anime and filter_lang: for i in anime[0]: i['url'] += filter_lang if not anime: anime = self.get_anime_simkl(anilist_id, filter_lang) return anime def get_anime_simkl(self, anilist_id, params): return simkl.SIMKLAPI().get_anime(anilist_id, params) def get_anime_init(self, anilist_id, filter_lang=None): show_meta = database.get_show(anilist_id) if not show_meta: from .AniListBrowser import AniListBrowser show_meta = AniListBrowser().get_anilist(anilist_id) if not show_meta['meta_ids']: name = ast.literal_eval(show_meta['kodi_meta'])['name'] trakt_id = trakt.TRAKTAPI().get_trakt_id(name) if not trakt_id: return self.get_anime_simkl(anilist_id, filter_lang) database.add_meta_ids(anilist_id, str(trakt_id)) return self.get_anime_trakt(anilist_id, filter_lang=filter_lang) def get_episodeList(self, show_id, pass_idx, filter_lang=None, rescrape=False): episodes = database.get_episode_list(int(show_id)) if episodes: items = trakt.TRAKTAPI()._process_trakt_episodes(show_id, '', episodes, '') else: items = simkl.SIMKLAPI().get_episodes(show_id) if rescrape: return items items = [i for i in items if self.is_aired(i['info'])] playlist = g.bulk_draw_items(items)[pass_idx:] for i in playlist: url = i[0] if filter_lang: url += filter_lang g.PLAYLIST.add(url=url, listitem=i[1]) def is_aired(self, info): try: try: air_date = info['aired'] except: air_date = info.get('premiered') if not air_date: return False if int(air_date[:4]) < 2019: return True todays_date = datetime.datetime.today().strftime('%Y-%m-%d') if air_date > todays_date: return False else: return True except: import traceback traceback.print_exc() # Assume an item is not aired if we do not have any information on it or fail to identify return False def get_sources(self, anilist_id, episode, filter_lang, media_type, rescrape=False): show = database.get_show(anilist_id) kodi_meta = ast.literal_eval(show['kodi_meta']) actionArgs = { 'query': kodi_meta['query'], 'anilist_id': anilist_id, 'episode': episode, 'status': kodi_meta['status'], 'filter_lang': filter_lang, 'media_type': media_type, 'rescrape': rescrape, 'get_backup': self.get_backup } sources = pages.getSourcesHelper(actionArgs) return sources def get_latest_sources(self, debrid_provider, hash_): resolvers = {'premiumize': premiumize.Premiumize, 'all_debrid': all_debrid.AllDebrid, 'real_debrid': real_debrid.RealDebrid} magnet = 'magnet:?xt=urn:btih:' + hash_ api = resolvers[debrid_provider] link = api().resolve_single_magnet(hash_, magnet) return link