mirror of
https://github.com/clinton-hall/nzbToMedia.git
synced 2024-11-14 09:30:21 -08:00
108 lines
4.3 KiB
Python
108 lines
4.3 KiB
Python
from __future__ import (
|
|
absolute_import,
|
|
division,
|
|
print_function,
|
|
unicode_literals,
|
|
)
|
|
|
|
from babelfish import Language
|
|
import subliminal
|
|
|
|
import core
|
|
from core import logger
|
|
|
|
import os
|
|
import re
|
|
|
|
for provider in subliminal.provider_manager.internal_extensions:
|
|
if provider not in [str(x) for x in subliminal.provider_manager.list_entry_points()]:
|
|
subliminal.provider_manager.register(str(provider))
|
|
|
|
|
|
def import_subs(filename):
|
|
if not core.GETSUBS:
|
|
return
|
|
try:
|
|
subliminal.region.configure('dogpile.cache.dbm', arguments={'filename': 'cachefile.dbm'})
|
|
except Exception:
|
|
pass
|
|
|
|
languages = set()
|
|
for item in core.SLANGUAGES:
|
|
try:
|
|
languages.add(Language(item))
|
|
except Exception:
|
|
pass
|
|
if not languages:
|
|
return
|
|
|
|
logger.info('Attempting to download subtitles for {0}'.format(filename), 'SUBTITLES')
|
|
try:
|
|
video = subliminal.scan_video(filename)
|
|
subtitles = subliminal.download_best_subtitles({video}, languages)
|
|
subliminal.save_subtitles(video, subtitles[video])
|
|
|
|
for subtitle in subtitles[video]:
|
|
subtitle_path = subliminal.subtitle.get_subtitle_path(video.name, subtitle.language)
|
|
os.chmod(subtitle_path, 0o644)
|
|
except Exception as e:
|
|
logger.error('Failed to download subtitles for {0} due to: {1}'.format(filename, e), 'SUBTITLES')
|
|
|
|
def rename_subs(path):
|
|
filepaths = []
|
|
sub_ext = ['.srt', '.sub', '.idx']
|
|
vidfiles = core.list_media_files(path, media=True, audio=False, meta=False, archives=False)
|
|
if not vidfiles or len(vidfiles) > 1: # If there is more than 1 video file, or no video files, we can't rename subs.
|
|
return
|
|
name = os.path.splitext(os.path.split(vidfiles[0])[1])[0]
|
|
for directory, _, filenames in os.walk(path):
|
|
for filename in filenames:
|
|
filepaths.extend([os.path.join(directory, filename)])
|
|
subfiles = [item for item in filepaths if os.path.splitext(item)[1] in sub_ext]
|
|
subfiles.sort() #This should sort subtitle names by language (alpha) and Number (where multiple)
|
|
renamed = []
|
|
for sub in subfiles:
|
|
subname, ext = os.path.splitext(os.path.basename(sub))
|
|
if name in subname: # The sub file name already includes the video name.
|
|
continue
|
|
words = re.findall('[a-zA-Z]+',str(subname)) # find whole words in string
|
|
# parse the words for language descriptors.
|
|
lan = None
|
|
for word in words:
|
|
try:
|
|
if len(word) == 2:
|
|
lan = Language.fromalpha2(word.lower())
|
|
elif len(word) == 3:
|
|
lan = Language(word.lower())
|
|
elif len(word) > 3:
|
|
lan = Language.fromname(word.lower())
|
|
if lan:
|
|
break
|
|
except: #if we didn't find a language, try next word.
|
|
continue
|
|
# rename the sub file as name.lan.ext
|
|
if not lan:
|
|
# could call ffprobe to parse the sub information and get language if lan unknown here.
|
|
new_sub_name = name
|
|
else:
|
|
new_sub_name = '{name}.{lan}'.format(name=name, lan=str(lan))
|
|
new_sub = os.path.join(directory, new_sub_name) # full path and name less ext
|
|
if '{new_sub}{ext}'.format(new_sub=new_sub, ext=ext) in renamed: # If duplicate names, add unique number before ext.
|
|
for i in range(1,len(renamed)+1):
|
|
if '{new_sub}.{i}{ext}'.format(new_sub=new_sub, i=i, ext=ext) in renamed:
|
|
continue
|
|
new_sub = '{new_sub}.{i}'.format(new_sub=new_sub, i=i)
|
|
break
|
|
new_sub = '{new_sub}{ext}'.format(new_sub=new_sub, ext=ext) # add extension now
|
|
if os.path.isfile(new_sub): # Don't copy over existing - final check.
|
|
logger.debug('Unable to rename sub file {old} as destination {new} already exists'.format(old=sub, new=new_sub))
|
|
continue
|
|
logger.debug('Renaming sub file from {old} to {new}'.format
|
|
(old=sub, new=new_sub))
|
|
renamed.append(new_sub)
|
|
try:
|
|
os.rename(sub, new_sub)
|
|
except Exception as error:
|
|
logger.error('Unable to rename sub file due to: {error}'.format(error=error))
|
|
return
|