def do_disable_check(): scrapers = relevant_scrapers() auto_disable = kodi.get_setting('auto-disable') check_freq = int(kodi.get_setting('disable-freq')) disable_thresh = int(kodi.get_setting('disable-thresh')) for cls in scrapers: last_check = db_connection.get_setting('%s_check' % (cls.get_name())) last_check = int(last_check) if last_check else 0 tries = kodi.get_setting('%s_try' % (cls.get_name())) tries = int(tries) if tries else 0 if tries > 0 and tries / check_freq > last_check / check_freq: kodi.set_setting('%s_check' % (cls.get_name()), str(tries)) success_rate = calculate_success(cls.get_name()) if success_rate < disable_thresh: if auto_disable == DISABLE_SETTINGS.ON: kodi.set_setting('%s-enable' % (cls.get_name()), 'false') kodi.notify(msg='[COLOR blue]%s[/COLOR] %s' % (i18n('scraper_disabled')), duration=5000) elif auto_disable == DISABLE_SETTINGS.PROMPT: dialog = xbmcgui.Dialog() line1 = i18n('disable_line1') % (cls.get_name(), 100 - success_rate, tries) line2 = i18n('disable_line2') line3 = i18n('disable_line3') ret = dialog.yesno('SALTS', line1, line2, line3, i18n('keep_enabled'), i18n('disable_it')) if ret: kodi.set_setting('%s-enable' % (cls.get_name()), 'false')
def do_disable_check(): scrapers = relevant_scrapers() auto_disable = ADDON.get_setting('auto-disable') check_freq = int(ADDON.get_setting('disable-freq')) disable_thresh = int(ADDON.get_setting('disable-thresh')) for cls in scrapers: last_check = db_connection.get_setting('%s_check' % (cls.get_name())) last_check = int(last_check) if last_check else 0 tries = ADDON.get_setting('%s_try' % (cls.get_name())) tries = int(tries) if tries else 0 if tries > 0 and tries / check_freq > last_check / check_freq: ADDON.set_setting('%s_check' % (cls.get_name()), str(tries)) success_rate = calculate_success(cls.get_name()) if success_rate < disable_thresh: if auto_disable == DISABLE_SETTINGS.ON: ADDON.set_setting('%s-enable' % (cls.get_name()), 'false') notify(msg='[COLOR blue]%s[/COLOR] %s' % (i18n('scraper_disabled')), duration=5000) elif auto_disable == DISABLE_SETTINGS.PROMPT: dialog = xbmcgui.Dialog() line1 = i18n('disable_line1') % (cls.get_name(), 100 - success_rate, tries) line2 = i18n('disable_line2') line3 = i18n('disable_line3') ret = dialog.yesno('SALTS', line1, line2, line3, i18n('keep_enabled'), i18n('disable_it')) if ret: ADDON.set_setting('%s-enable' % (cls.get_name()), 'false')
def get_resume_choice(trakt_id, season, episode): if kodi.get_setting("trakt_bookmark") == "true": resume_point = "%s%%" % (trakt_api.get_bookmark(trakt_id, season, episode)) header = i18n("trakt_bookmark_exists") else: resume_point = format_time(db_connection.get_bookmark(trakt_id, season, episode)) header = i18n("local_bookmark_exists") question = i18n("resume_from") % (resume_point) return xbmcgui.Dialog().yesno(header, question, "", "", i18n("start_from_beginning"), i18n("resume")) == 1
def get_resume_choice(trakt_id, season, episode): if kodi.get_setting('trakt_bookmark') == 'true': resume_point = '%s%%' % (trakt_api.get_bookmark(trakt_id, season, episode)) header = i18n('trakt_bookmark_exists') else: resume_point = format_time(db_connection.get_bookmark(trakt_id, season, episode)) header = i18n('local_bookmark_exists') question = i18n('resume_from') % (resume_point) return xbmcgui.Dialog().yesno(header, question, '', '', i18n('start_from_beginning'), i18n('resume')) == 1
def choose_list(username=None): lists = trakt_api.get_lists(username) if username is None: lists.insert(0, {'name': 'watchlist', 'ids': {'slug': WATCHLIST_SLUG}}) if lists: dialog = xbmcgui.Dialog() index = dialog.select(i18n('pick_a_list'), [list_data['name'] for list_data in lists]) if index > -1: return lists[index]['ids']['slug'] else: kodi.notify(msg=i18n('no_lists_for_user') % (username), duration=5000)
def choose_list(username=None): lists = trakt_api.get_lists(username) if username is None: lists.insert(0, {"name": "watchlist", "ids": {"slug": WATCHLIST_SLUG}}) if lists: dialog = xbmcgui.Dialog() index = dialog.select(i18n("pick_a_list"), [list_data["name"] for list_data in lists]) if index > -1: return lists[index]["ids"]["slug"] else: kodi.notify(msg=i18n("no_lists_for_user") % (username), duration=5000)
def onClick(self, control): # print 'onClick: %s' % (control) if control == AUTH_BUTTON: if not self.__get_token(): kodi.notify(msg=i18n('pin_auth_failed'), duration=5000) return self.auth = True if control == LATER_BUTTON: kodi.notify(msg=i18n('remind_in_24hrs'), duration=5000) kodi.set_setting('last_reminder', str(int(time.time()))) if control == NEVER_BUTTON: kodi.notify(msg=i18n('use_addon_settings'), duration=5000) kodi.set_setting('last_reminder', '-1') if control in [AUTH_BUTTON, LATER_BUTTON, NEVER_BUTTON]: self.close()
def parallel_get_url(q, scraper, video): worker = threading.current_thread() log_utils.log('Worker: %s (%s) for %s url' % (worker.name, worker, scraper.get_name()), log_utils.LOGDEBUG) url = scraper.get_url(video) log_utils.log('%s returned url %s from %s' % (scraper.get_name(), url, worker), log_utils.LOGDEBUG) if not url: url = '' if url == FORCE_NO_MATCH: label = '[%s] [COLOR green]%s[/COLOR]' % (scraper.get_name(), i18n('force_no_match')) else: label = '[%s] %s' % (scraper.get_name(), url) related = {'class': scraper, 'url': url, 'name': scraper.get_name(), 'label': label} q.put(related)
def get_section_params(section): section_params = {} section_params['section'] = section if section == SECTIONS.TV: section_params['next_mode'] = MODES.SEASONS section_params['folder'] = True section_params['video_type'] = VIDEO_TYPES.TVSHOW section_params['content_type'] = CONTENT_TYPES.TVSHOWS section_params['search_img'] = 'television_search.png' section_params['label_plural'] = i18n('tv_shows') section_params['label_single'] = i18n('tv_show') else: section_params['next_mode'] = MODES.GET_SOURCES section_params['folder'] = kodi.get_setting('source-win') == 'Directory' and kodi.get_setting('auto-play') == 'false' section_params['video_type'] = VIDEO_TYPES.MOVIE section_params['content_type'] = CONTENT_TYPES.MOVIES section_params['search_img'] = 'movies_search.png' section_params['label_plural'] = i18n('movies') section_params['label_single'] = i18n('movie') return section_params
def perform_auto_conf(responses): length = len(responses) TOTAL = 12 if length < TOTAL: responses += [True] * (TOTAL - length) if responses[0]: kodi.set_setting('trakt_timeout', '60') if responses[1]: kodi.set_setting('calendar-day', '-1') if responses[2]: kodi.set_setting('calendar_time', '2') if responses[3]: kodi.set_setting('source_timeout', '20') if responses[4]: kodi.set_setting('include_watchlist_next', 'true') if responses[5]: kodi.set_setting('filter_direct', 'true') if responses[6]: kodi.set_setting('filter_unusable', 'true') if responses[7]: kodi.set_setting('show_debrid', 'true') if responses[8]: kodi.set_setting('source_results', '0') if responses[9]: kodi.set_setting('enable_sort', 'true') kodi.set_setting('sort1_field', '2') kodi.set_setting('sort2_field', '5') kodi.set_setting('sort3_field', '6') kodi.set_setting('sort4_field', '1') kodi.set_setting('sort5_field', '3') kodi.set_setting('sort6_field', '4') if responses[10]: tiers = ['Local', 'Furk.net', 'Premiumize.me', 'EasyNews', 'DD.tv', 'NoobRoom', ['WatchHD', 'IFlix', 'MoviesPlanet', 'CyberReel', '9Movies', '123Movies', 'niter.tv', 'ororo.tv'], ['movietv.to', 'StreamLord', 'tunemovie', 'afdah.org', 'xmovies8', 'xmovies8.v2', 'alluc.com'], ['torba.se', 'IzlemeyeDeger', 'Rainierland', 'zumvo.com', 'PutMV', 'MiraDeTodo', 'beinmovie'], ['SezonLukDizi', 'Dizimag', 'Dizilab', 'Dizigold', 'Diziay', 'Dizipas', 'OneClickTVShows'], ['DDLValley', 'ReleaseBB', 'MyVideoLinks.eu', 'OCW', 'RLSSource.net', 'TVRelease.Net'], ['IceFilms', 'PrimeWire', 'Flixanity', 'wso.ch', 'WatchSeries', 'UFlix.org', 'Putlocker', 'MovieHut'], ['funtastic-vids', 'WatchFree.to', 'pftv', 'streamallthis.is', 'Movie4K', 'afdah', 'SolarMovie', 'yify-streaming'], ['CouchTunerV2', 'CouchTunerV1', 'Watch8Now', 'yshows', 'TwoMovies.us', 'iWatchOnline', 'vidics.ch', 'pubfilm'], ['OnlineMoviesIs', 'OnlineMoviesPro', 'ViewMovies', 'movie25', 'viooz.ac', 'view47', 'MoviesHD', 'wmo.ch'], ['ayyex', 'stream-tv.co', 'clickplay.to', 'MintMovies', 'MovieNight', 'cmz', 'ch131', 'filmikz.ch'], ['MovieTube', 'LosMovies', 'FilmStreaming.in', 'moviestorm.eu', 'MerDB'], 'MoviesOnline7'] sso = [] random_sso = kodi.get_setting('random_sso') == 'true' for tier in tiers: if isinstance(tier, basestring): sso.append(tier) else: if random_sso: random.shuffle(tier) sso += tier kodi.set_setting('source_sort_order', '|'.join(sso)) if responses[11]: reset_base_url() kodi.set_setting('filter-unknown', 'false') kodi.notify(msg=i18n('auto_conf_complete'))
def do_disable_check(): auto_disable = kodi.get_setting('auto-disable') disable_limit = int(kodi.get_setting('disable-limit')) for cls in relevant_scrapers(): setting = '%s_last_results' % (cls.get_name()) fails = kodi.get_setting(setting) fails = int(fails) if fails else 0 if fails >= disable_limit: if auto_disable == DISABLE_SETTINGS.ON: kodi.set_setting('%s-enable' % (cls.get_name()), 'false') kodi.notify(msg='[COLOR blue]%s[/COLOR] %s' % (cls.get_name(), i18n('scraper_disabled')), duration=5000) kodi.set_setting(setting, '0') elif auto_disable == DISABLE_SETTINGS.PROMPT: dialog = xbmcgui.Dialog() line1 = i18n('disable_line1') % (cls.get_name(), fails) line2 = i18n('disable_line2') line3 = i18n('disable_line3') ret = dialog.yesno('SALTS', line1, line2, line3, i18n('keep_enabled'), i18n('disable_it')) if ret: kodi.set_setting('%s-enable' % (cls.get_name()), 'false') kodi.set_setting(setting, '0') else: kodi.set_setting(setting, '-1')
def parallel_get_url(q, scraper, video): worker = threading.current_thread() log_utils.log("Worker: %s (%s) for %s url" % (worker.name, worker, scraper.get_name()), log_utils.LOGDEBUG) url = scraper.get_url(video) log_utils.log("%s returned url %s from %s" % (scraper.get_name(), url, worker), log_utils.LOGDEBUG) if not url: url = "" if url == FORCE_NO_MATCH: label = "[%s] [COLOR green]%s[/COLOR]" % (scraper.get_name(), i18n("force_no_match")) else: label = "[%s] %s" % (scraper.get_name(), url) related = {"class": scraper, "url": url, "name": scraper.get_name(), "label": label} q.put(related)
def get_section_params(section): section_params = {} section_params["section"] = section if section == SECTIONS.TV: section_params["next_mode"] = MODES.SEASONS section_params["folder"] = True section_params["video_type"] = VIDEO_TYPES.TVSHOW section_params["content_type"] = CONTENT_TYPES.TVSHOWS section_params["search_img"] = "television_search.png" section_params["label_plural"] = i18n("tv_shows") section_params["label_single"] = i18n("tv_show") else: section_params["next_mode"] = MODES.GET_SOURCES section_params["folder"] = ( kodi.get_setting("source-win") == "Directory" and kodi.get_setting("auto-play") == "false" ) section_params["video_type"] = VIDEO_TYPES.MOVIE section_params["content_type"] = CONTENT_TYPES.MOVIES section_params["search_img"] = "movies_search.png" section_params["label_plural"] = i18n("movies") section_params["label_single"] = i18n("movie") return section_params
def do_disable_check(): auto_disable = kodi.get_setting("auto-disable") disable_limit = int(kodi.get_setting("disable-limit")) for cls in relevant_scrapers(): setting = "%s_last_results" % (cls.get_name()) fails = kodi.get_setting(setting) fails = int(fails) if fails else 0 if fails >= disable_limit: if auto_disable == DISABLE_SETTINGS.ON: kodi.set_setting("%s-enable" % (cls.get_name()), "false") kodi.notify(msg="[COLOR blue]%s[/COLOR] %s" % (cls.get_name(), i18n("scraper_disabled")), duration=5000) kodi.set_setting(setting, "0") elif auto_disable == DISABLE_SETTINGS.PROMPT: dialog = xbmcgui.Dialog() line1 = i18n("disable_line1") % (cls.get_name(), fails) line2 = i18n("disable_line2") line3 = i18n("disable_line3") ret = dialog.yesno("SALTS", line1, line2, line3, i18n("keep_enabled"), i18n("disable_it")) if ret: kodi.set_setting("%s-enable" % (cls.get_name()), "false") kodi.set_setting(setting, "0") else: kodi.set_setting(setting, "-1")
def __attempt_db_recovery(self): self.__recovery = True header = i18n('recovery_header') if xbmcgui.Dialog().yesno(header, i18n('rec_mig_1'), i18n('rec_mig_2')): try: self.init_database('0.0.0') except Exception as e: log_utils.log('DB Migration Failed: %s' % (e), log_utils.LOGWARNING) if xbmcgui.Dialog().yesno(header, i18n('rec_reset_1'), i18n('rec_reset_2'), i18n('rec_reset_3')): try: result = self.reset_db() except Exception as e: log_utils.log('Reset Failed: %s' % (e), log_utils.LOGWARNING) kodi.notify(msg=i18n('reset_failed') % e, duration=5000) else: if result: kodi.notify(msg=i18n('db_reset_success'))
def attempt_db_recovery(self): header = i18n('recovery_header') if xbmcgui.Dialog().yesno(header, i18n('rec_mig_1'), i18n('rec_mig_2')): try: self.init_database('0.0.0') except Exception as e: log_utils.log('DB Migration Failed: %s' % (e), log_utils.LOGWARNING) if self.db_type == DB_TYPES.SQLITE: if xbmcgui.Dialog().yesno(header, i18n('rec_reset_1'), i18n('rec_reset_2'), i18n('rec_reset_3')): try: self.reset_db() except Exception as e: log_utils.log('Reset Failed: %s' % (e), log_utils.LOGWARNING) try: msg = i18n('reset_failed') % (e) except: msg = 'Reset Failed: %s' % (e) else: msg = i18n('db_reset_success') kodi.notify(msg=msg, duration=5000)
def perform_auto_conf(responses): length = len(responses) TOTAL = 12 if length < TOTAL: responses += [True] * (TOTAL - length) if responses[0]: kodi.set_setting('trakt_timeout', '60') if responses[1]: kodi.set_setting('calendar-day', '-1') if responses[2]: kodi.set_setting('calendar_time', '2') if responses[3]: kodi.set_setting('source_timeout', '20') if responses[4]: kodi.set_setting('include_watchlist_next', 'true') if responses[5]: kodi.set_setting('filter_direct', 'true') if responses[6]: kodi.set_setting('filter_unusable', 'true') if responses[7]: kodi.set_setting('show_debrid', 'true') if responses[8]: kodi.set_setting('source_results', '0') if responses[9]: kodi.set_setting('enable_sort', 'true') kodi.set_setting('sort1_field', '2') kodi.set_setting('sort2_field', '5') kodi.set_setting('sort3_field', '6') kodi.set_setting('sort4_field', '1') kodi.set_setting('sort5_field', '3') kodi.set_setting('sort6_field', '4') if responses[10]: tiers = [ 'Local', 'Furk.net', 'Premiumize.me', 'EasyNews', 'DD.tv', 'NoobRoom', [ 'WatchHD', 'IFlix', 'MoviesPlanet', 'CyberReel', '9Movies', '123Movies', 'niter.tv', 'ororo.tv' ], [ 'movietv.to', 'StreamLord', 'tunemovie', 'afdah.org', 'xmovies8', 'xmovies8.v2', 'alluc.com' ], [ 'torba.se', 'IzlemeyeDeger', 'Rainierland', 'zumvo.com', 'PutMV', 'MiraDeTodo', 'beinmovie' ], [ 'SezonLukDizi', 'Dizimag', 'Dizilab', 'Dizigold', 'Diziay', 'Dizipas', 'OneClickTVShows', 'MovieFarsi' ], [ 'DDLValley', 'ReleaseBB', 'MyVideoLinks.eu', 'OneClickWatch', 'RLSSource.net', 'TVRelease.Net' ], [ 'IceFilms', 'PrimeWire', 'Flixanity', 'wso.ch', 'WatchSeries', 'UFlix.org', 'Putlocker', 'MovieHut' ], [ 'funtastic-vids', 'WatchFree.to', 'pftv', 'streamallthis.is', 'Movie4K', 'afdah', 'SolarMovie', 'yify-streaming' ], [ 'CouchTunerV2', 'CouchTunerV1', 'Watch8Now', 'yshows', 'TwoMovies.us', 'iWatchOnline', 'vidics.ch', 'pubfilm' ], [ 'OnlineMoviesIs', 'OnlineMoviesPro', 'ViewMovies', 'movie25', 'viooz.ac', 'view47', 'MoviesHD', 'wmo.ch' ], [ 'ayyex', 'stream-tv.co', 'clickplay.to', 'MintMovies', 'MovieNight', 'cmz', 'ch131', 'filmikz.ch' ], [ 'MovieTube', 'LosMovies', 'FilmStreaming.in', 'moviestorm.eu', 'MerDB' ], 'MoviesOnline7' ] sso = [] random_sso = kodi.get_setting('random_sso') == 'true' for tier in tiers: if isinstance(tier, basestring): sso.append(tier) else: if random_sso: random.shuffle(tier) sso += tier kodi.set_setting('source_sort_order', '|'.join(sso)) if responses[11]: reset_base_url() kodi.set_setting('filter-unknown', 'false') kodi.notify(msg=i18n('auto_conf_complete'))
def download_media(url, path, file_name): try: progress = int(ADDON.get_setting('down_progress')) import urllib2 request = urllib2.Request(url) request.add_header('User-Agent', USER_AGENT) request.add_unredirected_header('Host', request.get_host()) response = urllib2.urlopen(request) content_length = 0 if 'Content-Length' in response.info(): content_length = int(response.info()['Content-Length']) file_name = file_name.replace('.strm', get_extension(url, response)) full_path = os.path.join(path, file_name) log_utils.log('Downloading: %s -> %s' % (url, full_path), xbmc.LOGDEBUG) path = xbmc.makeLegalFilename(path) if not xbmcvfs.exists(path): try: try: xbmcvfs.mkdirs(path) except: os.mkdir(path) except Exception as e: raise Exception(i18n('failed_create_dir')) file_desc = xbmcvfs.File(full_path, 'w') total_len = 0 if progress: if progress == PROGRESS.WINDOW: dialog = xbmcgui.DialogProgress() else: dialog = xbmcgui.DialogProgressBG() dialog.create('Stream All The Sources', i18n('downloading') % (file_name)) dialog.update(0) while True: data = response.read(CHUNK_SIZE) if not data: break if progress == PROGRESS.WINDOW and dialog.iscanceled(): break total_len += len(data) if not file_desc.write(data): raise Exception('failed_write_file') percent_progress = (total_len) * 100 / content_length if content_length > 0 else 0 log_utils.log('Position : %s / %s = %s%%' % (total_len, content_length, percent_progress), xbmc.LOGDEBUG) if progress == PROGRESS.WINDOW: dialog.update(percent_progress) elif progress == PROGRESS.BACKGROUND: dialog.update(percent_progress, 'Stream All The Sources') else: notify(msg=i18n('download_complete') % (file_name), duration=5000) log_utils.log('Download Complete: %s -> %s' % (url, full_path), xbmc.LOGDEBUG) file_desc.close() if progress: dialog.close() except Exception as e: log_utils.log('Error (%s) during download: %s -> %s' % (str(e), url, file_name), xbmc.LOGERROR) notify(msg=i18n('download_error') % (str(e), file_name), duration=5000)
def download_media(url, path, file_name): try: progress = int(kodi.get_setting("down_progress")) request = urllib2.Request(url) request.add_header("User-Agent", USER_AGENT) request.add_unredirected_header("Host", request.get_host()) response = urllib2.urlopen(request) content_length = 0 if "Content-Length" in response.info(): content_length = int(response.info()["Content-Length"]) file_name = file_name.replace(".strm", get_extension(url, response)) full_path = os.path.join(path, file_name) log_utils.log("Downloading: %s -> %s" % (url, full_path), log_utils.LOGDEBUG) path = xbmc.makeLegalFilename(path) if not xbmcvfs.exists(path): try: try: xbmcvfs.mkdirs(path) except: os.mkdir(path) except Exception as e: raise Exception(i18n("failed_create_dir")) file_desc = xbmcvfs.File(full_path, "w") total_len = 0 if progress: if progress == PROGRESS.WINDOW: dialog = xbmcgui.DialogProgress() else: dialog = xbmcgui.DialogProgressBG() dialog.create("Stream All The Sources", i18n("downloading") % (file_name)) dialog.update(0) while True: data = response.read(CHUNK_SIZE) if not data: break if progress == PROGRESS.WINDOW and dialog.iscanceled(): break total_len += len(data) if not file_desc.write(data): raise Exception("failed_write_file") percent_progress = (total_len) * 100 / content_length if content_length > 0 else 0 log_utils.log( "Position : %s / %s = %s%%" % (total_len, content_length, percent_progress), log_utils.LOGDEBUG ) if progress == PROGRESS.WINDOW: dialog.update(percent_progress) elif progress == PROGRESS.BACKGROUND: dialog.update(percent_progress, "Stream All The Sources") else: kodi.notify(msg=i18n("download_complete") % (file_name), duration=5000) log_utils.log("Download Complete: %s -> %s" % (url, full_path), log_utils.LOGDEBUG) file_desc.close() if progress: dialog.close() except Exception as e: log_utils.log("Error (%s) during download: %s -> %s" % (str(e), url, file_name), log_utils.LOGERROR) kodi.notify(msg=i18n("download_error") % (str(e), file_name), duration=5000)
def get_pin(): AUTH_BUTTON = 200 LATER_BUTTON = 201 NEVER_BUTTON = 202 ACTION_PREVIOUS_MENU = 10 ACTION_BACK = 92 CENTER_Y = 6 CENTER_X = 2 class PinAuthDialog(xbmcgui.WindowXMLDialog): auth = False def onInit(self): self.pin_edit_control = self.__add_editcontrol(30, 240, 40, 450) self.setFocus(self.pin_edit_control) auth = self.getControl(AUTH_BUTTON) never = self.getControl(NEVER_BUTTON) self.pin_edit_control.controlUp(never) self.pin_edit_control.controlLeft(never) self.pin_edit_control.controlDown(auth) self.pin_edit_control.controlRight(auth) auth.controlUp(self.pin_edit_control) auth.controlLeft(self.pin_edit_control) never.controlDown(self.pin_edit_control) never.controlRight(self.pin_edit_control) def onAction(self, action): # print 'Action: %s' % (action.getId()) if action == ACTION_PREVIOUS_MENU or action == ACTION_BACK: self.close() def onControl(self, control): # print 'onControl: %s' % (control) pass def onFocus(self, control): # print 'onFocus: %s' % (control) pass def onClick(self, control): # print 'onClick: %s' % (control) if control == AUTH_BUTTON: if not self.__get_token(): kodi.notify(msg=i18n('pin_auth_failed'), duration=5000) return self.auth = True if control == LATER_BUTTON: kodi.notify(msg=i18n('remind_in_24hrs'), duration=5000) kodi.set_setting('last_reminder', str(int(time.time()))) if control == NEVER_BUTTON: kodi.notify(msg=i18n('use_addon_settings'), duration=5000) kodi.set_setting('last_reminder', '-1') if control in [AUTH_BUTTON, LATER_BUTTON, NEVER_BUTTON]: self.close() def __get_token(self): pin = self.pin_edit_control.getText().strip() if pin: try: trakt_api = Trakt_API(use_https=use_https, timeout=trakt_timeout) result = trakt_api.get_token(pin=pin) kodi.set_setting('trakt_oauth_token', result['access_token']) kodi.set_setting('trakt_refresh_token', result['refresh_token']) profile = trakt_api.get_user_profile(cached=False) kodi.set_setting( 'trakt_user', '%s (%s)' % (profile['username'], profile['name'])) return True except Exception as e: log_utils.log('Trakt Authorization Failed: %s' % (e), log_utils.LOGDEBUG) return False return False # have to add edit controls programatically because getControl() (hard) crashes XBMC on them def __add_editcontrol(self, x, y, height, width): media_path = os.path.join(kodi.get_path(), 'resources', 'skins', 'Default', 'media') temp = xbmcgui.ControlEdit( 0, 0, 0, 0, '', font='font12', textColor='0xFFFFFFFF', focusTexture=os.path.join(media_path, 'button-focus2.png'), noFocusTexture=os.path.join(media_path, 'button-nofocus.png'), _alignment=CENTER_Y | CENTER_X) temp.setPosition(x, y) temp.setHeight(height) temp.setWidth(width) self.addControl(temp) return temp dialog = PinAuthDialog('TraktPinAuthDialog.xml', kodi.get_path()) dialog.doModal() if dialog.auth: kodi.notify(msg=i18n('trakt_auth_complete'), duration=3000) del dialog
def get_pin(): AUTH_BUTTON = 200 LATER_BUTTON = 201 NEVER_BUTTON = 202 ACTION_PREVIOUS_MENU = 10 ACTION_BACK = 92 CENTER_Y = 6 CENTER_X = 2 class PinAuthDialog(xbmcgui.WindowXMLDialog): auth = False def onInit(self): self.pin_edit_control = self.__add_editcontrol(30, 240, 40, 450) self.setFocus(self.pin_edit_control) auth = self.getControl(AUTH_BUTTON) never = self.getControl(NEVER_BUTTON) self.pin_edit_control.controlUp(never) self.pin_edit_control.controlLeft(never) self.pin_edit_control.controlDown(auth) self.pin_edit_control.controlRight(auth) auth.controlUp(self.pin_edit_control) auth.controlLeft(self.pin_edit_control) never.controlDown(self.pin_edit_control) never.controlRight(self.pin_edit_control) def onAction(self, action): # print 'Action: %s' % (action.getId()) if action == ACTION_PREVIOUS_MENU or action == ACTION_BACK: self.close() def onControl(self, control): # print 'onControl: %s' % (control) pass def onFocus(self, control): # print 'onFocus: %s' % (control) pass def onClick(self, control): # print 'onClick: %s' % (control) if control == AUTH_BUTTON: if not self.__get_token(): kodi.notify(msg=i18n('pin_auth_failed'), duration=5000) return self.auth = True if control == LATER_BUTTON: kodi.notify(msg=i18n('remind_in_24hrs'), duration=5000) kodi.set_setting('last_reminder', str(int(time.time()))) if control == NEVER_BUTTON: kodi.notify(msg=i18n('use_addon_settings'), duration=5000) kodi.set_setting('last_reminder', '-1') if control in [AUTH_BUTTON, LATER_BUTTON, NEVER_BUTTON]: self.close() def __get_token(self): pin = self.pin_edit_control.getText().strip() if pin: try: trakt_api = Trakt_API(use_https=use_https, timeout=trakt_timeout) result = trakt_api.get_token(pin=pin) kodi.set_setting('trakt_oauth_token', result['access_token']) kodi.set_setting('trakt_refresh_token', result['refresh_token']) profile = trakt_api.get_user_profile(cached=False) kodi.set_setting('trakt_user', '%s (%s)' % (profile['username'], profile['name'])) return True except Exception as e: log_utils.log('Trakt Authorization Failed: %s' % (e), log_utils.LOGDEBUG) return False return False # have to add edit controls programatically because getControl() (hard) crashes XBMC on them def __add_editcontrol(self, x, y, height, width): media_path = os.path.join(kodi.get_path(), 'resources', 'skins', 'Default', 'media') temp = xbmcgui.ControlEdit(0, 0, 0, 0, '', font='font12', textColor='0xFFFFFFFF', focusTexture=os.path.join(media_path, 'button-focus2.png'), noFocusTexture=os.path.join(media_path, 'button-nofocus.png'), _alignment=CENTER_Y | CENTER_X) temp.setPosition(x, y) temp.setHeight(height) temp.setWidth(width) self.addControl(temp) return temp dialog = PinAuthDialog('TraktPinAuthDialog.xml', kodi.get_path()) dialog.doModal() if dialog.auth: kodi.notify(msg=i18n('trakt_auth_complete'), duration=3000) del dialog
def do_auto_config(): ACTION_PREVIOUS_MENU = 10 ACTION_BACK = 92 CONTINUE_BUTTON = 200 CANCEL_BUTTON = 201 starty = 60 posx = 30 gap = 35 RADIO_BUTTONS = [ i18n('set_trakt_timeout'), i18n('set_cal_start'), i18n('set_cal_airtime'), i18n('set_scraper_timeout'), i18n('set_wl_mne'), i18n('set_test_direct'), i18n('set_filter_unusable'), i18n('set_show_debrid'), i18n('set_no_limit'), i18n('set_source_sort'), i18n('set_sso'), i18n('set_reset_url'), i18n('select_all_none')] class AutoConfDialog(xbmcgui.WindowXMLDialog): def onInit(self): log_utils.log('onInit:', log_utils.LOGDEBUG) self.OK = False self.radio_buttons = [] posy = starty for label in RADIO_BUTTONS: self.radio_buttons.append(self.__get_radio_button(posx, posy, label)) posy += gap try: responses = json.loads(kodi.get_setting('prev_responses')) except: responses = [True] * len(self.radio_buttons) if len(responses) < len(self.radio_buttons): responses += [True] * (len(self.radio_buttons) - len(responses)) self.addControls(self.radio_buttons) last_button = None for response, radio_button in zip(responses, self.radio_buttons): radio_button.setSelected(response) if last_button is not None: radio_button.controlUp(last_button) radio_button.controlLeft(last_button) last_button.controlDown(radio_button) last_button.controlRight(radio_button) last_button = radio_button continue_ctrl = self.getControl(CONTINUE_BUTTON) cancel_ctrl = self.getControl(CANCEL_BUTTON) self.radio_buttons[0].controlUp(cancel_ctrl) self.radio_buttons[0].controlLeft(cancel_ctrl) self.radio_buttons[-1].controlDown(continue_ctrl) self.radio_buttons[-1].controlRight(continue_ctrl) continue_ctrl.controlUp(self.radio_buttons[-1]) continue_ctrl.controlLeft(self.radio_buttons[-1]) cancel_ctrl.controlDown(self.radio_buttons[0]) cancel_ctrl.controlRight(self.radio_buttons[0]) def __get_radio_button(self, x, y, label): kwargs = {'font': 'font12', 'focusTexture': 'button-focus2.png', 'noFocusTexture': 'button-nofocus.png', 'focusOnTexture': 'radiobutton-focus.png', 'noFocusOnTexture': 'radiobutton-focus.png', 'focusOffTexture': 'radiobutton-nofocus.png', 'noFocusOffTexture': 'radiobutton-nofocus.png'} temp = xbmcgui.ControlRadioButton(x, y, 450, 30, label, **kwargs) return temp def onAction(self, action): # log_utils.log('Action: %s' % (action.getId()), log_utils.LOGDEBUG) if action == ACTION_PREVIOUS_MENU or action == ACTION_BACK: self.close() def onControl(self, control): # log_utils.log('onControl: %s' % (control), log_utils.LOGDEBUG) pass def onFocus(self, control): # log_utils.log('onFocus: %s' % (control), log_utils.LOGDEBUG) pass def onClick(self, control): # log_utils.log('onClick: %s' % (control), log_utils.LOGDEBUG) focus_button = self.getControl(control) if focus_button == self.radio_buttons[-1]: all_status = focus_button.isSelected() for button in self.radio_buttons: button.setSelected(all_status) if control == CONTINUE_BUTTON: self.OK = True if control == CANCEL_BUTTON: self.OK = False if control == CONTINUE_BUTTON or control == CANCEL_BUTTON: self.close() def get_responses(self): return [bool(button.isSelected()) for button in self.radio_buttons] dialog = AutoConfDialog('AutoConfDialog.xml', kodi.get_path()) dialog.doModal() if dialog.OK: responses = dialog.get_responses() kodi.set_setting('prev_responses', json.dumps(responses)) perform_auto_conf(responses) del dialog
def download_media(url, path, file_name): try: progress = int(kodi.get_setting('down_progress')) request = urllib2.Request(url) request.add_header('User-Agent', USER_AGENT) request.add_unredirected_header('Host', request.get_host()) response = urllib2.urlopen(request) content_length = 0 if 'Content-Length' in response.info(): content_length = int(response.info()['Content-Length']) file_name = file_name.replace('.strm', get_extension(url, response)) full_path = os.path.join(path, file_name) log_utils.log('Downloading: %s -> %s' % (url, full_path), log_utils.LOGDEBUG) path = xbmc.makeLegalFilename(path) if not xbmcvfs.exists(path): try: try: xbmcvfs.mkdirs(path) except: os.mkdir(path) except Exception as e: raise Exception(i18n('failed_create_dir')) file_desc = xbmcvfs.File(full_path, 'w') total_len = 0 if progress: if progress == PROGRESS.WINDOW: dialog = xbmcgui.DialogProgress() else: dialog = xbmcgui.DialogProgressBG() dialog.create('Stream All The Sources', i18n('downloading') % (file_name)) dialog.update(0) while True: data = response.read(CHUNK_SIZE) if not data: break if progress == PROGRESS.WINDOW and dialog.iscanceled(): break total_len += len(data) if not file_desc.write(data): raise Exception('failed_write_file') percent_progress = (total_len) * 100 / content_length if content_length > 0 else 0 log_utils.log('Position : %s / %s = %s%%' % (total_len, content_length, percent_progress), log_utils.LOGDEBUG) if progress == PROGRESS.WINDOW: dialog.update(percent_progress) elif progress == PROGRESS.BACKGROUND: dialog.update(percent_progress, 'Stream All The Sources') else: kodi.notify(msg=i18n('download_complete') % (file_name), duration=5000) log_utils.log('Download Complete: %s -> %s' % (url, full_path), log_utils.LOGDEBUG) file_desc.close() if progress: dialog.close() except Exception as e: log_utils.log('Error (%s) during download: %s -> %s' % (str(e), url, file_name), log_utils.LOGERROR) kodi.notify(msg=i18n('download_error') % (str(e), file_name), duration=5000)
def do_auto_config(): ACTION_PREVIOUS_MENU = 10 ACTION_BACK = 92 CONTINUE_BUTTON = 200 CANCEL_BUTTON = 201 starty = 60 posx = 30 gap = 35 RADIO_BUTTONS = [ i18n('set_trakt_timeout'), i18n('set_cal_start'), i18n('set_cal_airtime'), i18n('set_scraper_timeout'), i18n('set_wl_mne'), i18n('set_test_direct'), i18n('set_filter_unusable'), i18n('set_show_debrid'), i18n('set_no_limit'), i18n('set_source_sort'), i18n('set_sso'), i18n('set_reset_url'), i18n('select_all_none') ] class AutoConfDialog(xbmcgui.WindowXMLDialog): def onInit(self): log_utils.log('onInit:', log_utils.LOGDEBUG) self.OK = False self.radio_buttons = [] posy = starty for label in RADIO_BUTTONS: self.radio_buttons.append( self.__get_radio_button(posx, posy, label)) posy += gap try: responses = json.loads(kodi.get_setting('prev_responses')) except: responses = [True] * len(self.radio_buttons) if len(responses) < len(self.radio_buttons): responses += [True ] * (len(self.radio_buttons) - len(responses)) self.addControls(self.radio_buttons) last_button = None for response, radio_button in zip(responses, self.radio_buttons): radio_button.setSelected(response) if last_button is not None: radio_button.controlUp(last_button) radio_button.controlLeft(last_button) last_button.controlDown(radio_button) last_button.controlRight(radio_button) last_button = radio_button continue_ctrl = self.getControl(CONTINUE_BUTTON) cancel_ctrl = self.getControl(CANCEL_BUTTON) self.radio_buttons[0].controlUp(cancel_ctrl) self.radio_buttons[0].controlLeft(cancel_ctrl) self.radio_buttons[-1].controlDown(continue_ctrl) self.radio_buttons[-1].controlRight(continue_ctrl) continue_ctrl.controlUp(self.radio_buttons[-1]) continue_ctrl.controlLeft(self.radio_buttons[-1]) cancel_ctrl.controlDown(self.radio_buttons[0]) cancel_ctrl.controlRight(self.radio_buttons[0]) def __get_radio_button(self, x, y, label): kwargs = { 'font': 'font12', 'focusTexture': 'button-focus2.png', 'noFocusTexture': 'button-nofocus.png', 'focusOnTexture': 'radiobutton-focus.png', 'noFocusOnTexture': 'radiobutton-focus.png', 'focusOffTexture': 'radiobutton-nofocus.png', 'noFocusOffTexture': 'radiobutton-nofocus.png' } temp = xbmcgui.ControlRadioButton(x, y, 450, 30, label, **kwargs) return temp def onAction(self, action): # log_utils.log('Action: %s' % (action.getId()), log_utils.LOGDEBUG) if action == ACTION_PREVIOUS_MENU or action == ACTION_BACK: self.close() def onControl(self, control): # log_utils.log('onControl: %s' % (control), log_utils.LOGDEBUG) pass def onFocus(self, control): # log_utils.log('onFocus: %s' % (control), log_utils.LOGDEBUG) pass def onClick(self, control): # log_utils.log('onClick: %s' % (control), log_utils.LOGDEBUG) focus_button = self.getControl(control) if focus_button == self.radio_buttons[-1]: all_status = focus_button.isSelected() for button in self.radio_buttons: button.setSelected(all_status) if control == CONTINUE_BUTTON: self.OK = True if control == CANCEL_BUTTON: self.OK = False if control == CONTINUE_BUTTON or control == CANCEL_BUTTON: self.close() def get_responses(self): return [bool(button.isSelected()) for button in self.radio_buttons] dialog = AutoConfDialog('AutoConfDialog.xml', kodi.get_path()) dialog.doModal() if dialog.OK: responses = dialog.get_responses() kodi.set_setting('prev_responses', json.dumps(responses)) perform_auto_conf(responses) del dialog