Use movie instaid of media model

couchpotato/core/media/__init__.py

@@ -1,7 +1,7 @@
 from couchpotato import get_session
 from couchpotato.core.event import addEvent, fireEventAsync, fireEvent
 from couchpotato.core.plugins.base import Plugin
-from couchpotato.core.settings.model import Media
+from couchpotato.core.settings.model import Movie
 
 
 class MediaBase(Plugin):
@@ -27,7 +27,7 @@ class MediaBase(Plugin):
 
         def onComplete():
             db = get_session()
-            media = db.query(Media).filter_by(id = id).first()
+            media = db.query(Movie).filter_by(id = id).first()
             fireEventAsync('%s.searcher.single' % media.type, media.to_dict(self.default_dict), on_complete = self.createNotifyFront(id))
             db.expire_all()
 
@@ -37,7 +37,7 @@ class MediaBase(Plugin):
 
         def notifyFront():
             db = get_session()
-            media = db.query(Media).filter_by(id = media_id).first()
+            media = db.query(Movie).filter_by(id = media_id).first()
             fireEvent('notify.frontend', type = '%s.update.%s' % (media.type, media.id), data = media.to_dict(self.default_dict))
             db.expire_all()
 

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

@@ -4,6 +4,7 @@ from couchpotato.core.event import fireEvent, fireEventAsync, addEvent
 from couchpotato.core.helpers.variable import splitString
 from couchpotato.core.logger import CPLog
 from couchpotato.core.media import MediaBase
+from couchpotato.core.settings.model import Movie
 
 log = CPLog(__name__)
 
@@ -25,7 +26,7 @@ class MediaPlugin(MediaBase):
         db = get_session()
 
         for x in splitString(id):
-            media = db.query(Media).filter_by(id = x).first()
+            media = db.query(Movie).filter_by(id = x).first()
 
             if media:
                 # Get current selected title