2019-04-23 12:35:50 +00:00
|
|
|
import enzyme
|
2019-10-20 12:03:55 +00:00
|
|
|
from enzyme.exceptions import MalformedMKVError
|
2019-04-23 12:35:50 +00:00
|
|
|
import logging
|
2019-04-23 20:19:25 +00:00
|
|
|
import os
|
2020-03-18 19:33:54 +00:00
|
|
|
from knowit import api
|
2019-04-23 12:35:50 +00:00
|
|
|
|
2019-04-23 20:19:25 +00:00
|
|
|
|
2019-04-23 12:35:50 +00:00
|
|
|
class EmbeddedSubsReader:
|
|
|
|
def __init__(self):
|
2020-07-19 20:02:38 +00:00
|
|
|
self.ffprobe = None
|
2019-04-23 12:35:50 +00:00
|
|
|
|
|
|
|
def list_languages(self, file):
|
2020-07-19 20:02:38 +00:00
|
|
|
from utils import get_binary
|
|
|
|
self.ffprobe = get_binary("ffprobe")
|
|
|
|
|
2019-04-27 12:13:47 +00:00
|
|
|
subtitles_list = []
|
|
|
|
|
2019-08-26 02:11:30 +00:00
|
|
|
if self.ffprobe:
|
2020-03-18 19:33:54 +00:00
|
|
|
api.initialize({'provider': 'ffmpeg', 'ffmpeg': self.ffprobe})
|
|
|
|
data = api.know(file)
|
|
|
|
|
|
|
|
if 'subtitle' in data:
|
|
|
|
for detected_language in data['subtitle']:
|
2020-03-22 13:23:06 +00:00
|
|
|
if 'language' in detected_language:
|
|
|
|
language = detected_language['language'].alpha3
|
|
|
|
forced = detected_language['forced'] if 'forced' in detected_language else None
|
|
|
|
codec = detected_language['format'] if 'format' in detected_language else None
|
2020-03-18 19:33:54 +00:00
|
|
|
subtitles_list.append([language, forced, codec])
|
|
|
|
else:
|
|
|
|
continue
|
2019-04-27 12:13:47 +00:00
|
|
|
else:
|
2019-08-26 02:11:30 +00:00
|
|
|
if os.path.splitext(file)[1] == '.mkv':
|
|
|
|
with open(file, 'rb') as f:
|
2019-10-20 12:03:55 +00:00
|
|
|
try:
|
|
|
|
mkv = enzyme.MKV(f)
|
|
|
|
except MalformedMKVError:
|
|
|
|
logging.error('BAZARR cannot analyze this MKV with our built-in MKV parser, you should install ffmpeg: ' + file)
|
2019-11-04 11:34:33 +00:00
|
|
|
else:
|
|
|
|
for subtitle_track in mkv.subtitle_tracks:
|
|
|
|
subtitles_list.append([subtitle_track.language, subtitle_track.forced, subtitle_track.codec_id])
|
2019-04-27 12:13:47 +00:00
|
|
|
|
|
|
|
return subtitles_list
|
|
|
|
|
2019-04-23 12:35:50 +00:00
|
|
|
|
2019-08-26 02:11:30 +00:00
|
|
|
embedded_subs_reader = EmbeddedSubsReader()
|