moparisthebest/SickRage
1
0
Fork 0
mirror of https://github.com/moparisthebest/SickRage synced 2024-12-13 11:32:20 -05:00
Code Issues Releases Wiki Activity

Improved newznab offset code

Browse Source
This commit is contained in:
echel0n 2014-09-07 00:44:48 -07:00
parent 20c0b4ea72
commit ab16430b1a
1 changed files with 12 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
sickbeard/providers/newznab.py
View File

@ -264,7 +264,10 @@ class NewznabProvider(generic.NZBProvider):
results = [] results = []
while True: # get and set total items available
offset = total = 0
while total >= offset:
search_url = self.url + 'api?' + urllib.urlencode(params) search_url = self.url + 'api?' + urllib.urlencode(params)
logger.log(u"Search url: " + search_url, logger.DEBUG) logger.log(u"Search url: " + search_url, logger.DEBUG)
data = self.cache.getRSSFeed(search_url) data = self.cache.getRSSFeed(search_url)
@ -281,37 +284,17 @@ class NewznabProvider(generic.NZBProvider):
u"The data returned from the " + self.name + " is incomplete, this result is unusable", u"The data returned from the " + self.name + " is incomplete, this result is unusable",
logger.DEBUG) logger.DEBUG)
# attempt to grab the total and offset newznab responses # get total and offset attribs
try: if total == 0:
total = int(data.feed.newznab_response['total']) total = int(data.feed.newznab_response['total'] or 0)
offset = int(data.feed.newznab_response['offset']) offset = int(data.feed.newznab_response['offset'] or 0)
except (AttributeError, TypeError):
break
# sanity check - limiting at 10 at getting 1000 results in-case incorrect total parameter is reported
if params['limit'] > 1000:
logger.log("Excessive results for search, ending search", logger.WARNING)
break
# sanity check - total should remain constant
if offset != 0 and total != initial_total:
logger.log("Total number of items on newznab response changed, ending search", logger.DEBUG)
break
else:
initial_total = total
# if there are more items available then the amount given in one call, grab some more # if there are more items available then the amount given in one call, grab some more
if (total - params['limit']) > offset == params['offset']: params['offset'] += params['limit']
params['offset'] += params['limit']
logger.log(str(
total - params['offset']) + " more items to be fetched from provider. Fetching another " + str(
params['limit']) + " items.", logger.DEBUG)
else:
break
logger.log(str(
else: total - offset) + " more items to be fetched from provider. Fetching another " + str(
break params['limit']) + " items.", logger.DEBUG)
return results return results