Use Media for all Movie db actions

This commit is contained in:
Ruud committed 2013-10-08 09:57:36 +02:00
1 parent 107606ce65
commit d1c3f0c241
11 files changed
+54 -54

No files matched your search

+2 -2
View File
@@ -4,7 +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
from couchpotato.core.settings.model import Media
log = CPLog(__name__)
@@ -26,7 +26,7 @@ class MediaPlugin(MediaBase):
db = get_session()
for x in splitString(id):
media = db.query(Movie).filter_by(id = x).first()
media = db.query(Media).filter_by(id = x).first()
if media:
# Get current selected title
@@ -5,7 +5,7 @@ from couchpotato.core.helpers.encoding import simplifyString, toUnicode
from couchpotato.core.helpers.variable import md5, getTitle
from couchpotato.core.logger import CPLog
from couchpotato.core.media._base.searcher.base import SearcherBase
from couchpotato.core.settings.model import Movie, Release, ReleaseInfo
from couchpotato.core.settings.model import Media, Release, ReleaseInfo
from couchpotato.environment import Env
from inspect import ismethod, isfunction
import datetime
@@ -104,7 +104,7 @@ class Searcher(SearcherBase):
if profile_type['quality_id'] == rls.quality.id and profile_type['finish']:
# Mark movie done
log.info('Renamer disabled, marking movie as finished: %s', log_movie)
mvie = db.query(Movie).filter_by(id = movie['id']).first()
mvie = db.query(Media).filter_by(id = movie['id']).first()
mvie.status_id = done_status.get('id')
mvie.last_edit = int(time.time())
db.commit()