Browse Source

Allow search on imdb urls

pull/3275/merge
Ruud 11 years ago
parent
commit
90a618bd7e
  1. 15
      couchpotato/core/media/_base/search/main.py

15
couchpotato/core/media/_base/search/main.py

@ -1,6 +1,6 @@
from couchpotato.api import addApiView from couchpotato.api import addApiView
from couchpotato.core.event import fireEvent, addEvent from couchpotato.core.event import fireEvent, addEvent
from couchpotato.core.helpers.variable import mergeDicts from couchpotato.core.helpers.variable import mergeDicts, getImdb
from couchpotato.core.logger import CPLog from couchpotato.core.logger import CPLog
from couchpotato.core.plugins.base import Plugin from couchpotato.core.plugins.base import Plugin
@ -35,12 +35,21 @@ class Search(Plugin):
elif isinstance(types, (list, tuple, set)): elif isinstance(types, (list, tuple, set)):
types = list(types) types = list(types)
imdb_identifier = getImdb(q)
if not types: if not types:
result = fireEvent('info.search', q = q, merge = True) if imdb_identifier:
result = fireEvent('movie.info', identifier = imdb_identifier, merge = True)
result = {result['type']: [result]}
else:
result = fireEvent('info.search', q = q, merge = True)
else: else:
result = {} result = {}
for media_type in types: for media_type in types:
result[media_type] = fireEvent('%s.search' % media_type) if imdb_identifier:
result[media_type] = fireEvent('%s.info' % media_type, identifier = imdb_identifier)
else:
result[media_type] = fireEvent('%s.search' % media_type, q = q)
return mergeDicts({ return mergeDicts({
'success': True, 'success': True,

Loading…
Cancel
Save