mirror of
https://github.com/unshackle-dl/unshackle.git
synced 2026-11-12 20:15:10 +00:00
imSp4rky
1f96a4bd78
Strip cue classes, <v>/<lang>/ruby and karaoke timestamps before conversion, decode entities for SRT/SSA/ASS, load pysubs2 as utf-8-sig, and route sdh_method: filter-subs through convert_before_strip instead of raising.
63 lines
2.4 KiB
Python
63 lines
2.4 KiB
Python
from unshackle.core.tracks.subtitle import Subtitle
|
|
|
|
|
|
def test_strip_webvtt_cue_classes():
|
|
text = (
|
|
"WEBVTT\n\n"
|
|
"00:00:01.000 --> 00:00:04.000\n"
|
|
"<c.bg_transparent>Mor, da vi blev gift,</c>\n"
|
|
"<c>kom Ben</c> <c.a.b>og jeg</c.a.b> til dig.\n"
|
|
"<i>Keep</i> <b>these</b> and <code>this</code>\n"
|
|
"<i.loud>class</i> <b.x>on</b> <u.y>styling</u>\n"
|
|
"<v.z Bob>voice</v> <lang ja>jp</lang> <ruby>\u6f22<rt>\u304b\u3093</rt></ruby>\n"
|
|
"kara <00:00:01.500>timing <00:01.500>short\n"
|
|
)
|
|
out = Subtitle.strip_webvtt_cue_classes(text)
|
|
assert "bg_transparent" not in out and "</c>" not in out and "<c>" not in out
|
|
assert "Mor, da vi blev gift," in out
|
|
assert "<i>Keep</i> <b>these</b> and <code>this</code>" in out
|
|
# class list dropped, the styling tag itself kept
|
|
assert "<i>class</i> <b>on</b> <u>styling</u>" in out
|
|
# WebVTT-only tags go, their text stays
|
|
assert "voice jp \u6f22\u304b\u3093" in out
|
|
assert "kara timing short" in out
|
|
|
|
|
|
def test_decode_entities(tmp_path):
|
|
from unshackle.core.tracks.subtitle_convert import decode_entities
|
|
|
|
out = tmp_path / "a.srt"
|
|
out.write_text(
|
|
"1\n00:00:01,000 --> 00:00:04,000\nROBERT & RA'ED x 'q'\nTom ¬ sure, AT&T\n",
|
|
encoding="utf8",
|
|
)
|
|
decode_entities(out)
|
|
text = out.read_text(encoding="utf8")
|
|
assert "ROBERT & RA'ED \xa0x 'q'" in text
|
|
# no-semicolon legacy forms stay as written
|
|
assert "Tom ¬ sure, AT&T" in text
|
|
|
|
|
|
def test_filter_subs_converts_before_stripping(tmp_path, monkeypatch):
|
|
"""``sdh_method: filter-subs`` used to hand subtitle-filter a .vtt and raise OSError."""
|
|
from unshackle.core import binaries
|
|
from unshackle.core.config import config
|
|
|
|
monkeypatch.setattr(binaries, "SubtitleEdit", None)
|
|
monkeypatch.setattr(config, "subtitle", {"sdh_method": "filter-subs"}, raising=False)
|
|
|
|
path = tmp_path / "sub.vtt"
|
|
path.write_text(
|
|
"WEBVTT\n\n00:00:01.000 --> 00:00:03.000\n[MUSIC PLAYING]\n\n"
|
|
"00:00:04.000 --> 00:00:06.000\nPlain dialogue stays.\n",
|
|
encoding="utf8",
|
|
)
|
|
sub = Subtitle(url="https://example.test/x", language="en", codec=Subtitle.Codec.WebVTT)
|
|
sub.path = path
|
|
sub.strip_hearing_impaired()
|
|
|
|
assert sub.codec == Subtitle.Codec.SubRip
|
|
text = sub.path.read_text(encoding="utf8")
|
|
assert "MUSIC PLAYING" not in text
|
|
assert "Plain dialogue stays." in text
|