diff --git a/couchpotato/core/media/__init__.py b/couchpotato/core/media/__init__.py index e6a249d5..cdf667c9 100644 --- a/couchpotato/core/media/__init__.py +++ b/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() diff --git a/couchpotato/core/media/_base/media/main.py b/couchpotato/core/media/_base/media/main.py index 286d4043..d0ddf113 100644 --- a/couchpotato/core/media/_base/media/main.py +++ b/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