close, do not use recursion to determine unresolvable, instead, only look at current layer

This commit is contained in:
Jesse Bannon 2026-01-04 11:47:57 -08:00
parent e86e414ad9
commit eeea7be52d
7 changed files with 108 additions and 5 deletions

View file

@ -695,3 +695,52 @@ class Script:
Names of all functions within the Script.
"""
return set(to_function_definition_name(name) for name in self._functions.keys())
def resolve_partial(
self,
unresolvable: Optional[Set[str]] = None,
) -> "Script":
"""
Returns
-------
Dict of resolved values
Raises
------
ScriptVariableNotResolved
If specifying a filter of variable to resolve, and one of them does not.
"""
resolved: Dict[Variable, Resolvable] = {}
unresolved: Dict[Variable, SyntaxTree] = {
Variable(name): ast for name, ast in self._variables.items() if name not in unresolvable
}
unresolvable: Set[Variable] = {Variable(name) for name in (unresolvable or {})}
partially_resolved = True
while partially_resolved:
partially_resolved = False
for variable in list(unresolved.keys()):
definition = unresolved[variable]
maybe_resolved = definition.partial_resolve(
resolved_variables=resolved, custom_functions=self._functions
)
if isinstance(maybe_resolved, Resolvable):
resolved[variable] = maybe_resolved
del unresolved[variable]
else:
unresolved[variable] = maybe_resolved
# If the definition changed, then the script changed
# which means we can iterate again
partially_resolved |= definition != maybe_resolved
return Script({}).add_parsed(
self._functions
| {var.name: definition for var, definition in resolved.items()}
| unresolved
)

View file

@ -49,10 +49,10 @@ class UnresolvedArray(_Array, VariableDependency, FutureResolvable):
)
def partial_resolve(
self,
self: TypeT,
resolved_variables: Dict[Variable, Resolvable],
custom_functions: Dict[str, "VariableDependency"],
) -> "UnresolvedArray" | Resolvable:
) -> TypeT | Resolvable:
maybe_resolvable_values, is_resolvable = VariableDependency.try_partial_resolve(
args=self.value,
resolved_variables=resolved_variables,

View file

@ -45,7 +45,7 @@ class Function(FunctionType, VariableDependency, ABC):
custom_functions: Dict[str, "VariableDependency"],
) -> TypeT | Resolvable:
maybe_resolvable_values, is_resolvable = VariableDependency.try_partial_resolve(
args=self.value,
args=self.args,
resolved_variables=resolved_variables,
custom_functions=custom_functions,
)

View file

@ -13,6 +13,7 @@ from ytdl_sub.script.types.resolvable import NonHashable
from ytdl_sub.script.types.resolvable import Resolvable
from ytdl_sub.script.types.resolvable import ResolvableToJson
from ytdl_sub.script.types.variable import Variable
from ytdl_sub.script.types.variable_dependency import TypeT
from ytdl_sub.script.types.variable_dependency import VariableDependency
from ytdl_sub.script.utils.exceptions import KeyNotHashableRuntimeException
@ -56,10 +57,10 @@ class UnresolvedMap(_Map, VariableDependency, FutureResolvable):
return Map(output)
def partial_resolve(
self,
self: TypeT,
resolved_variables: Dict[Variable, Resolvable],
custom_functions: Dict[str, "VariableDependency"],
) -> "UnresolvedMap" | Resolvable:
) -> TypeT | Resolvable:
maybe_resolvable_keys, is_keys_resolvable = VariableDependency.try_partial_resolve(
args=self.value.keys(),
resolved_variables=resolved_variables,

View file

@ -7,6 +7,7 @@ from ytdl_sub.script.types.resolvable import Argument
from ytdl_sub.script.types.resolvable import Resolvable
from ytdl_sub.script.types.resolvable import String
from ytdl_sub.script.types.variable import Variable
from ytdl_sub.script.types.variable_dependency import TypeT
from ytdl_sub.script.types.variable_dependency import VariableDependency
@ -40,6 +41,25 @@ class SyntaxTree(VariableDependency):
# Otherwise, to concat multiple resolved outputs, we must concat as strings
return String("".join([str(res) for res in resolved]))
def partial_resolve(
self: TypeT,
resolved_variables: Dict[Variable, Resolvable],
custom_functions: Dict[str, VariableDependency],
) -> TypeT | Resolvable:
maybe_resolvable_values, is_resolvable = VariableDependency.try_partial_resolve(
args=self.ast,
resolved_variables=resolved_variables,
custom_functions=custom_functions,
)
if is_resolvable:
return self.resolve(
resolved_variables=resolved_variables,
custom_functions=custom_functions,
)
return SyntaxTree(ast=maybe_resolvable_values)
@property
def maybe_resolvable(self) -> Optional[Resolvable]:
"""
@ -76,3 +96,10 @@ class ResolvedSyntaxTree(SyntaxTree):
@property
def maybe_resolvable(self) -> Optional[Resolvable]:
return self.ast[0]
def partial_resolve(
self: TypeT,
resolved_variables: Dict[Variable, Resolvable],
custom_functions: Dict[str, VariableDependency],
) -> TypeT | Resolvable:
return self.ast[0]

View file

@ -271,7 +271,14 @@ class VariableDependency(ABC):
if not isinstance(maybe_resolvable_args[-1], Resolvable):
is_resolvable = False
elif isinstance(arg, Variable):
if arg not in resolved_variables:
is_resolvable = False
maybe_resolvable_args.append(arg)
else:
maybe_resolvable_args.append(arg)
if is_resolvable:
print("hmm")
return maybe_resolvable_args, is_resolvable

View file

@ -598,3 +598,22 @@ def test_default_docker_config_and_subscriptions(
"year": "{episode_year}",
},
}
def test_default_docker_config_and_subscriptions(
docker_default_subscription_path: Path, output_directory: str
):
default_config = ConfigFile.from_file_path("docker/root/defaults/config.yaml")
default_subs = Subscription.from_file_path(
config=default_config, subscription_path=docker_default_subscription_path
)
assert len(default_subs) == 1
unresolvable = default_subs[0].plugins.get_all_variables(
additional_options=[default_subs[0].downloader_options, default_subs[0].output_options]
)
unresolvable.add("entry_metadata")
unresolvable.add("sibling_metadata")
out = default_subs[0].overrides.script.resolve_partial(unresolvable=unresolvable)
print("hi")