diff --git a/src/ytdl_sub/validators/string_formatter_validators.py b/src/ytdl_sub/validators/string_formatter_validators.py index 53fbc0dc..77ed8f75 100644 --- a/src/ytdl_sub/validators/string_formatter_validators.py +++ b/src/ytdl_sub/validators/string_formatter_validators.py @@ -10,7 +10,7 @@ from yt_dlp.utils import sanitize_filename from ytdl_sub.utils.exceptions import InvalidVariableNameException from ytdl_sub.utils.exceptions import StringFormattingException from ytdl_sub.utils.exceptions import StringFormattingVariableNotFoundException -from ytdl_sub.validators.validators import ListValidator +from ytdl_sub.validators.validators import ListValidator, StringValidator from ytdl_sub.validators.validators import LiteralDictValidator from ytdl_sub.validators.validators import Validator @@ -47,7 +47,7 @@ def is_valid_source_variable_name(input_str: str, raise_exception: bool = False) return is_source_variable_name -class StringFormatterValidator(Validator): +class StringFormatterValidator(StringValidator): """ String that can use :class:`source variables ` @@ -73,7 +73,6 @@ class StringFormatterValidator(Validator): and would resolve to something like ``sweet_tv_show.s2022.e502.mp4``. """ - _expected_value_type = str _expected_value_type_name = "format string" _variable_not_found_error_msg_formatter = ( "Format variable '{variable_name}' does not exist. Available variables: {available_fields}" diff --git a/src/ytdl_sub/validators/validators.py b/src/ytdl_sub/validators/validators.py index 1220cfca..5bd4dbaa 100644 --- a/src/ytdl_sub/validators/validators.py +++ b/src/ytdl_sub/validators/validators.py @@ -138,6 +138,12 @@ class StringValidator(ValueValidator[str]): _expected_value_type = str _expected_value_type_name = "string" + def __init__(self, name: str, value: Any): + if isinstance(value, (int, float, bool)): + value = str(value) + + super().__init__(name, value) + class FloatValidator(ValueValidator[float]): _expected_value_type = (int, float) diff --git a/tests/unit/validators/test_string_validator.py b/tests/unit/validators/test_string_validator.py index b495bdef..93b930b6 100644 --- a/tests/unit/validators/test_string_validator.py +++ b/tests/unit/validators/test_string_validator.py @@ -4,12 +4,12 @@ from ytdl_sub.utils.exceptions import ValidationException from ytdl_sub.validators.validators import StringValidator -@pytest.mark.parametrize("value", ["a", "unicode 💩", ""]) +@pytest.mark.parametrize("value", ["a", "unicode 💩", "", 1, 3.14, True, False]) def test_string_validator(value): string_validator = StringValidator(name="good_str_validator", value=value) assert string_validator._name == "good_str_validator" - assert string_validator._value == value - assert string_validator.value == value + assert string_validator._value == str(value) + assert string_validator.value == str(value) @pytest.mark.parametrize("value", [None, {}, True, 0])