feat(tracks): add Video.hdr10_base and forward s_lang over remote and REST

A DV track with an HDR10 base (profile 8.1) lists as DV+HDR10; dl_params now carry s_lang.
This commit is contained in:
imSp4rky
2026-10-27 13:06:48 -06:00
parent 0440f2ba54
commit 9017ab95b7
12 changed files with 54 additions and 13 deletions
+6
View File
@@ -215,3 +215,9 @@ def test_a_quoted_dl_workers_value_works_like_the_cli(monkeypatch: pytest.Monkey
service = SimpleNamespace(get_tracks=lambda title: SimpleNamespace(videos=[hevc("v", 2160)]), session=None)
monkeypatch.setitem(config.dl, "workers", "8")
assert probe_calls(monkeypatch, service) == [8]
def test_a_dv_track_with_an_hdr10_base_lists_as_dv_plus_hdr10() -> None:
video = Video(url="https://x/v.mp4", language="en", codec=Video.Codec.HEVC, range_=Video.Range.DV, hdr10_base=True)
assert "[H.265, DV+HDR10]" in str(video)
assert Video.from_dict(video.to_dict()).hdr10_base
+5 -3
View File
@@ -192,17 +192,18 @@ class _CapturedCtx(Exception):
@pytest.mark.parametrize(
("params", "expected"),
[
({}, (["orig"], [], [], [], False, False)),
({}, (["orig"], [], [], ["all"], [], False, False)),
(
{
"lang": ["es-419"],
"v_lang": ["en"],
"a_lang": ["ja"],
"s_lang": ["ko"],
"acodec": "AAC,EC3",
"forced_subs": True,
"exact_lang": True,
},
(["es-419"], ["en"], ["ja"], ["AAC", "EC3"], True, True),
(["es-419"], ["en"], ["ja"], ["ko"], ["AAC", "EC3"], True, True),
),
],
)
@@ -231,7 +232,8 @@ def test_perform_download_puts_the_lang_selection_in_the_service_ctx(
ctx = exc.value.args[0]
p = ctx.params
selection = (p["lang"], p["v_lang"], p["a_lang"], [c.name for c in p["acodec"]], p["forced_subs"], p["exact_lang"])
acodec = [c.name for c in p["acodec"]]
selection = (p["lang"], p["v_lang"], p["a_lang"], p["s_lang"], acodec, p["forced_subs"], p["exact_lang"])
assert selection == expected
@@ -6,6 +6,7 @@ from enum import Enum
import pytest
from unshackle.core.api.handlers import serialize_video_track
from unshackle.core.remote_service import (
RemoteService,
build_title,
@@ -52,6 +53,11 @@ def test_deserialize_video_minimal() -> None:
assert v.range is Video.Range.SDR
def test_a_dv_track_keeps_its_hdr10_base_label_over_remote() -> None:
v = Video(url="https://x/v.mp4", language="en", codec=Video.Codec.HEVC, range_=Video.Range.DV, hdr10_base=True)
assert deserialize_video(serialize_video_track(v)).range_label == "DV+HDR10"
def test_deserialize_video_unknown_codec_falls_back_to_none() -> None:
v = deserialize_video({"id": "v2", "codec": "MADE_UP", "width": 0, "height": 0})
assert v.codec is None
@@ -624,6 +630,7 @@ def test_dl_selection_travels_nested_and_a_service_option_cannot_override_it() -
"lang": ["orig", "es-419"],
"v_lang": [],
"a_lang": [],
"s_lang": ["ko"],
"acodec": [Audio.Codec.EC3],
"forced_subs": True,
"exact_lang": True,
@@ -633,6 +640,7 @@ def test_dl_selection_travels_nested_and_a_service_option_cannot_override_it() -
"lang": ["orig", "es-419"],
"v_lang": [],
"a_lang": [],
"s_lang": ["ko"],
"acodec": ["EC3"],
"forced_subs": True,
"exact_lang": True,
+11 -1
View File
@@ -162,7 +162,15 @@ def _create_with_real_ctx(monkeypatch, data):
return svc
DL_DEFAULTS = {"lang": ["orig"], "v_lang": [], "a_lang": [], "acodec": [], "forced_subs": False, "exact_lang": False}
DL_DEFAULTS = {
"lang": ["orig"],
"v_lang": [],
"a_lang": [],
"s_lang": ["all"],
"acodec": [],
"forced_subs": False,
"exact_lang": False,
}
def test_nested_dl_params_reach_the_parent_ctx(monkeypatch):
@@ -173,6 +181,7 @@ def test_nested_dl_params_reach_the_parent_ctx(monkeypatch):
"lang": ["es-419", "en"],
"v_lang": ["ja"],
"a_lang": ["es-419"],
"s_lang": ["ko"],
"acodec": ["EC3", "NOPE", "AAC"],
"forced_subs": True,
"exact_lang": True,
@@ -182,6 +191,7 @@ def test_nested_dl_params_reach_the_parent_ctx(monkeypatch):
assert svc.parent_params["lang"] == ["es-419", "en"]
assert svc.parent_params["v_lang"] == ["ja"]
assert svc.parent_params["a_lang"] == ["es-419"]
assert svc.parent_params["s_lang"] == ["ko"]
assert svc.parent_params["acodec"] == [Audio.Codec.EC3, Audio.Codec.AAC]
assert svc.parent_params["forced_subs"] is True
assert svc.parent_params["exact_lang"] is True