Coverage for pydantic/_internal/_decorators_v1.py: 95.00%

1"""Logic for V1 validators, e.g. `@validator` and `@root_validator`.""" 

2 

3from __future__ import annotations as _annotations 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

4 

5from inspect import Parameter, signature 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

6from typing import Any, Dict, Tuple, Union, cast 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

7 

8from pydantic_core import core_schema 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

9from typing_extensions import Protocol 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

10 

11from ..errors import PydanticUserError 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

12from ._decorators import can_be_positional 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

13 

14 

15class V1OnlyValueValidator(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

16 """A simple validator, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

17 

18 def __call__(self, __value: Any) -> Any: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

19 

20 

21class V1ValidatorWithValues(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

22 """A validator with `values` argument, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

23 

24 def __call__(self, __value: Any, values: dict[str, Any]) -> Any: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

25 

26 

27class V1ValidatorWithValuesKwOnly(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

28 """A validator with keyword only `values` argument, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

29 

30 def __call__(self, __value: Any, *, values: dict[str, Any]) -> Any: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

31 

32 

33class V1ValidatorWithKwargs(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

34 """A validator with `kwargs` argument, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

35 

36 def __call__(self, __value: Any, **kwargs: Any) -> Any: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

37 

38 

39class V1ValidatorWithValuesAndKwargs(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

40 """A validator with `values` and `kwargs` arguments, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

41 

42 def __call__(self, __value: Any, values: dict[str, Any], **kwargs: Any) -> Any: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

43 

44 

45V1Validator = Union[ 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

46 V1ValidatorWithValues, V1ValidatorWithValuesKwOnly, V1ValidatorWithKwargs, V1ValidatorWithValuesAndKwargs 

47] 

48 

49 

50def can_be_keyword(param: Parameter) -> bool: 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

51 return param.kind in (Parameter.POSITIONAL_OR_KEYWORD, Parameter.KEYWORD_ONLY) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

52 

53 

54def make_generic_v1_field_validator(validator: V1Validator) -> core_schema.WithInfoValidatorFunction: 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

55 """Wrap a V1 style field validator for V2 compatibility. 

56 

57 Args: 

58 validator: The V1 style field validator. 

59 

60 Returns: 

61 A wrapped V2 style field validator. 

62 

63 Raises: 

64 PydanticUserError: If the signature is not supported or the parameters are 

65 not available in Pydantic V2. 

66 """ 

67 sig = signature(validator) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

68 

69 needs_values_kw = False 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

70 

71 for param_num, (param_name, parameter) in enumerate(sig.parameters.items()): 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

72 if can_be_keyword(parameter) and param_name in ('field', 'config'): 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

73 raise PydanticUserError( 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

74 'The `field` and `config` parameters are not available in Pydantic V2, ' 

75 'please use the `info` parameter instead.', 

76 code='validator-field-config-info', 

77 ) 

78 if parameter.kind is Parameter.VAR_KEYWORD: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

79 needs_values_kw = True 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

80 elif can_be_keyword(parameter) and param_name == 'values': 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

81 needs_values_kw = True 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

82 elif can_be_positional(parameter) and param_num == 0: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

83 # value 

84 continue 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

85 elif parameter.default is Parameter.empty: # ignore params with defaults e.g. bound by functools.partial 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

86 raise PydanticUserError( 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

87 f'Unsupported signature for V1 style validator {validator}: {sig} is not supported.', 

88 code='validator-v1-signature', 

89 ) 

90 

91 if needs_values_kw: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

92 # (v, **kwargs), (v, values, **kwargs), (v, *, values, **kwargs) or (v, *, values) 

93 val1 = cast(V1ValidatorWithValues, validator) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

94 

95 def wrapper1(value: Any, info: core_schema.ValidationInfo) -> Any: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

96 return val1(value, values=info.data) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

97 

98 return wrapper1 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

99 else: 

100 val2 = cast(V1OnlyValueValidator, validator) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

101 

102 def wrapper2(value: Any, _: core_schema.ValidationInfo) -> Any: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

103 return val2(value) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

104 

105 return wrapper2 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

106 

107 

108RootValidatorValues = Dict[str, Any] 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

109# technically tuple[model_dict, model_extra, fields_set] | tuple[dataclass_dict, init_vars] 

110RootValidatorFieldsTuple = Tuple[Any, ...] 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

111 

112 

113class V1RootValidatorFunction(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

114 """A simple root validator, supported for V1 validators and V2 validators.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

115 

116 def __call__(self, __values: RootValidatorValues) -> RootValidatorValues: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

117 

118 

119class V2CoreBeforeRootValidator(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

120 """V2 validator with mode='before'.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

121 

122 def __call__(self, __values: RootValidatorValues, __info: core_schema.ValidationInfo) -> RootValidatorValues: ... 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

123 

124 

125class V2CoreAfterRootValidator(Protocol): 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

126 """V2 validator with mode='after'.""" 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

127 

128 def __call__( 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

129 self, __fields_tuple: RootValidatorFieldsTuple, __info: core_schema.ValidationInfo 9 ctx1abcdefGgh

130 ) -> RootValidatorFieldsTuple: ... 9 ctx1abcdefGgh

131 

132 

133def make_v1_generic_root_validator( 41 ctx1ijklmnopabcdqrstuvwxefHIJKLMNOGyzABCDEFgh

134 validator: V1RootValidatorFunction, pre: bool 

135) -> V2CoreBeforeRootValidator | V2CoreAfterRootValidator: 

136 """Wrap a V1 style root validator for V2 compatibility. 

137 

138 Args: 

139 validator: The V1 style field validator. 

140 pre: Whether the validator is a pre validator. 

141 

142 Returns: 

143 A wrapped V2 style validator. 

144 """ 

145 if pre is True: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

146 # mode='before' for pydantic-core 

147 def _wrapper1(values: RootValidatorValues, _: core_schema.ValidationInfo) -> RootValidatorValues: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

148 return validator(values) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

149 

150 return _wrapper1 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

151 

152 # mode='after' for pydantic-core 

153 def _wrapper2(fields_tuple: RootValidatorFieldsTuple, _: core_schema.ValidationInfo) -> RootValidatorFieldsTuple: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

154 if len(fields_tuple) == 2: 154 ↛ 156line 154 didn't jump to line 156, because the condition on line 154 was never true32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

155 # dataclass, this is easy 

156 values, init_vars = fields_tuple 

157 values = validator(values) 

158 return values, init_vars 

159 else: 

160 # ugly hack: to match v1 behaviour, we merge values and model_extra, then split them up based on fields 

161 # afterwards 

162 model_dict, model_extra, fields_set = fields_tuple 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

163 if model_extra: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

164 fields = set(model_dict.keys()) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

165 model_dict.update(model_extra) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

166 model_dict_new = validator(model_dict) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

167 for k in list(model_dict_new.keys()): 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

168 if k not in fields: 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

169 model_extra[k] = model_dict_new.pop(k) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

170 else: 

171 model_dict_new = validator(model_dict) 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

172 return model_dict_new, model_extra, fields_set 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh

173 

174 return _wrapper2 32 ctx1ijklmnopabcdqrstuvwxefyzABCDEFgh