dinislage/CouchPotatoServer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Return movie_id after rename/download. fix #362

Browse Source
This commit is contained in:
Ruud
2012-05-28 12:21:07 +02:00
parent 9575dff838
commit 57b8f0e9b7
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
couchpotato/core/plugins/scanner/main.py
+7 -1
View File
@@ -4,7 +4,7 @@ from couchpotato.core.helpers.encoding import toUnicode, simplifyString
from couchpotato.core.helpers.variable import getExt, getImdb, tryInt from couchpotato.core.helpers.variable import getExt, getImdb, tryInt
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
from couchpotato.core.settings.model import File from couchpotato.core.settings.model import File, Movie
from couchpotato.environment import Env from couchpotato.environment import Env
from enzyme.exceptions import NoParserError, ParseError from enzyme.exceptions import NoParserError, ParseError
from guessit import guess_movie_info from guessit import guess_movie_info
@@ -161,6 +161,8 @@ class Scanner(Plugin):
except: except:
log.error('Failed getting files from %s: %s' % (folder, traceback.format_exc())) log.error('Failed getting files from %s: %s' % (folder, traceback.format_exc()))
db = get_session()
for file_path in files: for file_path in files:
if not os.path.exists(file_path): if not os.path.exists(file_path):
@@ -359,6 +361,10 @@ class Scanner(Plugin):
group['library'] = self.determineMovie(group) group['library'] = self.determineMovie(group)
if not group['library']: if not group['library']:
log.error('Unable to determine movie: %s' % group['identifiers']) log.error('Unable to determine movie: %s' % group['identifiers'])
else:
movie = db.query(Movie).filter_by(library_id = group['library']['id']).first()
group['movie_id'] = None if not movie else movie.id
processed_movies[identifier] = group processed_movies[identifier] = group