From 8c922531e2aca0bf1c0b08a754c4ed3863a75250 Mon Sep 17 00:00:00 2001 From: Ruud Date: Sun, 15 Apr 2012 23:18:48 +0200 Subject: [PATCH] Sort tmdb search results --- couchpotato/core/providers/movie/themoviedb/main.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/couchpotato/core/providers/movie/themoviedb/main.py b/couchpotato/core/providers/movie/themoviedb/main.py index 2838c4ce..50aa2392 100644 --- a/couchpotato/core/providers/movie/themoviedb/main.py +++ b/couchpotato/core/providers/movie/themoviedb/main.py @@ -3,6 +3,7 @@ from couchpotato.core.helpers.encoding import simplifyString, toUnicode from couchpotato.core.logger import CPLog from couchpotato.core.providers.movie.base import MovieProvider from libs.themoviedb import tmdb +import re log = CPLog(__name__) @@ -67,8 +68,12 @@ class TheMovieDb(MovieProvider): if raw: try: nr = 0 - for movie in raw: + # Sort on returned score first when year is in q + if re.search('\s\d{4}', q): + movies = sorted(raw, key = lambda k: k['score'], reverse = True) + + for movie in movies: results.append(self.parseMovie(movie)) nr += 1