mirror of
https://github.com/spaam/svtplay-dl.git
synced 2024-11-24 12:15:40 +01:00
524 lines
17 KiB
Python
524 lines
17 KiB
Python
import binascii
|
|
import json
|
|
import logging
|
|
import re
|
|
import xml.etree.ElementTree as ET
|
|
from io import StringIO
|
|
|
|
from cryptography.hazmat.backends import default_backend
|
|
from cryptography.hazmat.primitives.ciphers import algorithms
|
|
from cryptography.hazmat.primitives.ciphers import Cipher
|
|
from cryptography.hazmat.primitives.ciphers import modes
|
|
from requests import __build__ as requests_version
|
|
from svtplay_dl.fetcher.m3u8 import M3U8
|
|
from svtplay_dl.utils.fetcher import filter_files
|
|
from svtplay_dl.utils.http import get_full_url
|
|
from svtplay_dl.utils.http import HTTP
|
|
from svtplay_dl.utils.output import find_dupes
|
|
from svtplay_dl.utils.output import formatname
|
|
from svtplay_dl.utils.text import decode_html_entities
|
|
|
|
|
|
def subtitle_probe(config, url, **kwargs):
|
|
httpobject = kwargs.get("httpobject", None)
|
|
if httpobject:
|
|
http = httpobject
|
|
else:
|
|
http = HTTP(config)
|
|
subdata = http.request("get", url, cookies=kwargs.get("cookies", None))
|
|
|
|
if subdata.text.startswith("WEBVTT"):
|
|
yield subtitle(config, "wrst", url, **kwargs)
|
|
elif subdata.text.startswith("#EXTM3U"):
|
|
m3u8 = M3U8(subdata.text)
|
|
yield subtitle(config, "wrstsegment", url, **kwargs, m3u8=m3u8)
|
|
elif "<?xml" in subdata.text:
|
|
xmldata = ET.fromstring(subdata.text)
|
|
if xmldata.tag.endswith("MPD"):
|
|
data = http.get(kwargs.get("files")[0]).content
|
|
if data.find(b"ftyp") > 0:
|
|
yield subtitle(config, "stpp", url, **kwargs)
|
|
elif data.startswith(b"WEBVTT"):
|
|
yield subtitle(config, "wrst", kwargs.get("files")[0], **kwargs)
|
|
elif xmldata.tag.endswith("tt"):
|
|
yield subtitle(config, "tt", url, **kwargs)
|
|
|
|
|
|
class subtitle:
|
|
def __init__(self, config, subtype, url, **kwargs):
|
|
self.url = url
|
|
self.subtitle = None
|
|
self.config = config
|
|
self.subtype = subtype
|
|
self.http = HTTP(config)
|
|
self.subfix = kwargs.get("subfix", None)
|
|
self.bom = False
|
|
self.output = kwargs.pop("output", None)
|
|
self.kwargs = kwargs
|
|
|
|
def __repr__(self):
|
|
return f"<Subtitle(type={self.subtype}, url={self.url}>"
|
|
|
|
def download(self):
|
|
output_ext = "srt"
|
|
if self.config.get("get_raw_subtitles"):
|
|
output_ext = self.subtype
|
|
|
|
if self.subfix and self.config.get("get_all_subtitles"):
|
|
self.output["ext"] = f"{self.subfix}.{output_ext}"
|
|
else:
|
|
self.output["ext"] = output_ext
|
|
|
|
subdata = self.http.request("get", self.url)
|
|
if subdata.status_code != 200:
|
|
logging.warning("Can't download subtitle file")
|
|
return
|
|
|
|
data = None
|
|
if "mtgx" in self.url and subdata.content[:3] == b"\xef\xbb\xbf":
|
|
subdata.encoding = "utf-8"
|
|
self.bom = True
|
|
|
|
if self.subtype == "tt":
|
|
data = self.tt(subdata)
|
|
if self.subtype == "json":
|
|
data = self.json(subdata)
|
|
if self.subtype == "sami":
|
|
data = self.sami(subdata)
|
|
if self.subtype == "smi":
|
|
data = self.smi(subdata)
|
|
if self.subtype == "wrst":
|
|
if "tv4play" in self.url and subdata.content[:3] == b"\xef\xbb\xbf":
|
|
self.bom = True
|
|
subdata.encoding = subdata.apparent_encoding
|
|
data = self.wrst(subdata)
|
|
if self.subtype == "wrstsegment":
|
|
data = self.wrstsegment(subdata)
|
|
if self.subtype == "raw":
|
|
data = self.raw(subdata)
|
|
if self.subtype == "stpp":
|
|
data = self.stpp(subdata)
|
|
|
|
if self.config.get("get_raw_subtitles"):
|
|
data = self.raw(subdata)
|
|
|
|
if len(data) > 0:
|
|
dupe, fileame = find_dupes(self.output, self.config, False)
|
|
if dupe and not self.config.get("force_subtitle"):
|
|
logging.warning("File (%s) already exists. Use --force-subtitle to overwrite", fileame.name)
|
|
return
|
|
self.save_file(data)
|
|
|
|
def save_file(self, data):
|
|
filename = formatname(self.output, self.config)
|
|
with open(filename, "w", encoding="utf-8") as file_d:
|
|
file_d.write(data)
|
|
|
|
def raw(self, subdata):
|
|
return subdata.text
|
|
|
|
def tt(self, subdata):
|
|
i = 1
|
|
subs = subdata.text
|
|
return self._tt(subs, i)
|
|
|
|
def _tt(self, subs, i):
|
|
data = ""
|
|
subdata = re.sub(' xmlns="[^"]+"', "", subs, count=1)
|
|
tree = ET.XML(subdata)
|
|
xml = tree.find("body")
|
|
if not xml:
|
|
return data
|
|
xml = xml.find("div")
|
|
if not xml:
|
|
return data
|
|
plist = list(xml.findall("p"))
|
|
for node in plist:
|
|
tag = norm(node.tag)
|
|
if tag in ("p", "span"):
|
|
begin = node.attrib["begin"]
|
|
if not ("dur" in node.attrib):
|
|
if "end" not in node.attrib:
|
|
duration = node.attrib["duration"]
|
|
else:
|
|
duration = node.attrib["dur"]
|
|
if not ("end" in node.attrib):
|
|
begin2 = begin.split(":")
|
|
duration2 = duration.split(":")
|
|
try:
|
|
sec = float(begin2[2]) + float(duration2[2])
|
|
except ValueError:
|
|
sec = 0.000
|
|
end = f"{int(begin2[0]):02d}:{int(begin2[1]):02d}:{sec:06.3f}"
|
|
else:
|
|
end = node.attrib["end"]
|
|
data += f"{i}\n{begin.replace('.', ',')} --> {end.replace('.', ',')}\n"
|
|
data = tt_text(node, data)
|
|
data += "\n"
|
|
i += 1
|
|
|
|
return data
|
|
|
|
def json(self, subdata):
|
|
data = json.loads(subdata.text)
|
|
number = 1
|
|
subs = ""
|
|
for i in data:
|
|
subs += f"{number}\n{timestr(int(i['startMillis']))} --> {timestr(int(i['endMillis']))}\n"
|
|
subs += f"{i['text']}\n\n"
|
|
number += 1
|
|
|
|
return subs
|
|
|
|
def sami(self, subdata):
|
|
text = subdata.text
|
|
text = re.sub(r"&", "&", text)
|
|
tree = ET.fromstring(text)
|
|
allsubs = tree.findall(".//Subtitle")
|
|
subs = ""
|
|
increase = 0
|
|
for sub in allsubs:
|
|
try:
|
|
number = int(sub.attrib["SpotNumber"])
|
|
except ValueError:
|
|
number = int(re.search(r"(\d+)", sub.attrib["SpotNumber"]).group(1))
|
|
increase += 1
|
|
n = number + increase
|
|
|
|
texts = sub.findall(".//Text")
|
|
all = ""
|
|
for text in texts:
|
|
line = ""
|
|
for txt in text.itertext():
|
|
line += f"{txt}"
|
|
all += f"{decode_html_entities(line.lstrip())}\n"
|
|
subs += f"{n}\n{timecolon(sub.attrib['TimeIn'])} --> {timecolon(sub.attrib['TimeOut'])}\n{all}\n"
|
|
subs = re.sub("&", r"&", subs)
|
|
return subs
|
|
|
|
def smi(self, subdata):
|
|
if requests_version < 0x20300:
|
|
subdata = subdata.content.decode("latin")
|
|
else:
|
|
subdata.encoding = "ISO-8859-1"
|
|
subdata = subdata.text
|
|
ssubdata = StringIO(subdata)
|
|
timea = 0
|
|
number = 1
|
|
data = None
|
|
subs = ""
|
|
TAG_RE = re.compile(r"<(?!\/?i).*?>")
|
|
bad_char = re.compile(r"\x96")
|
|
for i in ssubdata.readlines():
|
|
i = i.rstrip()
|
|
sync = re.search(r"<SYNC Start=(\d+)>", i)
|
|
if sync:
|
|
if int(sync.group(1)) != int(timea):
|
|
if data and data != " ":
|
|
subs += f"{number}\n{timestr(timea)} --> {timestr(sync.group(1))}\n"
|
|
text = decode_html_entities("%s\n" % TAG_RE.sub("", data.replace("<br>", "\n")))
|
|
if text[len(text) - 2] != "\n":
|
|
text += "\n"
|
|
subs += text
|
|
number += 1
|
|
timea = sync.group(1)
|
|
text = re.search("<P Class=SVCC>(.*)", i)
|
|
if text:
|
|
data = text.group(1)
|
|
recomp = re.compile(r"\r")
|
|
text = bad_char.sub("-", recomp.sub("", subs))
|
|
return text
|
|
|
|
def wrst(self, subdata):
|
|
return self._wrst(subdata.text)
|
|
|
|
def _wrst(self, data):
|
|
ssubdata = StringIO(data)
|
|
srt = ""
|
|
subtract = False
|
|
number_b = 1
|
|
number = 0
|
|
block = 0
|
|
subnr = False
|
|
cuetime = False
|
|
|
|
for i in ssubdata.readlines():
|
|
match = re.search(r"^[\r\n]+", i)
|
|
match2 = re.search(r"([\d:\.]+ --> [\d:\.]+)", i)
|
|
match3 = re.search(r"^(\d+)\s", i)
|
|
if match and number_b == 1 and self.bom:
|
|
continue
|
|
elif match and number_b > 1:
|
|
block = 0
|
|
srt += "\n"
|
|
cuetime = False
|
|
elif match2:
|
|
cuetime = True
|
|
if not subnr:
|
|
srt += f"{number_b}\n"
|
|
matchx = re.search(r"(?P<h1>\d+):(?P<m1>\d+):(?P<s1>[\d\.]+) --> (?P<h2>\d+):(?P<m2>\d+):(?P<s2>[\d\.]+)", i)
|
|
if matchx:
|
|
hour1 = int(matchx.group("h1"))
|
|
hour2 = int(matchx.group("h2"))
|
|
if int(number) == 1:
|
|
if hour1 > 9:
|
|
subtract = True
|
|
if subtract:
|
|
hour1 -= 10
|
|
hour2 -= 10
|
|
else:
|
|
matchx = re.search(r"(?P<m1>\d+):(?P<s1>[\d\.]+) --> (?P<m2>\d+):(?P<s2>[\d\.]+)", i)
|
|
hour1 = 0
|
|
hour2 = 0
|
|
time = (
|
|
f"{hour1:02d}:{matchx.group('m1')}:{matchx.group('s1').replace('.', ',')} --> "
|
|
f"{hour2:02d}:{matchx.group('m2')}:{matchx.group('s2').replace('.', ',')}\n"
|
|
)
|
|
srt += time
|
|
block = 1
|
|
subnr = False
|
|
number_b += 1
|
|
elif match3 and block == 0:
|
|
number = match3.group(1)
|
|
srt += f"{number}\n"
|
|
subnr = True
|
|
else:
|
|
if not cuetime:
|
|
continue
|
|
sub = _wsrt_colors(self.config.get("convert_subtitle_colors"), i)
|
|
srt += sub.strip()
|
|
srt += "\n"
|
|
srt = decode_html_entities(srt)
|
|
return srt
|
|
|
|
def wrstsegment(self, subdata):
|
|
pretext = []
|
|
if self.kwargs.get("filter", False):
|
|
self.kwargs["m3u8"] = filter_files(self.kwargs["m3u8"])
|
|
|
|
for _, i in enumerate(self.kwargs["m3u8"].media_segment):
|
|
itemurl = get_full_url(i["URI"], self.url)
|
|
cont = self.http.get(itemurl)
|
|
if self.kwargs["m3u8"].encrypted:
|
|
keyurl = get_full_url(i["EXT-X-KEY"]["URI"], self.url)
|
|
key = self.http.request("get", keyurl).content
|
|
iv = binascii.unhexlify(i["EXT-X-KEY"]["IV"][2:].zfill(32)) if "IV" in i["EXT-X-KEY"] else b"\x00" * 16
|
|
backend = default_backend()
|
|
cipher = Cipher(algorithms.AES(key), modes.CBC(iv), backend=backend)
|
|
decryptor = cipher.decryptor()
|
|
if decryptor:
|
|
data = decryptor.update(cont.content).decode("utf-8")
|
|
else:
|
|
cont.encoding = "utf-8"
|
|
data = cont.text
|
|
pretext.append(data)
|
|
return _wrstsegments(pretext, self.config.get("convert_subtitle_colors"))
|
|
|
|
def stpp(self, subdata):
|
|
nr = 1
|
|
entries = []
|
|
|
|
for i in self.kwargs["files"]:
|
|
res = self.http.get(i)
|
|
start = res.content.find(b"mdat") + 4
|
|
if start > 3:
|
|
_data = self._tt(res.content[start:].decode(), nr)
|
|
if _data:
|
|
entries.append(_data.split("\n\n"))
|
|
nr += 1
|
|
|
|
new_entries = []
|
|
for entry in entries:
|
|
for i in entry:
|
|
if i:
|
|
new_entries.append(i.split("\n"))
|
|
|
|
entries = new_entries
|
|
changed = True
|
|
while changed:
|
|
changed, entries = _resolv(entries)
|
|
|
|
nr = 1
|
|
data = ""
|
|
for entry in entries:
|
|
for item in entry:
|
|
data += f"{item}\n"
|
|
data += "\n"
|
|
|
|
return data
|
|
|
|
|
|
def _wrstsegments(entries: list, convert=False) -> str:
|
|
time = 0
|
|
subs = []
|
|
for cont in entries:
|
|
cont = re.sub(r"\n\n\d+\n", "\n", cont) # remove sequence numbers
|
|
text = cont.split("\n")
|
|
for t in text: # is in text[1] for tv4play, but this should be more future proof
|
|
if "X-TIMESTAMP-MAP=MPEGTS" in t:
|
|
time = float(re.search(r"X-TIMESTAMP-MAP=MPEGTS:(\d+)", t).group(1)) / 90000
|
|
if time > 0:
|
|
time -= 10
|
|
itmes = []
|
|
if len(text) > 1:
|
|
for n in text:
|
|
if n: # don't get the empty lines.
|
|
itmes.append(n)
|
|
|
|
several_items = False
|
|
skip = False
|
|
pre_date_skip = True
|
|
sub = []
|
|
for x in range(len(itmes)):
|
|
item = itmes[x].rstrip()
|
|
if not item.rstrip():
|
|
continue
|
|
if strdate(item) and len(subs) > 0 and itmes[x + 1] == subs[-1][1]:
|
|
ha = strdate(subs[-1][0])
|
|
ha3 = strdate(item)
|
|
second = str2sec(ha3.group(4)) + time
|
|
subs[-1][0] = f"{ha.group(1).replace('.', ',')} --> {sec2str(second).replace('.', ',')}"
|
|
skip = True
|
|
pre_date_skip = False
|
|
continue
|
|
has_date = strdate(item)
|
|
if has_date:
|
|
if several_items:
|
|
subs.append(sub)
|
|
sub = []
|
|
skip = False
|
|
first = str2sec(has_date.group(1)) + time
|
|
second = str2sec(has_date.group(4)) + time
|
|
sub.append(f"{sec2str(first).replace('.', ',')} --> {sec2str(second).replace('.', ',')}")
|
|
several_items = True
|
|
pre_date_skip = False
|
|
elif has_date is None and skip is False and pre_date_skip is False:
|
|
sub.append(_wsrt_colors(convert, item))
|
|
if sub:
|
|
subs.append(sub)
|
|
string = ""
|
|
nr = 1
|
|
for sub in subs:
|
|
string += "{}\n{}\n\n".format(nr, "\n".join(sub))
|
|
nr += 1
|
|
|
|
string = re.sub("\r", "", string)
|
|
return string
|
|
|
|
|
|
def _resolv(entries):
|
|
skip = False
|
|
changed = False
|
|
new_entries = []
|
|
for nr, i in enumerate(entries):
|
|
if skip:
|
|
skip = False
|
|
continue
|
|
time_match = strdate(i[1].replace(",", "."))
|
|
time_match_next = None
|
|
if nr + 1 < len(entries):
|
|
time_match_next = strdate(entries[nr + 1][1].replace(",", "."))
|
|
left_time = time_match.group(1)
|
|
right_time = time_match.group(4)
|
|
if time_match_next and time_match.group(4) == time_match_next.group(1):
|
|
right_time = time_match_next.group(4)
|
|
skip = True
|
|
changed = True
|
|
next_entries = [nr + 1, f"{left_time} --> {right_time}"]
|
|
next_entries.extend(i[2:])
|
|
new_entries.append(next_entries)
|
|
return changed, new_entries
|
|
|
|
|
|
def timestr(msec):
|
|
"""
|
|
Convert a millisecond value to a string of the following
|
|
format:
|
|
|
|
HH:MM:SS,SS
|
|
|
|
with 10 millisecond precision. Note the , seperator in
|
|
the seconds.
|
|
"""
|
|
sec = float(msec) / 1000
|
|
|
|
hours = int(sec / 3600)
|
|
sec -= hours * 3600
|
|
|
|
minutes = int(sec / 60)
|
|
sec -= minutes * 60
|
|
|
|
return f"{hours:02d}:{minutes:02d}:{sec:06.3f}".replace(".", ",")
|
|
|
|
|
|
def timecolon(data):
|
|
match = re.search(r"(\d+:\d+:\d+):(\d+)", data)
|
|
return f"{match.group(1)},{match.group(2)}"
|
|
|
|
|
|
def norm(name):
|
|
if name[0] == "{":
|
|
_, tag = name[1:].split("}")
|
|
return tag
|
|
else:
|
|
return name
|
|
|
|
|
|
def tt_text(node, data):
|
|
if node.text:
|
|
data += "%s\n" % node.text.strip(" \t\n\r")
|
|
for i in node:
|
|
if i.text:
|
|
data += "%s\n" % i.text.strip(" \t\n\r")
|
|
if i.tail:
|
|
text = i.tail.strip(" \t\n\r")
|
|
if text:
|
|
data += f"{text}\n"
|
|
return data
|
|
|
|
|
|
def strdate(datestring):
|
|
match = re.search(r"^((\d+:\d+:\d+[\.,]*[0-9]*)?(\d+:\d+[\.,]*[0-9]*)?) --> ((\d+:\d+:\d+[\.,]*[0-9]*)?(\d+:\d+[\.,]*[0-9]*)?)$", datestring)
|
|
return match
|
|
|
|
|
|
def sec2str(seconds):
|
|
m, s = divmod(seconds, 60)
|
|
h, m = divmod(m, 60)
|
|
return f"{int(h):02d}:{int(m):02d}:{s:06.3f}"
|
|
|
|
|
|
def str2sec(string):
|
|
seconds = [3600, 60, 1]
|
|
return sum(x * float(t) for x, t in zip(seconds[3 - len(string.split(":")) :], string.split(":")))
|
|
|
|
|
|
def _wsrt_colors(convert, text):
|
|
if convert:
|
|
colors = {
|
|
"30": "#000000",
|
|
"31": "#ff0000",
|
|
"32": "#00ff00",
|
|
"33": "#ffff00",
|
|
"34": "#0000ff",
|
|
"35": "#ff00ff",
|
|
"36": "#00ffff",
|
|
"37": "#ffffff",
|
|
"c.black": "#000000",
|
|
"c.red": "#ff0000",
|
|
"c.green": "#00ff00",
|
|
"c.yellow": "#ffff00",
|
|
"c.blue": "#0000ff",
|
|
"c.magenta": "#ff00ff",
|
|
"c.cyan": "#00ffff",
|
|
"c.gray": "#ffffff",
|
|
}
|
|
for tag, color in colors.items():
|
|
regex1 = "<" + tag + ">"
|
|
replace = '<font color="' + color + '">'
|
|
text = re.sub(regex1, replace, text)
|
|
text = re.sub(f'</{tag.split(".")[0]}>', "</font>", text)
|
|
else:
|
|
text = re.sub("<[^>]*>", "", text)
|
|
return text
|