mirror of https://github.com/morpheus65535/bazarr
110 lines
3.4 KiB
Python
110 lines
3.4 KiB
Python
# coding=utf-8
|
|
import re
|
|
import logging
|
|
|
|
from subzero.modification.processors.re_processor import ReProcessor, NReProcessor
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class SubtitleModification(object):
|
|
identifier = None
|
|
description = None
|
|
long_description = None
|
|
exclusive = False
|
|
advanced = False # has parameters
|
|
args_mergeable = False
|
|
order = None
|
|
modifies_whole_file = False # operates on the whole file, not individual entries
|
|
apply_last = False
|
|
only_uppercase = False
|
|
pre_processors = []
|
|
processors = []
|
|
post_processors = []
|
|
languages = []
|
|
|
|
def __init__(self, parent):
|
|
return
|
|
|
|
def _process(self, content, processors, debug=False, parent=None, index=None, **kwargs):
|
|
if not content:
|
|
return
|
|
|
|
# processors may be a list or a callable
|
|
#if callable(processors):
|
|
# _processors = processors()
|
|
#else:
|
|
# _processors = processors
|
|
_processors = processors
|
|
|
|
new_content = content
|
|
for processor in _processors:
|
|
if not processor.supported(parent):
|
|
if debug and processor.enabled:
|
|
logger.debug("Processor not supported, skipping: %s", processor.name)
|
|
processor.enabled = False
|
|
continue
|
|
|
|
old_content = new_content
|
|
new_content = processor.process(new_content, debug=debug, **kwargs)
|
|
if not new_content:
|
|
if debug:
|
|
logger.debug("Processor returned empty line: %s", processor.name)
|
|
break
|
|
if debug:
|
|
if old_content == new_content:
|
|
continue
|
|
logger.debug("%d: %s: %s -> %s", index, processor.name, repr(old_content), repr(new_content))
|
|
|
|
return new_content
|
|
|
|
def pre_process(self, content, debug=False, parent=None, **kwargs):
|
|
return self._process(content, self.pre_processors, debug=debug, parent=parent, **kwargs)
|
|
|
|
def process(self, content, debug=False, parent=None, **kwargs):
|
|
return self._process(content, self.processors, debug=debug, parent=parent, **kwargs)
|
|
|
|
def post_process(self, content, debug=False, parent=None, **kwargs):
|
|
return self._process(content, self.post_processors, debug=debug, parent=parent, **kwargs)
|
|
|
|
def modify(self, content, debug=False, parent=None, **kwargs):
|
|
if not content:
|
|
return
|
|
|
|
new_content = content
|
|
for method in ("pre_process", "process", "post_process"):
|
|
if not new_content:
|
|
return
|
|
new_content = getattr(self, method)(new_content, debug=debug, parent=parent, **kwargs)
|
|
|
|
return new_content
|
|
|
|
@classmethod
|
|
def get_signature(cls, **kwargs):
|
|
string_args = ",".join(["%s=%s" % (key, value) for key, value in kwargs.iteritems()])
|
|
return "%s(%s)" % (cls.identifier, string_args)
|
|
|
|
@classmethod
|
|
def merge_args(cls, args1, args2):
|
|
raise NotImplementedError
|
|
|
|
|
|
class SubtitleTextModification(SubtitleModification):
|
|
pass
|
|
|
|
|
|
TAG = ur"(?:\s*{\\[iusb][0-1]}\s*)*"
|
|
EMPTY_TAG_PROCESSOR = ReProcessor(re.compile(r'({\\\w1})[\s.,-_!?]*({\\\w0})'), "", name="empty_tag")
|
|
|
|
empty_line_post_processors = [
|
|
# empty tag
|
|
EMPTY_TAG_PROCESSOR,
|
|
|
|
# empty line (needed?)
|
|
NReProcessor(re.compile(r'^[\s-]+$'), "", name="empty_line"),
|
|
]
|
|
|
|
|
|
class EmptyEntryError(Exception):
|
|
pass
|