panni
160 lines
5.2 KiB
Python
160 lines
5.2 KiB
Python
# coding=utf-8
|
|
|
|
import traceback
|
|
|
|
import re
|
|
|
|
import pysubs2
|
|
import logging
|
|
|
|
from pysubs2 import SSAStyle
|
|
from pysubs2.subrip import ms_to_timestamp
|
|
from pysubs2.substation import parse_tags
|
|
from registry import registry
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SubtitleModifications(object):
|
|
debug = False
|
|
language = None
|
|
initialized_mods = {}
|
|
|
|
def __init__(self, debug=False):
|
|
self.debug = debug
|
|
self.initialized_mods = {}
|
|
|
|
def load(self, fn=None, content=None, language=None, encoding="utf-8"):
|
|
"""
|
|
|
|
:param encoding: used for decoding the content when fn is given, not used in case content is given
|
|
:param language: babelfish.Language language of the subtitle
|
|
:param fn: filename
|
|
:param content: unicode
|
|
:return:
|
|
"""
|
|
self.language = language
|
|
self.initialized_mods = {}
|
|
try:
|
|
if fn:
|
|
self.f = pysubs2.load(fn, encoding=encoding)
|
|
elif content:
|
|
self.f = pysubs2.SSAFile.from_string(content)
|
|
except (IOError,
|
|
UnicodeDecodeError,
|
|
pysubs2.exceptions.UnknownFPSError,
|
|
pysubs2.exceptions.UnknownFormatIdentifierError,
|
|
pysubs2.exceptions.FormatAutodetectionError):
|
|
if fn:
|
|
logger.exception("Couldn't load subtitle: %s: %s", fn, traceback.format_exc())
|
|
elif content:
|
|
logger.exception("Couldn't load subtitle: %s", traceback.format_exc())
|
|
|
|
@classmethod
|
|
def parse_identifier(cls, identifier):
|
|
# simple identifier
|
|
if identifier in registry.mods:
|
|
return identifier, {}
|
|
|
|
# identifier with params; identifier(param=value)
|
|
split_args = identifier[identifier.find("(")+1:-1].split(",")
|
|
args = dict((key, value) for key, value in [sub.split("=") for sub in split_args])
|
|
return identifier[:identifier.find("(")], args
|
|
|
|
@classmethod
|
|
def get_mod_class(cls, identifier):
|
|
identifier, args = cls.parse_identifier(identifier)
|
|
return registry.mods[identifier]
|
|
|
|
@classmethod
|
|
def get_mod_signature(cls, identifier, **kwargs):
|
|
return cls.get_mod_class(identifier).get_signature(**kwargs)
|
|
|
|
def modify(self, *mods):
|
|
new_f = []
|
|
|
|
parsed_mods = [SubtitleModifications.parse_identifier(mod) for mod in mods]
|
|
line_mods = []
|
|
non_line_mods = []
|
|
|
|
for identifier, args in parsed_mods:
|
|
if identifier not in registry.mods:
|
|
logger.error("Mod %s not loaded", identifier)
|
|
continue
|
|
|
|
mod_cls = registry.mods[identifier]
|
|
if mod_cls.modifies_whole_file:
|
|
non_line_mods.append((identifier, args))
|
|
else:
|
|
line_mods.append((identifier, args))
|
|
|
|
if identifier not in self.initialized_mods:
|
|
self.initialized_mods[identifier] = mod_cls(self)
|
|
|
|
# apply file mods
|
|
if non_line_mods:
|
|
for identifier, args in non_line_mods:
|
|
mod = self.initialized_mods[identifier]
|
|
mod.modify(None, debug=self.debug, parent=self, **args)
|
|
|
|
# apply line mods
|
|
if line_mods:
|
|
for line in self.f:
|
|
applied_mods = []
|
|
skip_line = False
|
|
for identifier, args in line_mods:
|
|
mod = self.initialized_mods[identifier]
|
|
|
|
# don't bother reapplying exclusive mods multiple times
|
|
if mod.exclusive and identifier in applied_mods:
|
|
continue
|
|
|
|
new_content = mod.modify(line.text, debug=self.debug, parent=self, **args)
|
|
if not new_content:
|
|
if self.debug:
|
|
logger.debug("%s: deleting %s", identifier, line)
|
|
skip_line = True
|
|
break
|
|
|
|
line.text = new_content
|
|
applied_mods.append(identifier)
|
|
if not skip_line:
|
|
new_f.append(line)
|
|
|
|
self.f.events = new_f
|
|
|
|
def to_unicode(self):
|
|
def prepare_text(text, style):
|
|
body = []
|
|
for fragment, sty in parse_tags(text, style, self.f.styles):
|
|
fragment = fragment.replace(ur"\h", u" ")
|
|
fragment = fragment.replace(ur"\n", u"\n")
|
|
fragment = fragment.replace(ur"\N", u"\n")
|
|
if sty.italic: fragment = u"<i>%s</i>" % fragment
|
|
if sty.underline: fragment = u"<u>%s</u>" % fragment
|
|
if sty.strikeout: fragment = u"<s>%s</s>" % fragment
|
|
body.append(fragment)
|
|
|
|
return re.sub(u"\n+", u"\n", u"".join(body).strip())
|
|
|
|
visible_lines = (line for line in self.f if not line.is_comment)
|
|
|
|
out = []
|
|
|
|
for i, line in enumerate(visible_lines, 1):
|
|
start = ms_to_timestamp(line.start)
|
|
end = ms_to_timestamp(line.end)
|
|
text = prepare_text(line.text, self.f.styles.get(line.style, SSAStyle.DEFAULT_STYLE))
|
|
|
|
out.append(u"%d\n" % i)
|
|
out.append(u"%s --> %s\n" % (start, end))
|
|
out.append(u"%s%s" % (text, "\n\n"))
|
|
|
|
return u"".join(out)
|
|
|
|
SubMod = SubtitleModifications
|
|
|
|
|
|
|
|
|