mirror of https://github.com/morpheus65535/bazarr
87 lines
2.3 KiB
Python
87 lines
2.3 KiB
Python
# coding=utf-8
|
|
|
|
from __future__ import absolute_import
|
|
import logging
|
|
|
|
from subzero.modification.processors import Processor
|
|
import six
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class StringProcessor(Processor):
|
|
"""
|
|
String replacement processor base
|
|
"""
|
|
|
|
def __init__(self, search, replace, name=None, parent=None, supported=None):
|
|
super(StringProcessor, self).__init__(name=name, supported=supported)
|
|
self.search = search
|
|
self.replace = replace
|
|
|
|
def process(self, content, debug=False, **kwargs):
|
|
return content.replace(self.search, self.replace)
|
|
|
|
|
|
class MultipleLineProcessor(Processor):
|
|
"""
|
|
replaces stuff in whole lines
|
|
|
|
takes a search/replace dict as first argument
|
|
Expects a dictionary in the form of:
|
|
dict = {
|
|
"data": {"old_value": "new_value"}
|
|
}
|
|
"""
|
|
def __init__(self, snr_dict, name=None, parent=None, supported=None):
|
|
super(MultipleLineProcessor, self).__init__(name=name, supported=supported)
|
|
self.snr_dict = snr_dict
|
|
|
|
def process(self, content, debug=False, **kwargs):
|
|
if not self.snr_dict["data"]:
|
|
return content
|
|
|
|
for key, value in six.iteritems(self.snr_dict["data"]):
|
|
if debug and key in content:
|
|
logger.debug(u"Replacing '%s' with '%s' in '%s'", key, value, content)
|
|
|
|
content = content.replace(key, value)
|
|
|
|
return content
|
|
|
|
|
|
class WholeLineProcessor(MultipleLineProcessor):
|
|
def process(self, content, debug=False, **kwargs):
|
|
if not self.snr_dict["data"]:
|
|
return content
|
|
content = content.strip()
|
|
|
|
for key, value in six.iteritems(self.snr_dict["data"]):
|
|
if content == key:
|
|
if debug:
|
|
logger.debug(u"Replacing '%s' with '%s'", key, value)
|
|
|
|
content = value
|
|
break
|
|
|
|
return content
|
|
|
|
|
|
class MultipleWordProcessor(MultipleLineProcessor):
|
|
"""
|
|
replaces words
|
|
|
|
takes a search/replace dict as first argument
|
|
Expects a dictionary in the form of:
|
|
dict = {
|
|
"data": {"old_value": "new_value"}
|
|
}
|
|
"""
|
|
def process(self, content, debug=False, **kwargs):
|
|
words = content.split(u" ")
|
|
new_words = []
|
|
for word in words:
|
|
new_words.append(self.snr_dict.get(word, word))
|
|
|
|
return u" ".join(new_words)
|