From 3f16dbd09c58570b4ee89f00ddcdc1c91e76f607 Mon Sep 17 00:00:00 2001 From: Ruud Date: Fri, 4 Apr 2014 17:52:33 +0200 Subject: [PATCH] Sort releases based on preferred method in api return --- couchpotato/core/plugins/release/main.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/couchpotato/core/plugins/release/main.py b/couchpotato/core/plugins/release/main.py index 437e736c..72a0a502 100644 --- a/couchpotato/core/plugins/release/main.py +++ b/couchpotato/core/plugins/release/main.py @@ -446,7 +446,14 @@ class Release(Plugin): raw_releases = list(db.get_many('release', media_id, with_doc = True)) releases = [] - for r in sorted(raw_releases, key = lambda k: k['doc'].get('info', {}).get('score', 0), reverse = True): + for r in raw_releases: releases.append(r['doc']) + releases = sorted(releases, key = lambda k: k.get('info', {}).get('score', 0), reverse = True) + + # Sort based on preferred search method + download_preference = self.conf('preferred_method', section = 'searcher') + if download_preference != 'both': + releases = sorted(releases, key = lambda k: k.get('info', {}).get('protocol', '')[:3], reverse = (download_preference == 'torrent')) + return releases