mirror of
https://github.com/morpheus65535/bazarr
synced 2024-12-22 07:42:32 +00:00
73 lines
2.7 KiB
Python
73 lines
2.7 KiB
Python
import os
|
|
import enzyme
|
|
import babelfish
|
|
import pycountry
|
|
import sqlite3
|
|
import ast
|
|
|
|
from get_general_settings import *
|
|
|
|
def store_subtitles(file):
|
|
languages = []
|
|
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()
|
|
enabled_languages = c_db.execute("SELECT code2 FROM table_settings_languages WHERE enabled = 1").fetchall()
|
|
|
|
for language in enabled_languages:
|
|
subtitle_path = os.path.splitext(file)[0] + "." + str(language[0]) + ".srt"
|
|
if os.path.isfile(subtitle_path):
|
|
languages.append([str(language[0]),str(path_replace_reverse(subtitle_path))])
|
|
try:
|
|
c_db.execute("UPDATE table_episodes SET subtitles = ? WHERE path = ?", (str(languages), path_replace_reverse(file)))
|
|
conn_db.commit()
|
|
except:
|
|
pass
|
|
c_db.close()
|
|
|
|
return languages
|
|
|
|
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()
|
|
all_path = c_db.execute("SELECT path FROM table_episodes").fetchall()
|
|
c_db.close()
|
|
|
|
for path in all_path:
|
|
print store_subtitles(path_replace(path[0]))
|
|
|
|
#print list_missing_subtitles('/tv/Fear the Walking Dead/Season 3/Fear.The.Walking.Dead.S03E01.CONVERT.720p.WEB.h264-TBS[rarbg].mkv')
|
|
#full_scan_subtitles()
|