Add library query

couchpotato/core/media/_base/providers/base.py

@@ -200,7 +200,7 @@ class YarrProvider(Provider):
             self._search(media, quality, results)
         # Search possible titles
         else:
-            media_title = fireEvent('library.query', media['library'], single = True)
+            media_title = fireEvent('library.query', media, single = True)
 
             for title in possibleTitles(media_title):
                 self._searchOnTitle(title, media, quality, results)

couchpotato/core/media/movie/library.py

@@ -0,0 +1,29 @@
+from couchpotato.core.event import addEvent
+from couchpotato.core.logger import CPLog
+from couchpotato.core.media._base.library.base import LibraryBase
+
+
+log = CPLog(__name__)
+
+autoload = 'MovieLibraryPlugin'
+
+
+class MovieLibraryPlugin(LibraryBase):
+
+    def __init__(self):
+        addEvent('library.query', self.query)
+
+    def query(self, media, first = True, include_year = True, **kwargs):
+        if media.get('type') != 'movie':
+            return
+
+        titles = media['info'].get('titles', [])
+
+        # Add year identifier to titles
+        if include_year:
+            titles = [title + (' %s' % str(media['info']['year'])) for title in titles]
+
+        if first:
+            return titles[0] if titles else None
+
+        return titles