530d2f7666
Adds the mechanism to partially resolve an entire script (aka every underlying variable in ytdl-sub). This will be used in the upcoming `inspect` sub-command where it will dump a subscription's raw contents at multiple levels of resolution. Potentially useful to optimize script execution as well.
118 lines
3.8 KiB
Python
118 lines
3.8 KiB
Python
from pathlib import Path
|
|
|
|
from resources import expected_json
|
|
|
|
from ytdl_sub.config.config_file import ConfigFile
|
|
from ytdl_sub.entries.script.variable_definitions import VARIABLES
|
|
from ytdl_sub.subscriptions.subscription import Subscription
|
|
from ytdl_sub.utils.script import ScriptUtils
|
|
|
|
|
|
def _ensure_partial_resolve(
|
|
sub: Subscription, preset_type: str, should_fully_resolve: bool
|
|
) -> None:
|
|
unresolvable = sub.plugins.get_all_variables(
|
|
additional_options=[sub.downloader_options, sub.output_options]
|
|
)
|
|
unresolvable.add("entry_metadata")
|
|
unresolvable.add("sibling_metadata")
|
|
|
|
if not should_fully_resolve:
|
|
unresolvable.update(VARIABLES.scripts().keys())
|
|
|
|
script = sub.overrides.script.add(
|
|
{
|
|
f"{preset_type}_directory": "tv_show_directory_path",
|
|
f"{preset_type}_directory_sanitized": "tv_show_directory_path",
|
|
}
|
|
)
|
|
partial_script = script.resolve_partial(unresolvable=unresolvable)
|
|
|
|
# Assert only overrides
|
|
out = {
|
|
name: ScriptUtils.to_native_script(partial_script._variables[name])
|
|
for name in sub.overrides.keys
|
|
if not name.startswith("%")
|
|
}
|
|
|
|
expected_out_filename = f"{preset_type}/inspect_overrides.json"
|
|
if should_fully_resolve:
|
|
expected_out_filename = f"{preset_type}/inspect_full_overrides.json"
|
|
|
|
assert out == expected_json(out, expected_out_filename)
|
|
|
|
# Assert all variables
|
|
out = {
|
|
name: ScriptUtils.to_native_script(partial_script._variables[name])
|
|
for name in partial_script.variable_names
|
|
}
|
|
|
|
expected_out_filename = f"{preset_type}/inspect_variables.json"
|
|
if should_fully_resolve:
|
|
expected_out_filename = f"{preset_type}/inspect_full_variables.json"
|
|
|
|
assert out == expected_json(out, expected_out_filename)
|
|
|
|
|
|
def test_partial_resolve_tv_show(config_file: ConfigFile, tv_show_subscriptions_path: Path):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=config_file, subscription_path=tv_show_subscriptions_path
|
|
)[0],
|
|
preset_type="tv_show",
|
|
should_fully_resolve=True,
|
|
)
|
|
|
|
|
|
def test_partial_resolve_tv_show_full(config_file: ConfigFile, tv_show_subscriptions_path: Path):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=config_file, subscription_path=tv_show_subscriptions_path
|
|
)[0],
|
|
preset_type="tv_show",
|
|
should_fully_resolve=False,
|
|
)
|
|
|
|
|
|
def test_partial_resolve_music(default_config: ConfigFile, music_subscriptions_path: Path):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=default_config, subscription_path=music_subscriptions_path
|
|
)[0],
|
|
preset_type="music",
|
|
should_fully_resolve=True,
|
|
)
|
|
|
|
|
|
def test_partial_resolve_music_full(default_config: ConfigFile, music_subscriptions_path: Path):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=default_config, subscription_path=music_subscriptions_path
|
|
)[0],
|
|
preset_type="music",
|
|
should_fully_resolve=False,
|
|
)
|
|
|
|
|
|
def test_partial_resolve_music_video(
|
|
default_config: ConfigFile, music_video_subscription_path: Path
|
|
):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=default_config, subscription_path=music_video_subscription_path
|
|
)[0],
|
|
preset_type="music_video",
|
|
should_fully_resolve=False,
|
|
)
|
|
|
|
|
|
def test_partial_resolve_music_video_full(
|
|
default_config: ConfigFile, music_video_subscription_path: Path
|
|
):
|
|
_ensure_partial_resolve(
|
|
sub=Subscription.from_file_path(
|
|
config=default_config, subscription_path=music_video_subscription_path
|
|
)[0],
|
|
preset_type="music_video",
|
|
should_fully_resolve=True,
|
|
)
|