Coverage for pydantic/_internal/_validate_call.py: 100.00%

1from __future__ import annotations as _annotations 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

2 

3import inspect 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

4from functools import partial 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

5from typing import Any, Awaitable, Callable 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

6 

7import pydantic_core 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

8 

9from ..config import ConfigDict 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

10from ..plugin._schema_validator import create_schema_validator 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

11from . import _generate_schema, _typing_extra 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

12from ._config import ConfigWrapper 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

13 

14 

15class ValidateCallWrapper: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

16 """This is a wrapper around a function that validates the arguments passed to it, and optionally the return value.""" 

17 

18 __slots__ = ( 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

19 '__pydantic_validator__', 

20 '__name__', 

21 '__qualname__', 

22 '__annotations__', 

23 '__dict__', # required for __module__ 

24 ) 

25 

26 def __init__( 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

27 self, 

28 function: Callable[..., Any], 

29 config: ConfigDict | None, 

30 validate_return: bool, 

31 namespace: dict[str, Any] | None, 

32 ): 

33 if isinstance(function, partial): 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

34 func = function.func 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

35 schema_type = func 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

36 self.__name__ = f'partial({func.__name__})' 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

37 self.__qualname__ = f'partial({func.__qualname__})' 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

38 self.__module__ = func.__module__ 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

39 else: 

40 schema_type = function 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

41 self.__name__ = function.__name__ 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

42 self.__qualname__ = function.__qualname__ 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

43 self.__module__ = function.__module__ 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

44 

45 global_ns = _typing_extra.add_module_globals(function, None) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

46 # TODO: this is a bit of a hack, we should probably have a better way to handle this 

47 # specifically, we shouldn't be pumping the namespace full of type_params 

48 # when we take namespace and type_params arguments in eval_type_backport 

49 type_params = getattr(schema_type, '__type_params__', ()) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

50 namespace = { 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

51 **{param.__name__: param for param in type_params}, 

52 **(global_ns or {}), 

53 **(namespace or {}), 

54 } 

55 config_wrapper = ConfigWrapper(config) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

56 gen_schema = _generate_schema.GenerateSchema(config_wrapper, namespace) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

57 schema = gen_schema.clean_schema(gen_schema.generate_schema(function)) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

58 core_config = config_wrapper.core_config(self) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

59 

60 self.__pydantic_validator__ = create_schema_validator( 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

61 schema, 

62 schema_type, 

63 self.__module__, 

64 self.__qualname__, 

65 'validate_call', 

66 core_config, 

67 config_wrapper.plugin_settings, 

68 ) 

69 

70 if validate_return: 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

71 signature = inspect.signature(function) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

72 return_type = signature.return_annotation if signature.return_annotation is not signature.empty else Any 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

73 gen_schema = _generate_schema.GenerateSchema(config_wrapper, namespace) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

74 schema = gen_schema.clean_schema(gen_schema.generate_schema(return_type)) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

75 validator = create_schema_validator( 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

76 schema, 

77 schema_type, 

78 self.__module__, 

79 self.__qualname__, 

80 'validate_call', 

81 core_config, 

82 config_wrapper.plugin_settings, 

83 ) 

84 if inspect.iscoroutinefunction(function): 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

85 

86 async def return_val_wrapper(aw: Awaitable[Any]) -> None: 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

87 return validator.validate_python(await aw) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

88 

89 self.__return_pydantic_validator__ = return_val_wrapper 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

90 else: 

91 self.__return_pydantic_validator__ = validator.validate_python 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

92 else: 

93 self.__return_pydantic_validator__ = None 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

94 

95 def __call__(self, *args: Any, **kwargs: Any) -> Any: 48 ctx1abcdefghijklmnopqrstuvwxyzMNOPQRSTUVABCDEFGHIJKL

96 res = self.__pydantic_validator__.validate_python(pydantic_core.ArgsKwargs(args, kwargs)) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

97 if self.__return_pydantic_validator__: 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

98 return self.__return_pydantic_validator__(res) 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL

99 return res 38 ctx1abcdefghijklmnopqrstuvwxyzABCDEFGHIJKL