bazarr/get_subtitle.py

82 lines
3.8 KiB
Python
Raw Normal View History

2017-11-21 20:42:18 +00:00
# coding: utf-8
from __future__ import unicode_literals
2017-09-17 00:11:47 +00:00
import os
2017-10-16 23:27:19 +00:00
import sqlite3
import ast
2017-09-17 00:11:47 +00:00
from babelfish import *
from subliminal import *
2017-09-17 13:02:16 +00:00
from pycountry import *
2017-10-16 23:27:19 +00:00
from get_general_settings import *
from list_subtitles import *
from utils import *
2017-09-17 00:11:47 +00:00
# configure the cache
2017-10-20 13:14:30 +00:00
region.configure('dogpile.cache.dbm', arguments={'filename': os.path.join(os.path.dirname(__file__), 'data/cache/cachefile.dbm')})
2017-09-17 00:11:47 +00:00
def download_subtitle(path, language, hi, providers):
2017-11-16 18:39:47 +00:00
video = scan_video(path)
best_subtitles = download_best_subtitles([video], {Language(language)}, providers=providers, hearing_impaired=hi)
try:
best_subtitle = best_subtitles[video][0]
result = save_subtitles(video, [best_subtitle])
downloaded_provider = str(result[0]).strip('<>').split(' ')[0][:-8]
downloaded_language = pycountry.languages.lookup(str(str(result[0]).strip('<>').split(' ')[2].strip('[]'))).name
message = downloaded_language + " subtitles downloaded from " + downloaded_provider + "."
return message
except:
return None
2017-10-16 23:27:19 +00:00
def series_download_subtitles(no):
2017-10-20 12:59:21 +00:00
conn_db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'))
2017-10-16 23:27:19 +00:00
c_db = conn_db.cursor()
2017-11-21 20:42:18 +00:00
episodes_details = c_db.execute("SELECT path, missing_subtitles, sonarrEpisodeId FROM table_episodes WHERE sonarrSeriesId = ?", (no,)).fetchall()
2017-10-16 23:27:19 +00:00
series_details = c_db.execute("SELECT hearing_impaired FROM table_shows WHERE sonarrSeriesId = ?", (no,)).fetchone()
enabled_providers = c_db.execute("SELECT name FROM table_settings_providers WHERE enabled = 1").fetchall()
c_db.close()
providers_list = []
for provider in enabled_providers:
providers_list.append(provider[0])
for episode in episodes_details:
for language in ast.literal_eval(episode[1]):
message = download_subtitle(path_replace(episode[0]), str(pycountry.languages.lookup(language).alpha_3), series_details[0], providers_list)
if message is not None:
store_subtitles(path_replace(episode[0]))
history_log(1, no, episode[2], message)
2017-11-16 18:42:23 +00:00
list_missing_subtitles(no)
2017-10-16 23:27:19 +00:00
def wanted_download_subtitles(path):
2017-11-16 18:46:30 +00:00
conn_db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'))
2017-11-16 19:09:40 +00:00
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()
enabled_providers = c_db.execute("SELECT name FROM table_settings_providers WHERE enabled = 1").fetchall()
c_db.close()
2017-10-16 23:27:19 +00:00
2017-11-16 19:09:40 +00:00
providers_list = []
for provider in enabled_providers:
providers_list.append(provider[0])
2017-10-16 23:27:19 +00:00
2017-11-16 19:09:40 +00:00
for episode in episodes_details:
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)
if message is not None:
store_subtitles(path_replace(episode[0]))
list_missing_subtitles(episode[3])
history_log(1, episode[3], episode[2], message)
2017-11-16 17:04:20 +00:00
2017-10-23 03:00:11 +00:00
def wanted_search_missing_subtitles():
db = sqlite3.connect(os.path.join(os.path.dirname(__file__), 'data/db/bazarr.db'))
db.create_function("path_substitution", 1, path_replace)
c = db.cursor()
c.execute("SELECT path_substitution(path) FROM table_episodes WHERE table_episodes.missing_subtitles != '[]'")
data = c.fetchall()
c.close()
for episode in data:
2017-11-21 20:42:18 +00:00
wanted_download_subtitles(episode[0])