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

fix int formatter

Browse source
This commit is contained in:
Jesse Bannon 2023-12-13 14:26:07 -08:00
parent c38ac2da4d
commit dd08c0aa8d
1 changed files with 6 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/ytdl_sub/validators/string_formatter_validators.py
View file

@ -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]):