This commit is contained in:
Louis Vézina 2017-11-16 13:42:23 -05:00
parent 2eef8c29e7
commit e1b972c561

View file

@ -51,26 +51,26 @@ def series_download_subtitles(no):
if message is not None: if message is not None:
store_subtitles(path_replace(episode[0])) store_subtitles(path_replace(episode[0]))
history_log(1, no, episode[2], message) history_log(1, no, episode[2], message)
#list_missing_subtitles(no) list_missing_subtitles(no)
def wanted_download_subtitles(path): def wanted_download_subtitles(path):
conn_db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db')) #conn_db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'))
c_db = conn_db.cursor() #c_db = conn_db.cursor()
episodes_details = c_db.execute("SELECT table_episodes.path, table_episodes.missing_subtitles, table_episodes.sonarrEpisodeId, table_episodes.sonarrSeriesId, table_shows.hearing_impaired FROM table_episodes INNER JOIN table_shows on table_shows.sonarrSeriesId = table_episodes.sonarrSeriesId WHERE table_episodes.path = ? AND missing_subtitles != '[]'", (path_replace_reverse(path),)).fetchall() #episodes_details = c_db.execute("SELECT table_episodes.path, table_episodes.missing_subtitles, table_episodes.sonarrEpisodeId, table_episodes.sonarrSeriesId, table_shows.hearing_impaired FROM table_episodes INNER JOIN table_shows on table_shows.sonarrSeriesId = table_episodes.sonarrSeriesId WHERE table_episodes.path = ? AND missing_subtitles != '[]'", (path_replace_reverse(path),)).fetchall()
enabled_providers = c_db.execute("SELECT name FROM table_settings_providers WHERE enabled = 1").fetchall() #enabled_providers = c_db.execute("SELECT name FROM table_settings_providers WHERE enabled = 1").fetchall()
c_db.close() #c_db.close()
providers_list = [] #providers_list = []
for provider in enabled_providers: #for provider in enabled_providers:
providers_list.append(provider[0]) # providers_list.append(provider[0])
for episode in episodes_details: #for episode in episodes_details:
for language in ast.literal_eval(episode[1]): # for language in ast.literal_eval(episode[1]):
message = download_subtitle(path_replace(episode[0]), str(pycountry.languages.lookup(language).alpha_3), episode[4], providers_list) # message = download_subtitle(path_replace(episode[0]), str(pycountry.languages.lookup(language).alpha_3), episode[4], providers_list)
if message is not None: # if message is not None:
store_subtitles(path_replace(episode[0])) # store_subtitles(path_replace(episode[0]))
list_missing_subtitles(episode[3]) # list_missing_subtitles(episode[3])
history_log(1, episode[3], episode[2], message) # history_log(1, episode[3], episode[2], message)
del conn_db del conn_db
del c_db del c_db