Coverage for pydantic/validate_call_decorator.py: 100.00%

1"""Decorator for validating function calls.""" 

2 

3from __future__ import annotations as _annotations 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

4 

5import inspect 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

6from functools import partial 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

7from types import BuiltinFunctionType 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

8from typing import TYPE_CHECKING, Any, Callable, TypeVar, cast, overload 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

9 

10from ._internal import _generate_schema, _typing_extra, _validate_call 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

11from .errors import PydanticUserError 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

12 

13__all__ = ('validate_call',) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

14 

15if TYPE_CHECKING: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

16 from .config import ConfigDict 

17 

18 AnyCallableT = TypeVar('AnyCallableT', bound=Callable[..., Any]) 

19 

20 

21_INVALID_TYPE_ERROR_CODE = 'validate-call-type' 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

22 

23 

24def _check_function_type(function: object) -> None: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

25 """Check if the input function is a supported type for `validate_call`.""" 

26 if isinstance(function, _generate_schema.VALIDATE_CALL_SUPPORTED_TYPES): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

27 try: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

28 inspect.signature(cast(_generate_schema.ValidateCallSupportedTypes, function)) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

29 except ValueError: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

30 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

31 f"Input function `{function}` doesn't have a valid signature", code=_INVALID_TYPE_ERROR_CODE 

32 ) 

33 

34 if isinstance(function, partial): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

35 try: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

36 assert not isinstance(partial.func, partial), 'Partial of partial' 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

37 _check_function_type(function.func) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

38 except PydanticUserError as e: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

39 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

40 f'Partial of `{function.func}` is invalid because the type of `{function.func}` is not supported by `validate_call`', 

41 code=_INVALID_TYPE_ERROR_CODE, 

42 ) from e 

43 

44 return 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

45 

46 if isinstance(function, BuiltinFunctionType): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

47 raise PydanticUserError(f'Input built-in function `{function}` is not supported', code=_INVALID_TYPE_ERROR_CODE) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

48 if isinstance(function, (classmethod, staticmethod, property)): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

49 name = type(function).__name__ 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

50 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

51 f'The `@{name}` decorator should be applied after `@validate_call` (put `@{name}` on top)', 

52 code=_INVALID_TYPE_ERROR_CODE, 

53 ) 

54 

55 if inspect.isclass(function): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

56 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

57 f'Unable to validate {function}: `validate_call` should be applied to functions, not classes (put `@validate_call` on top of `__init__` or `__new__` instead)', 

58 code=_INVALID_TYPE_ERROR_CODE, 

59 ) 

60 if callable(function): 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

61 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

62 f'Unable to validate {function}: `validate_call` should be applied to functions, not instances or other callables. Use `validate_call` explicitly on `__call__` instead.', 

63 code=_INVALID_TYPE_ERROR_CODE, 

64 ) 

65 

66 raise PydanticUserError( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

67 f'Unable to validate {function}: `validate_call` should be applied to one of the following: function, method, partial, or lambda', 

68 code=_INVALID_TYPE_ERROR_CODE, 

69 ) 

70 

71 

72@overload 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

73def validate_call( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

74 *, config: ConfigDict | None = None, validate_return: bool = False 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

75) -> Callable[[AnyCallableT], AnyCallableT]: ... 14 ctx1abcdstefghijkl

76 

77 

78@overload 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

79def validate_call(func: AnyCallableT, /) -> AnyCallableT: ... 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

80 

81 

82def validate_call( 30 ctx1mnopqrabcduvwxyzefghABCDEFijkl

83 func: AnyCallableT | None = None, 

84 /, 

85 *, 

86 config: ConfigDict | None = None, 

87 validate_return: bool = False, 

88) -> AnyCallableT | Callable[[AnyCallableT], AnyCallableT]: 

89 """!!! abstract "Usage Documentation" 

90 [Validation Decorator](../concepts/validation_decorator.md) 

91 

92 Returns a decorated wrapper around the function that validates the arguments and, optionally, the return value. 

93 

94 Usage may be either as a plain decorator `@validate_call` or with arguments `@validate_call(...)`. 

95 

96 Args: 

97 func: The function to be decorated. 

98 config: The configuration dictionary. 

99 validate_return: Whether to validate the return value. 

100 

101 Returns: 

102 The decorated function. 

103 """ 

104 parent_namespace = _typing_extra.parent_frame_namespace() 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

105 

106 def validate(function: AnyCallableT) -> AnyCallableT: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

107 _check_function_type(function) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

108 validate_call_wrapper = _validate_call.ValidateCallWrapper( 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

109 cast(_generate_schema.ValidateCallSupportedTypes, function), config, validate_return, parent_namespace 

110 ) 

111 return _validate_call.update_wrapper_attributes(function, validate_call_wrapper.__call__) # type: ignore 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

112 

113 if func is not None: 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

114 return validate(func) 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl

115 else: 

116 return validate 32 ctx1mnopqrabcdstuvwxyzefghABCDEFijkl