danvics/ytdl-sub
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix adding custom functions

Browse source
This commit is contained in:
Jesse Bannon 2024-03-29 23:38:04 -07:00
parent d44e2fcd5f
commit cda9fc73ac
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/ytdl_sub/script/script.py
View file

@ -477,7 +477,7 @@ class Script:
added_variables_to_validate.add(name) added_variables_to_validate.add(name)
if name in functions_to_add: if name in functions_to_add:
self._functions[_function_name(name)] = parsed self._functions[name] = parsed
else: else:
self._variables[name] = parsed self._variables[name] = parsed

6
tests/unit/script/test_script.py
View file

@ -80,4 +80,8 @@ class TestScript:
} }
) )
assert script.resolve_once({"url": "{ %bilateral_url('nope') }"})["url"].native == "nope" script.add({"%bilateral_url_wrap": "{ %bilateral_url($0) }"})
assert (
script.resolve_once({"url": "{ %bilateral_url_wrap('nope') }"})["url"].native == "nope"
)