Try next on failed download from provider

This commit is contained in:
Ruud
2012-11-15 18:30:45 +01:00
parent 2b49a4b5d6
commit e53a9ed30a
6 changed files with 12 additions and 10 deletions
+1 -1
View File
@@ -239,7 +239,7 @@ class Searcher(Plugin):
filedata = None
if data.get('download') and (ismethod(data.get('download')) or isfunction(data.get('download'))):
filedata = data.get('download')(url = data.get('url'), nzb_id = data.get('id'))
if filedata is 'try_next':
if filedata == 'try_next':
return filedata
successful = fireEvent('download', data = data, movie = movie, manual = manual, filedata = filedata, single = True)
+7 -1
View File
@@ -6,6 +6,7 @@ from couchpotato.environment import Env
from urlparse import urlparse
import re
import time
import traceback
log = CPLog(__name__)
@@ -56,7 +57,12 @@ class YarrProvider(Provider):
addEvent('nzb.feed', self.feed)
def download(self, url = '', nzb_id = ''):
return self.urlopen(url)
try:
return self.urlopen(url, headers = {'User-Agent': Env.getIdentifier()}, show_error = False)
except:
log.error('Failed getting nzb from %s: %s', (self.getName(), traceback.format_exc()))
return 'try_next'
def feed(self):
return []
@@ -144,7 +144,8 @@ class Newzbin(NZBProvider, RSS):
}, show_error = False)
except Exception, e:
log.error('Failed downloading from newzbin, check credit: %s', e)
return False
return 'try_next'
def getFormatId(self, format):
for id, quality in self.format_ids.iteritems():
@@ -219,4 +219,5 @@ class Newznab(NZBProvider, RSS):
return 'try_next'
log.error('Failed download from %s', (host, traceback.format_exc()))
raise
return 'try_next'
@@ -98,9 +98,6 @@ class NZBMatrix(NZBProvider, RSS):
return results
def download(self, url = '', nzb_id = ''):
return self.urlopen(url, headers = {'User-Agent': Env.getIdentifier()})
def getApiExt(self):
return '&username=%s&apikey=%s' % (self.conf('username'), self.conf('api_key'))
@@ -97,8 +97,5 @@ class Nzbsrus(NZBProvider, RSS):
return results
def download(self, url = '', nzb_id = ''):
return self.urlopen(url, headers = {'User-Agent': Env.getIdentifier()})
def getApiExt(self):
return '/%s/' % (self.conf('userid'))