Sort tmdb search results

This commit is contained in:
Ruud committed 2012-04-15 23:18:48 +02:00
1 parent e3a9bb1301
commit 8c922531e2
1 file changed
+6 -1
@@ -3,6 +3,7 @@ from couchpotato.core.helpers.encoding import simplifyString, toUnicode
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.providers.movie.base import MovieProvider from couchpotato.core.providers.movie.base import MovieProvider
from libs.themoviedb import tmdb from libs.themoviedb import tmdb
import re
log = CPLog(__name__) log = CPLog(__name__)
@@ -67,8 +68,12 @@ class TheMovieDb(MovieProvider):
if raw: if raw:
try: try:
nr = 0 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)) results.append(self.parseMovie(movie))
nr += 1 nr += 1