From af6f9473c42f6364fc8e637ad84dea9a6328d6d8 Mon Sep 17 00:00:00 2001
From: Ruud <ruud@crashdummy.nl>
Date: Thu, 28 Jun 2012 23:14:27 +0200
Subject: [PATCH] Simplified preferred method

---
 couchpotato/core/plugins/searcher/__init__.py |  7 ++++---
 couchpotato/core/plugins/searcher/main.py     | 22 +++++-----------------
 2 files changed, 9 insertions(+), 20 deletions(-)

diff --git a/couchpotato/core/plugins/searcher/__init__.py b/couchpotato/core/plugins/searcher/__init__.py
index c68ada0..f499e2b 100644
--- a/couchpotato/core/plugins/searcher/__init__.py
+++ b/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')],
                 },
             ],
         }, {
diff --git a/couchpotato/core/plugins/searcher/main.py b/couchpotato/core/plugins/searcher/main.py
index 22f297a..0f4dacc 100644
--- a/couchpotato/core/plugins/searcher/main.py
+++ b/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