mirror of https://github.com/morpheus65535/bazarr
102 lines
3.5 KiB
Python
102 lines
3.5 KiB
Python
import os
|
|
import enzyme
|
|
import babelfish
|
|
from subliminal import *
|
|
import pycountry
|
|
import sqlite3
|
|
import ast
|
|
|
|
from get_general_settings import *
|
|
|
|
def list_subtitles(file):
|
|
languages = []
|
|
actual_subtitles = []
|
|
if os.path.exists(file):
|
|
if os.path.splitext(file)[1] == '.mkv':
|
|
try:
|
|
with open(file, 'rb') as f:
|
|
mkv = enzyme.MKV(f)
|
|
|
|
for subtitle_track in mkv.subtitle_tracks:
|
|
try:
|
|
languages.append([str(pycountry.languages.lookup(subtitle_track.language).alpha_2),None])
|
|
except:
|
|
pass
|
|
except:
|
|
pass
|
|
|
|
subtitles = core.search_external_subtitles(file)
|
|
|
|
for subtitle, language in subtitles.iteritems():
|
|
actual_subtitles.append([str(language), path_replace_reverse(os.path.join(os.path.dirname(file), subtitle))])
|
|
|
|
return actual_subtitles
|
|
|
|
def store_subtitles(file):
|
|
languages = []
|
|
actual_subtitles = []
|
|
if os.path.exists(file):
|
|
if os.path.splitext(file)[1] == '.mkv':
|
|
try:
|
|
with open(file, 'rb') as f:
|
|
mkv = enzyme.MKV(f)
|
|
|
|
for subtitle_track in mkv.subtitle_tracks:
|
|
try:
|
|
languages.append([str(pycountry.languages.lookup(subtitle_track.language).alpha_2),None])
|
|
except:
|
|
pass
|
|
except:
|
|
pass
|
|
|
|
conn_db = sqlite3.connect('bazarr.db')
|
|
c_db = conn_db.cursor()
|
|
|
|
subtitles = core.search_external_subtitles(file)
|
|
|
|
for subtitle, language in subtitles.iteritems():
|
|
actual_subtitles.append([str(language), path_replace_reverse(os.path.join(os.path.dirname(file), subtitle))])
|
|
try:
|
|
c_db.execute("UPDATE table_episodes SET subtitles = ? WHERE path = ?", (str(actual_subtitles), path_replace_reverse(file)))
|
|
conn_db.commit()
|
|
except:
|
|
pass
|
|
c_db.close()
|
|
|
|
return actual_subtitles
|
|
|
|
def list_missing_subtitles(file):
|
|
conn_db = sqlite3.connect('bazarr.db')
|
|
c_db = conn_db.cursor()
|
|
actual_subtitles_long = c_db.execute("SELECT sonarrSeriesId, subtitles FROM table_episodes WHERE path = ?", (file,)).fetchone()
|
|
desired_subtitles = c_db.execute("SELECT languages FROM table_shows WHERE sonarrSeriesId = ?", (actual_subtitles_long[0],)).fetchone()
|
|
c_db.close()
|
|
missing_subtitles = []
|
|
actual_subtitles = []
|
|
if desired_subtitles[0] == "None":
|
|
pass
|
|
else:
|
|
actual_subtitles_long = ast.literal_eval(actual_subtitles_long[1])
|
|
for actual_subtitle in actual_subtitles_long:
|
|
actual_subtitles.append(actual_subtitle[0])
|
|
|
|
desired_subtitles = ast.literal_eval(desired_subtitles[0])
|
|
|
|
missing_subtitles = (list(set(desired_subtitles) - set(actual_subtitles)))
|
|
return str(missing_subtitles)
|
|
|
|
def full_scan_subtitles():
|
|
conn_db = sqlite3.connect('bazarr.db')
|
|
c_db = conn_db.cursor()
|
|
c_db.execute("SELECT path FROM table_episodes").fetchall()
|
|
c_db.close()
|
|
|
|
def new_scan_subtitles():
|
|
conn_db = sqlite3.connect('bazarr.db')
|
|
c_db = conn_db.cursor()
|
|
c_db.execute("SELECT path FROM table_episodes WHERE subtitles is null").fetchall()
|
|
c_db.close()
|
|
|
|
#full_scan_subtitles()
|
|
#new_scan_subtitles()
|