bazarr/get_subtitle.py

20 lines
792 B
Python
Raw Normal View History

2017-09-17 00:11:47 +00:00
import os
from babelfish import *
from subliminal import *
2017-09-17 13:02:16 +00:00
from pycountry import *
2017-09-17 00:11:47 +00:00
# configure the cache
region.configure('dogpile.cache.dbm', arguments={'filename': 'cachefile.dbm'})
def download_subtitle(path, language, hi, providers):
video = scan_video(path)
best_subtitles = download_best_subtitles([video], {Language(language)}, providers=providers, hearing_impaired=hi)
best_subtitle = best_subtitles[video][0]
2017-09-17 13:02:16 +00:00
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 + "."
2017-09-17 00:11:47 +00:00
2017-09-17 13:02:16 +00:00
return message