mirror
/
bazarr
mirror of https://github.com/morpheus65535/bazarr
1
0
Fork 0
Code Issues Releases Wiki Activity

Added defaults languages migration.

This commit is contained in:
morpheus65535 2021-02-13 09:59:43 -05:00
parent 21a6d4b6fb
commit 9398789fb7
1 changed files with 36 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
bazarr/database.py
View File

@ -168,10 +168,29 @@ def db_upgrade():
"cutoff INTEGER, items TEXT NOT NULL)") "cutoff INTEGER, items TEXT NOT NULL)")
if not lang_table_exist: if not lang_table_exist:
series_default = []
for language in ast.literal_eval(settings.general.serie_default_language):
if settings.general.serie_default_forced == 'Both':
series_default.append([language, 'True', settings.general.serie_default_hi])
series_default.append([language, 'False', settings.general.serie_default_hi])
else:
series_default.append([language, settings.general.serie_default_forced,
settings.general.serie_default_hi])
movies_default = []
for language in ast.literal_eval(settings.general.movie_default_language):
if settings.general.movie_default_forced == 'Both':
movies_default.append([language, 'True', settings.general.movie_default_hi])
movies_default.append([language, 'False', settings.general.movie_default_hi])
else:
movies_default.append([language, settings.general.movie_default_forced,
settings.general.movie_default_hi])
profiles_to_create = database.execute("SELECT DISTINCT languages, hearing_impaired, forced " profiles_to_create = database.execute("SELECT DISTINCT languages, hearing_impaired, forced "
"FROM (SELECT languages, hearing_impaired, forced FROM table_shows " "FROM (SELECT languages, hearing_impaired, forced FROM table_shows "
"UNION ALL SELECT languages, hearing_impaired, forced FROM table_movies) " "UNION ALL SELECT languages, hearing_impaired, forced FROM table_movies) "
"a WHERE languages NOT null and languages NOT IN ('None', '[]')") "a WHERE languages NOT null and languages NOT IN ('None', '[]')")
for profile in profiles_to_create: for profile in profiles_to_create:
profile_items = [] profile_items = []
languages_list = ast.literal_eval(profile['languages']) languages_list = ast.literal_eval(profile['languages'])
@ -184,6 +203,7 @@ def db_upgrade():
else: else:
profile_items.append({'id': i, 'language': language, 'forced': profile['forced'], profile_items.append({'id': i, 'language': language, 'forced': profile['forced'],
'hi': profile['hearing_impaired'], 'audio_exclude': 'False'}) 'hi': profile['hearing_impaired'], 'audio_exclude': 'False'})
# Create profiles # Create profiles
new_profile_name = profile['languages'] + ' (' + profile['hearing_impaired'] + '/' + profile['forced'] + ')' new_profile_name = profile['languages'] + ' (' + profile['hearing_impaired'] + '/' + profile['forced'] + ')'
database.execute("INSERT INTO table_languages_profiles (name, cutoff, items) VALUES(" database.execute("INSERT INTO table_languages_profiles (name, cutoff, items) VALUES("
@ -198,6 +218,22 @@ def db_upgrade():
"forced = ?", (created_profile_id, profile['languages'], profile['hearing_impaired'], "forced = ?", (created_profile_id, profile['languages'], profile['hearing_impaired'],
profile['forced'])) profile['forced']))
# Save new defaults
profile_items_list = []
for item in profile_items:
profile_items_list.append([item['language'], item['forced'], item['hi']])
try:
if created_profile_id and profile_items_list == series_default:
settings.general.serie_default_profile = str(created_profile_id)
except:
pass
try:
if created_profile_id and profile_items_list == movies_default:
settings.general.movie_default_profile = str(created_profile_id)
except:
pass
# null languages, forced and hearing_impaired for all series and movies # null languages, forced and hearing_impaired for all series and movies
database.execute("UPDATE table_shows SET languages = null, forced = null, hearing_impaired = null") database.execute("UPDATE table_shows SET languages = null, forced = null, hearing_impaired = null")
database.execute("UPDATE table_movies SET languages = null, forced = null, hearing_impaired = null") database.execute("UPDATE table_movies SET languages = null, forced = null, hearing_impaired = null")