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/")