Browse Source

Simplified preferred method

pull/477/merge
Ruud 13 years ago
parent
commit
af6f9473c4
  1. 7
      couchpotato/core/plugins/searcher/__init__.py
  2. 22
      couchpotato/core/plugins/searcher/main.py

7
couchpotato/core/plugins/searcher/__init__.py

@ -33,10 +33,11 @@ config = [{
},
{
'name': 'preferred_method',
'label': 'Preferred download method',
'default': 'both',
'label': 'First search',
'description': 'Which of the methods do you prefer',
'default': 'nzb',
'type': 'dropdown',
'values': [('usenet & torrents', 'both'), ('usenet', 'usenet'), ('torrents', 'torrents')],
'values': [('usenet & torrents', 'both'), ('usenet', 'nzb'), ('torrents', 'torrent')],
},
],
}, {

22
couchpotato/core/plugins/searcher/main.py

@ -105,28 +105,16 @@ class Searcher(Plugin):
log.info('Search for %s in %s', (default_title, quality_type['quality']['label']))
quality = fireEvent('quality.single', identifier = quality_type['quality']['identifier'], single = True)
download_preference = self.conf('preferred_method')
if download_preference == 'torrents':
log.info('First searching torrents')
results = fireEvent('torrent.search', movie, quality, merge = True)
if not results:
log.info('No results from torrents, searching usenet')
results = fireEvent('nzb.search', movie, quality, merge = True)
elif download_preference == 'usenet':
log.info('First searching usenet')
results = fireEvent('nzb.search', movie, quality, merge = True)
if not results:
log.info('No results from usenet, searching torrents')
results = fireEvent('torrent.search', movie, quality, merge = True)
else:
results = fireEvent('yarr.search', movie, quality, merge = True)
results = fireEvent('yarr.search', movie, quality, merge = True)
sorted_results = sorted(results, key = lambda k: k['score'], reverse = True)
if len(sorted_results) == 0:
log.debug('Nothing found for %s in %s', (default_title, quality_type['quality']['label']))
download_preference = self.conf('preferred_method')
if download_preference != 'both':
sorted_results = sorted(sorted_results, key = lambda k: k['type'], reverse = (download_preference == 'torrent'))
# Check if movie isn't deleted while searching
if not db.query(Movie).filter_by(id = movie.get('id')).first():
break

Loading…
Cancel
Save