91 lines
2.8 KiB
Python
91 lines
2.8 KiB
Python
import codecs
|
|
import os
|
|
import re
|
|
import sys
|
|
|
|
import pysrt
|
|
|
|
|
|
class sdh_remover:
|
|
def __init__(self,):
|
|
self.__replace__ = "empty_line"
|
|
self.content = []
|
|
|
|
def cleanLine(self, line, regex):
|
|
line = re.sub("</i>", "", line)
|
|
line = re.sub("<i>", "", line)
|
|
if re.search(r"\[(.*)?\n(.*)?\]", line):
|
|
line = re.sub(
|
|
re.search(r"\[(.*)?\n(.*)?\]", line).group(), self.__replace__, line
|
|
)
|
|
|
|
if re.search(r"\((.*)?\n(.*)?\)", line):
|
|
line = re.sub(
|
|
re.search(r"\((.*)?\n(.*)?\)", line).group(), self.__replace__, line
|
|
)
|
|
|
|
try:
|
|
# is it inside a markup tag?
|
|
match = regex.match(line).group(1)
|
|
tag = re.compile("(<[A-z]+[^>]*>)").match(match).group(1)
|
|
line = re.sub(match, tag + self.__replace__, line)
|
|
except:
|
|
try:
|
|
line = re.sub(regex, self.__replace__, line)
|
|
except:
|
|
pass
|
|
return line
|
|
|
|
def _save(self, Output):
|
|
|
|
file = codecs.open(Output, "w", encoding="utf-8")
|
|
|
|
for idx, text in enumerate(self.content, start=1):
|
|
file.write(
|
|
"{}\n{} --> {}\n{}\n\n".format(
|
|
str(idx), text["start"], text["end"], text["text"].strip(),
|
|
)
|
|
)
|
|
|
|
file.close()
|
|
|
|
def clean(self):
|
|
if not self.content == []:
|
|
temp = self.content
|
|
self.content = []
|
|
|
|
for text in temp:
|
|
if text["text"].strip() == self.__replace__:
|
|
continue
|
|
text.update({"text": re.sub(self.__replace__, "", text["text"])})
|
|
|
|
if not text["text"].strip() == "":
|
|
self.content.append(text)
|
|
|
|
return
|
|
|
|
def noHI(self, Input=None, Output=None, content=None):
|
|
|
|
srt = pysrt.open(Input, encoding="utf-8")
|
|
for idx, line in enumerate(srt, start=1):
|
|
number = str(idx)
|
|
start = line.start
|
|
end = line.end
|
|
text = line.text
|
|
|
|
text = self.cleanLine(text, re.compile(r"(\[(.+)?\]|\[(.+)?|^(.+)?\])"))
|
|
text = self.cleanLine(text, re.compile(r"(\((.+)?\)|\((.+)?|^(.+)?\))"))
|
|
text = self.cleanLine(text, re.compile(r"(\[(.+)?\]|\[(.+)?|^(.+)?\])"))
|
|
text = self.cleanLine(
|
|
text,
|
|
re.compile(r"([♩♪♫♭♮♯]+(.+)?[♩♪♫♭♮♯]+|[♩♪♫♭♮♯]+(.+)?|^(.+)?[♩♪♫♭♮♯]+)"),
|
|
)
|
|
text = self.cleanLine(text, re.compile(r"(<font[^>]*>)|(<\/font>)"))
|
|
|
|
self.content.append(
|
|
{"number": number, "start": start, "end": end, "text": text,}
|
|
)
|
|
|
|
self.clean()
|
|
self._save(Output)
|