open-source-mirrors
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Continuing development

Browse Source
pull/302/head
Louis Vézina 6 years ago
parent aa77ab68a9
commit cfea295198
3 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File

15
bazarr/get_movies.py
Unescape View File

@ -77,6 +77,9 @@ def update_movies():
if movie['alternativeTitles'] != None: if movie['alternativeTitles'] != None:
alternativeTitles = str([item['title'] for item in movie['alternativeTitles']]) alternativeTitles = str([item['title'] for item in movie['alternativeTitles']])
if 'imdbId' in movie: imdbId = movie['imdbId']
else: imdbId = None
try: try:
format, resolution = movie['movieFile']['quality']['quality']['name'].split('-') format, resolution = movie['movieFile']['quality']['quality']['name'].split('-')
except: except:
@ -111,7 +114,7 @@ def update_movies():
profile_id_to_language(movie['qualityProfileId']), sceneName, profile_id_to_language(movie['qualityProfileId']), sceneName,
unicode(bool(movie['monitored'])), movie['sortTitle'], unicode(bool(movie['monitored'])), movie['sortTitle'],
movie['year'], alternativeTitles, format, resolution, movie['year'], alternativeTitles, format, resolution,
videoCodec, audioCodec, movie["tmdbId"])) videoCodec, audioCodec, imdbId, movie["tmdbId"]))
else: else:
if movie_default_enabled is True: if movie_default_enabled is True:
movies_to_add.append((movie["title"], movies_to_add.append((movie["title"],
@ -121,7 +124,7 @@ def update_movies():
profile_id_to_language(movie['qualityProfileId']), sceneName, profile_id_to_language(movie['qualityProfileId']), sceneName,
unicode(bool(movie['monitored'])), movie['sortTitle'], unicode(bool(movie['monitored'])), movie['sortTitle'],
movie['year'], alternativeTitles, format, resolution, movie['year'], alternativeTitles, format, resolution,
videoCodec, audioCodec)) videoCodec, audioCodec, imdbId))
else: else:
movies_to_add.append((movie["title"], movies_to_add.append((movie["title"],
movie["path"] + separator + movie['movieFile'][ movie["path"] + separator + movie['movieFile'][
@ -130,7 +133,7 @@ def update_movies():
profile_id_to_language(movie['qualityProfileId']), sceneName, profile_id_to_language(movie['qualityProfileId']), sceneName,
unicode(bool(movie['monitored'])), movie['sortTitle'], unicode(bool(movie['monitored'])), movie['sortTitle'],
movie['year'], alternativeTitles, format, resolution, movie['year'], alternativeTitles, format, resolution,
videoCodec, audioCodec)) videoCodec, audioCodec, imdbId))
else: else:
logging.error( logging.error(
'BAZARR Radarr returned a movie without a file path: ' + movie["path"] + separator + 'BAZARR Radarr returned a movie without a file path: ' + movie["path"] + separator +
@ -141,18 +144,18 @@ def update_movies():
c = db.cursor() c = db.cursor()
updated_result = c.executemany( updated_result = c.executemany(
'''UPDATE table_movies SET title = ?, path = ?, tmdbId = ?, radarrId = ?, overview = ?, poster = ?, fanart = ?, `audio_language` = ?, sceneName = ?, monitored = ?, sortTitle = ?, year = ?, alternativeTitles = ?, format = ?, resolution = ?, video_codec = ?, audio_codec = ? WHERE tmdbid = ?''', '''UPDATE table_movies SET title = ?, path = ?, tmdbId = ?, radarrId = ?, overview = ?, poster = ?, fanart = ?, `audio_language` = ?, sceneName = ?, monitored = ?, sortTitle = ?, year = ?, alternativeTitles = ?, format = ?, resolution = ?, video_codec = ?, audio_codec = ?, imdbId = ? WHERE tmdbid = ?''',
movies_to_update) movies_to_update)
db.commit() db.commit()
if movie_default_enabled is True: if movie_default_enabled is True:
added_result = c.executemany( added_result = c.executemany(
'''INSERT OR IGNORE INTO table_movies(title, path, tmdbId, languages, subtitles,`hearing_impaired`, radarrId, overview, poster, fanart, `audio_language`, sceneName, monitored, sortTitle, year, alternativeTitles, format, resolution, video_codec, audio_codec) VALUES (?,?,?,?,?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''', '''INSERT OR IGNORE INTO table_movies(title, path, tmdbId, languages, subtitles,`hearing_impaired`, radarrId, overview, poster, fanart, `audio_language`, sceneName, monitored, sortTitle, year, alternativeTitles, format, resolution, video_codec, audio_codec, imdbId) VALUES (?,?,?,?,?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
movies_to_add) movies_to_add)
db.commit() db.commit()
else: else:
added_result = c.executemany( added_result = c.executemany(
'''INSERT OR IGNORE INTO table_movies(title, path, tmdbId, languages, subtitles,`hearing_impaired`, radarrId, overview, poster, fanart, `audio_language`, sceneName, monitored, sortTitle, year, alternativeTitles, format, resolution, video_codec, audio_codec) VALUES (?,?,?,(SELECT languages FROM table_movies WHERE tmdbId = ?), '[]',(SELECT `hearing_impaired` FROM table_movies WHERE tmdbId = ?), ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''', '''INSERT OR IGNORE INTO table_movies(title, path, tmdbId, languages, subtitles,`hearing_impaired`, radarrId, overview, poster, fanart, `audio_language`, sceneName, monitored, sortTitle, year, alternativeTitles, format, resolution, video_codec, audio_codec, imdbId) VALUES (?,?,?,(SELECT languages FROM table_movies WHERE tmdbId = ?), '[]',(SELECT `hearing_impaired` FROM table_movies WHERE tmdbId = ?), ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)''',
movies_to_add) movies_to_add)
db.commit() db.commit()

3
bazarr/get_subtitle.py
Unescape View File

@ -708,11 +708,12 @@ def refine_from_db(path, video):
elif isinstance(video, Movie): elif isinstance(video, Movie):
db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30) db = sqlite3.connect(os.path.join(args.config_dir, 'db', 'bazarr.db'), timeout=30)
c = db.cursor() c = db.cursor()
data = c.execute("SELECT title, year, alternativeTitles, format, resolution, video_codec, audio_codec FROM table_movies WHERE path = ?", (path_replace_reverse_movie(path),)).fetchone() data = c.execute("SELECT title, year, alternativeTitles, format, resolution, video_codec, audio_codec, imdbId FROM table_movies WHERE path = ?", (path_replace_reverse_movie(path),)).fetchone()
db.close() db.close()
if data: if data:
video.title = re.sub(r'(\(\d\d\d\d\))' , '', data[0]) video.title = re.sub(r'(\(\d\d\d\d\))' , '', data[0])
if int(data[1]) > 0: video.year = int(data[1]) if int(data[1]) > 0: video.year = int(data[1])
if data[7]: video.imdb_id = data[7]
video.alternative_titles = ast.literal_eval(data[2]) video.alternative_titles = ast.literal_eval(data[2])
if not video.format: if not video.format:
if data[3]: video.format = data[3] if data[3]: video.format = data[3]

1
bazarr/update_db.py
Unescape View File

@ -111,6 +111,7 @@ if os.path.exists(os.path.join(args.config_dir, 'db', 'bazarr.db')):
c.execute('alter table table_movies add column "resolution" "text"') c.execute('alter table table_movies add column "resolution" "text"')
c.execute('alter table table_movies add column "video_codec" "text"') c.execute('alter table table_movies add column "video_codec" "text"')
c.execute('alter table table_movies add column "audio_codec" "text"') c.execute('alter table table_movies add column "audio_codec" "text"')
c.execute('alter table table_movies add column "imdbId" "text"')
db.commit() db.commit()
except: except:
pass pass

Write Preview
Loading…
Cancel
Save