def makeNewznabProvider(configString): if not configString: return None search_mode = 'eponly' search_fallback = 0 enable_daily = 0 enable_backlog = 0 try: values = configString.split('|') if len(values) == 9: name, url, key, catIDs, enabled, search_mode, search_fallback, enable_daily, enable_backlog = values else: name = values[0] url = values[1] key = values[2] catIDs = values[3] enabled = values[4] except ValueError: logger.log(u"Skipping Newznab provider string: '" + configString + "', incorrect format", logger.ERROR) return None # newznab = sys.modules['sickbeard.providers.newznab'] newProvider = newznab.NewznabProvider(name, url, key=key, catIDs=catIDs, search_mode=search_mode, search_fallback=search_fallback, enable_daily=enable_daily, enable_backlog=enable_backlog) newProvider.enabled = enabled == '1' return newProvider
def test_parse_ids(self): ids_test_cases = [] for k in newznab.NewznabConstants.providerToIndexerMapping.iterkeys(): rand_id = random.randrange(1, 99999999) ids_test_cases.append(({ k: rand_id }, { newznab.NewznabConstants.providerToIndexerMapping[k]: rand_id })) all_case = {} all_case_ex = {} for k in newznab.NewznabConstants.providerToIndexerMapping.iterkeys(): rand_id = random.randrange(1, 99999999) all_case.update({k: rand_id}) all_case_ex.update({ newznab.NewznabConstants.providerToIndexerMapping[k]: rand_id }) ids_test_cases.append((all_case, all_case_ex)) newznab_provider = newznab.NewznabProvider('test', '') for cur_test, cur_expected in ids_test_cases: item = self._create_item('Show.Name.S01E01.x264-Group', 'https://test.h', ids=cur_test) result = newznab_provider.cache.parse_ids(item, BasicTests.ns_parsed) self.assertEqual(cur_expected, result)
def getNewznabCategories(name, url, key): """ Retrieves a list of possible categories with category id's Using the default url/api?cat http://yournewznaburl.com/api?t=caps&apikey=yourapikey """ error = "" success = False if not name: error += "\n" + _("No Provider Name specified") if not url: error += "\n" + _("No Provider Url specified") if not key: error += "\n" + _("No Provider Api key specified") if error: return json.dumps({'success': False, 'error': error}) # Get newznabprovider obj with provided name tempProvider = newznab.NewznabProvider(name, url, key) success, tv_categories, error = tempProvider.get_newznab_categories() return json.dumps({'success': success, 'tv_categories': tv_categories, 'error': error})
def test_parse_pub_date(self): newznab_provider = newznab.NewznabProvider('test', '') for cur_test in pubdate_test_cases: item = self._create_item('Show.Name.S01E01.x264-Group', 'https://test.h', pubdate=cur_test) result = newznab_provider._parse_pub_date(item) cur_expected = self._parse_pub_date(cur_test) self.assertEqual(cur_expected, result)
def test_get_size_uid(self): newznab_provider = newznab.NewznabProvider('test', '') for cur_test, cur_expected in size_test_cases: item = self._create_item('Show.Name.S01E01.x264-Group', 'http://test.h', cur_test[0], cur_test[1]) result = newznab_provider.get_size_uid(item, name_space=BasicTests.ns_parsed) self.assertEqual(cur_expected, result)
def test_title_and_url(self): if VERBOSE: print('Running tests') newznab_provider = newznab.NewznabProvider('test', '') for cur_test, cur_expected in item_parse_test_cases: item = self._create_item(cur_test[0], cur_test[1]) result = newznab_provider._title_and_url(item) self.assertEqual(cur_expected, result)
def saveProviders(self, newznab_string='', torrentrss_string='', provider_order=None, **kwargs): newznabProviderDict = dict( zip((x.get_id() for x in sickbeard.newznabProviderList), sickbeard.newznabProviderList)) finished_names = [] # add all the newznab info we got into our list # if not newznab_string: # logger.log('No newznab_string passed to saveProviders', logger.DEBUG) for curNewznabProviderStr in newznab_string.split('!!!'): if not curNewznabProviderStr: continue cur_name, cur_url, cur_key, cur_cat = curNewznabProviderStr.split('|') cur_url = config.clean_url(cur_url) cur_id = GenericProvider.make_id(cur_name) # if it does not already exist then add it if cur_id not in newznabProviderDict: new_provider = newznab.NewznabProvider(cur_name, cur_url, key=cur_key, catIDs=cur_cat) sickbeard.newznabProviderList.append(new_provider) newznabProviderDict[cur_id] = new_provider # set all params newznabProviderDict[cur_id].name = cur_name newznabProviderDict[cur_id].url = cur_url newznabProviderDict[cur_id].key = cur_key newznabProviderDict[cur_id].catIDs = cur_cat # a 0 in the key spot indicates that no key is needed newznabProviderDict[cur_id].needs_auth = cur_key and cur_key != '0' newznabProviderDict[cur_id].search_mode = str(kwargs.get(cur_id + '_search_mode', 'eponly')).strip() newznabProviderDict[cur_id].search_fallback = config.checkbox_to_value(kwargs.get(cur_id + 'search_fallback', 0), value_on=1, value_off=0) newznabProviderDict[cur_id].enable_daily = config.checkbox_to_value(kwargs.get(cur_id + 'enable_daily', 0), value_on=1, value_off=0) newznabProviderDict[cur_id].enable_backlog = config.checkbox_to_value(kwargs.get(cur_id + 'enable_backlog', 0), value_on=1, value_off=0) # mark it finished finished_names.append(cur_id) # delete anything that is in the list that was not processed just now if newznab_string: for curProvider in sickbeard.newznabProviderList: if curProvider.get_id() not in finished_names: sickbeard.newznabProviderList.remove(curProvider) del newznabProviderDict[curProvider.get_id()] # if not torrentrss_string: # logger.log('No torrentrss_string passed to saveProviders', logger.DEBUG) torrentRssProviderDict = dict( zip((x.get_id() for x in sickbeard.torrentRssProviderList), sickbeard.torrentRssProviderList)) finished_names = [] if torrentrss_string: for curTorrentRssProviderStr in torrentrss_string.split('!!!'): if not curTorrentRssProviderStr: continue cur_name, cur_url, cur_cookies, cur_title_tag = curTorrentRssProviderStr.split('|') cur_url = config.clean_url(cur_url) cur_id = GenericProvider.make_id(cur_name) # if it does not already exist then create it if cur_id not in torrentRssProviderDict: new_provider = rsstorrent.TorrentRssProvider(cur_name, cur_url, cur_cookies, cur_title_tag) sickbeard.torrentRssProviderList.append(new_provider) torrentRssProviderDict[cur_id] = new_provider # update values torrentRssProviderDict[cur_id].name = cur_name torrentRssProviderDict[cur_id].url = cur_url torrentRssProviderDict[cur_id].cookies = cur_cookies torrentRssProviderDict[cur_id].cur_title_tag = cur_title_tag # mark it finished finished_names.append(cur_id) # delete anything that is in the list that was not processed just now if torrentrss_string: for curProvider in sickbeard.torrentRssProviderList: if curProvider.get_id() not in finished_names: sickbeard.torrentRssProviderList.remove(curProvider) del torrentRssProviderDict[curProvider.get_id()] # do the enable/disable enabled_provider_list = [] disabled_provider_list = [] for cur_id, cur_enabled in (cur_provider_str.split(':') for cur_provider_str in provider_order.split()): cur_enabled = bool(try_int(cur_enabled)) cur_provider_obj = [x for x in sickbeard.providers.sortedProviderList() if x.get_id() == cur_id and hasattr(x, 'enabled')] if cur_provider_obj: cur_provider_obj[0].enabled = cur_enabled if cur_enabled: enabled_provider_list.append(cur_id) else: disabled_provider_list.append(cur_id) if cur_id in newznabProviderDict: newznabProviderDict[cur_id].enabled = cur_enabled elif cur_id in torrentRssProviderDict: torrentRssProviderDict[cur_id].enabled = cur_enabled # dynamically load provider settings for curProvider in sickbeard.providers.sortedProviderList(): if hasattr(curProvider, 'custom_url'): curProvider.custom_url = str(kwargs.get(curProvider.get_id('_custom_url'), '')).strip() if hasattr(curProvider, 'minseed'): curProvider.minseed = int(str(kwargs.get(curProvider.get_id('_minseed'), 0)).strip()) if hasattr(curProvider, 'minleech'): curProvider.minleech = int(str(kwargs.get(curProvider.get_id('_minleech'), 0)).strip()) if hasattr(curProvider, 'ratio'): if curProvider.get_id('_ratio') in kwargs: ratio = str(kwargs.get(curProvider.get_id('_ratio'))).strip() print (ratio) if ratio in ('None', None, ''): curProvider.ratio = None else: curProvider.ratio = max(float(ratio), -1) else: curProvider.ratio = None if hasattr(curProvider, 'digest'): curProvider.digest = str(kwargs.get(curProvider.get_id('_digest'), '')).strip() or None if hasattr(curProvider, 'hash'): curProvider.hash = str(kwargs.get(curProvider.get_id('_hash'), '')).strip() or None if hasattr(curProvider, 'api_key'): curProvider.api_key = str(kwargs.get(curProvider.get_id('_api_key'), '')).strip() or None if hasattr(curProvider, 'username'): curProvider.username = str(kwargs.get(curProvider.get_id('_username'), '')).strip() or None if hasattr(curProvider, 'password'): curProvider.password = filters.unhide(curProvider.password, str(kwargs.get(curProvider.get_id('_password'), '')).strip()) if hasattr(curProvider, 'passkey'): curProvider.passkey = filters.unhide(curProvider.passkey, str(kwargs.get(curProvider.get_id('_passkey'), '')).strip()) if hasattr(curProvider, 'pin'): curProvider.pin = filters.unhide(curProvider.pin, str(kwargs.get(curProvider.get_id('_pin'), '')).strip()) if hasattr(curProvider, 'confirmed'): curProvider.confirmed = config.checkbox_to_value(kwargs.get(curProvider.get_id('_confirmed'))) if hasattr(curProvider, 'ranked'): curProvider.ranked = config.checkbox_to_value(kwargs.get(curProvider.get_id('_ranked'))) if hasattr(curProvider, 'engrelease'): curProvider.engrelease = config.checkbox_to_value(kwargs.get(curProvider.get_id('_engrelease'))) if hasattr(curProvider, 'onlyspasearch'): curProvider.onlyspasearch = config.checkbox_to_value(kwargs.get(curProvider.get_id('_onlyspasearch'))) if hasattr(curProvider, 'sorting'): curProvider.sorting = str(kwargs.get(curProvider.get_id('_sorting'), 'seeders')).strip() if hasattr(curProvider, 'freeleech'): curProvider.freeleech = config.checkbox_to_value(kwargs.get(curProvider.get_id('_freeleech'))) if hasattr(curProvider, 'search_mode'): curProvider.search_mode = str(kwargs.get(curProvider.get_id('_search_mode'), 'eponly')).strip() if hasattr(curProvider, 'search_fallback'): curProvider.search_fallback = config.checkbox_to_value(kwargs.get(curProvider.get_id('_search_fallback'))) if hasattr(curProvider, 'enable_daily'): curProvider.enable_daily = curProvider.can_daily and config.checkbox_to_value(kwargs.get(curProvider.get_id('_enable_daily'))) if hasattr(curProvider, 'enable_backlog'): curProvider.enable_backlog = curProvider.can_backlog and config.checkbox_to_value(kwargs.get(curProvider.get_id('_enable_backlog'))) if hasattr(curProvider, 'cat'): curProvider.cat = int(str(kwargs.get(curProvider.get_id('_cat'), 0)).strip()) if hasattr(curProvider, 'subtitle'): curProvider.subtitle = config.checkbox_to_value(kwargs.get(curProvider.get_id('_subtitle'))) if curProvider.enable_cookies: curProvider.cookies = str(kwargs.get(curProvider.get_id('_cookies'))).strip() sickbeard.NEWZNAB_DATA = '!!!'.join([x.configStr() for x in sickbeard.newznabProviderList]) sickbeard.PROVIDER_ORDER = enabled_provider_list + disabled_provider_list sickbeard.save_config() # Add a site_message if no providers are enabled for daily and/or backlog sickbeard.providers.check_enabled_providers() ui.notifications.message(_('Configuration Saved'), ek(os.path.join, sickbeard.CONFIG_FILE)) return self.redirect("/config/providers/")