Assume builtins don't accept an info argument (pydantic#6754)

FacerAin · Jul 19, 2023 · 98a8d56 · 98a8d56
1 parent c588b5d
commit 98a8d56
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 0 deletions.
diff --git a/pydantic/_internal/_decorators.py b/pydantic/_internal/_decorators.py
@@ -504,6 +504,9 @@ def inspect_validator(validator: Callable[..., Any], mode: FieldValidatorModes)
  Returns:
  Whether the validator takes an info argument.
  """
+ if getattr(validator, '__module__', None) == 'builtins':
+ # int, str, etc.
+ return False
  sig = signature(validator)
  n_positional = count_positional_params(sig)
  if mode == 'wrap':

diff --git a/tests/test_validators.py b/tests/test_validators.py
@@ -58,6 +58,16 @@ class Model(BaseModel):
  assert Model(x='1.0').x == 1.0
 
 
+def test_annotated_validator_builtin() -> None:
+ """https://github.com/pydantic/pydantic/issues/6752"""
+ TruncatedFloat = Annotated[float, BeforeValidator(int)]
+
+ class Model(BaseModel):
+ x: TruncatedFloat
+
+ assert Model(x=1.234).x == 1
+
+
 def test_annotated_validator_plain() -> None:
  MyInt = Annotated[int, PlainValidator(lambda x, _info: x if x != -1 else 0)]