From dd08c0aa8dcbb520e32a88332b7182e56c801046 Mon Sep 17 00:00:00 2001 From: Jesse Bannon Date: Wed, 13 Dec 2023 14:26:07 -0800 Subject: [PATCH] fix int formatter --- .../validators/string_formatter_validators.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/ytdl_sub/validators/string_formatter_validators.py b/src/ytdl_sub/validators/string_formatter_validators.py index 3f6f3047..546816b5 100644 --- a/src/ytdl_sub/validators/string_formatter_validators.py +++ b/src/ytdl_sub/validators/string_formatter_validators.py @@ -15,6 +15,7 @@ from ytdl_sub.validators.validators import DictValidator from ytdl_sub.validators.validators import ListValidator from ytdl_sub.validators.validators import LiteralDictValidator from ytdl_sub.validators.validators import StringValidator +from ytdl_sub.validators.validators import Validator _fields_validator = re.compile(r"{([a-z][a-z0-9_]*?)}") @@ -122,15 +123,15 @@ class OverridesStringFormatterValidator(StringFormatterValidator): class OverridesIntegerFormatterValidator(StringFormatterValidator): _expected_value_type_name = "integer" - def apply_formatter(self, variable_dict: Dict[str, str]) -> str: - output = super().apply_formatter(variable_dict) + def post_process(self, resolved: str) -> str: try: - int(output) + int(resolved) except Exception as exc: raise self._validation_exception( - f"Expected an integer, but received '{output}'" + f"Expected an integer, but received '{resolved}'" ) from exc - return output + + return resolved class ListFormatterValidator(ListValidator[StringFormatterValidator]):