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

lint, unit test

Browse source
This commit is contained in:
Jesse Bannon 2026-03-13 16:16:22 -07:00
parent 6cffcf08e1
commit 01e7cfdad7
3 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/ytdl_sub/entries/script/variable_definitions.py
View file

@ -1216,6 +1216,9 @@ class VariableDefinitions(
} | self.injected_variables() } | self.injected_variables()
def get(self, name: str) -> Optional[Variable]: def get(self, name: str) -> Optional[Variable]:
"""
Returns the variable attribute if it exists. None otherwise.
"""
if not hasattr(self, name): if not hasattr(self, name):
return None return None
return getattr(self, name) return getattr(self, name)

1
src/ytdl_sub/utils/script.py
View file

@ -18,6 +18,7 @@ from ytdl_sub.script.utils.name_validation import is_function
# pylint: disable=too-many-return-statements # pylint: disable=too-many-return-statements
# pylint: disable=too-many-branches # pylint: disable=too-many-branches
class ScriptUtils: class ScriptUtils:
@classmethod @classmethod
def add_sanitized_variables(cls, variables: Dict[str, str]) -> Dict[str, str]: def add_sanitized_variables(cls, variables: Dict[str, str]) -> Dict[str, str]:

4
tests/unit/validators/test_string_formatter_validator.py
View file

@ -113,12 +113,12 @@ class TestUnstructuredDictFormatterValidator(object):
assert len(validator.dict) == 8 assert len(validator.dict) == 8
assert all(isinstance(val, expected_formatter_class) for val in validator.dict.values()) assert all(isinstance(val, expected_formatter_class) for val in validator.dict.values())
assert validator.dict_with_format_strings == { assert validator.dict_with_format_strings == {
"key1": '{ %concat( "string with ", variable ) }', "key1": "string with { variable }",
"key2": "no variables", "key2": "no variables",
"key3": "{ %int(3) }", "key3": "{ %int(3) }",
"key4": "{ %float(4.132) }", "key4": "{ %float(4.132) }",
"key5": "{ %bool(True) }", "key5": "{ %bool(True) }",
"key6": '{ { %concat( variable, "_key" ): "value", "static_key": %concat( variable, "_value" ) } }', "key6": '{ { %concat( variable, "_key" ): "value", "static_key": %concat( variable, "_value" ) } }',
"key7": '{ [ "list_1", %concat( "list_", variable_2 ) ] }', "key7": '{ [ "list_1", %concat( "list_", variable_2 ) ] }',
"key8": '{ %concat( "string ", variable1, " with multiple ", variable2 ) }', "key8": "string { variable1 } with multiple { variable2 }",
} }