Coverage for pydantic/_internal/_decorators.py: 97.58%

1"""Logic related to validators applied to models etc. via the `@field_validator` and `@model_validator` decorators.""" 

2 

3from __future__ import annotations as _annotations 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

4 

5import types 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

6from collections import deque 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

7from collections.abc import Iterable 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

8from dataclasses import dataclass, field 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

9from functools import cached_property, partial, partialmethod 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

10from inspect import Parameter, Signature, isdatadescriptor, ismethoddescriptor, signature 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

11from itertools import islice 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

12from typing import TYPE_CHECKING, Any, Callable, ClassVar, Generic, Literal, TypeVar, Union 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

13 

14from pydantic_core import PydanticUndefined, PydanticUndefinedType, core_schema 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

15from typing_extensions import TypeAlias, is_typeddict 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

16 

17from ..errors import PydanticUserError 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

18from ._core_utils import get_type_ref 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

19from ._internal_dataclass import slots_true 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

20from ._namespace_utils import GlobalsNamespace, MappingNamespace 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

21from ._typing_extra import get_function_type_hints 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

22from ._utils import can_be_positional 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

23 

24if TYPE_CHECKING: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

25 from ..fields import ComputedFieldInfo 

26 from ..functional_validators import FieldValidatorModes 

27 from ._config import ConfigWrapper 

28 

29 

30@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

31class ValidatorDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

32 """A container for data from `@validator` so that we can access it 

33 while building the pydantic-core schema. 

34 

35 Attributes: 

36 decorator_repr: A class variable representing the decorator string, '@validator'. 

37 fields: A tuple of field names the validator should be called on. 

38 mode: The proposed validator mode. 

39 each_item: For complex objects (sets, lists etc.) whether to validate individual 

40 elements rather than the whole object. 

41 always: Whether this method and other validators should be called even if the value is missing. 

42 check_fields: Whether to check that the fields actually exist on the model. 

43 """ 

44 

45 decorator_repr: ClassVar[str] = '@validator' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

46 

47 fields: tuple[str, ...] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

48 mode: Literal['before', 'after'] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

49 each_item: bool 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

50 always: bool 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

51 check_fields: bool | None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

52 

53 

54@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

55class FieldValidatorDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

56 """A container for data from `@field_validator` so that we can access it 

57 while building the pydantic-core schema. 

58 

59 Attributes: 

60 decorator_repr: A class variable representing the decorator string, '@field_validator'. 

61 fields: A tuple of field names the validator should be called on. 

62 mode: The proposed validator mode. 

63 check_fields: Whether to check that the fields actually exist on the model. 

64 json_schema_input_type: The input type of the function. This is only used to generate 

65 the appropriate JSON Schema (in validation mode) and can only specified 

66 when `mode` is either `'before'`, `'plain'` or `'wrap'`. 

67 """ 

68 

69 decorator_repr: ClassVar[str] = '@field_validator' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

70 

71 fields: tuple[str, ...] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

72 mode: FieldValidatorModes 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

73 check_fields: bool | None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

74 json_schema_input_type: Any 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

75 

76 

77@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

78class RootValidatorDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

79 """A container for data from `@root_validator` so that we can access it 

80 while building the pydantic-core schema. 

81 

82 Attributes: 

83 decorator_repr: A class variable representing the decorator string, '@root_validator'. 

84 mode: The proposed validator mode. 

85 """ 

86 

87 decorator_repr: ClassVar[str] = '@root_validator' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

88 mode: Literal['before', 'after'] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

89 

90 

91@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

92class FieldSerializerDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

93 """A container for data from `@field_serializer` so that we can access it 

94 while building the pydantic-core schema. 

95 

96 Attributes: 

97 decorator_repr: A class variable representing the decorator string, '@field_serializer'. 

98 fields: A tuple of field names the serializer should be called on. 

99 mode: The proposed serializer mode. 

100 return_type: The type of the serializer's return value. 

101 when_used: The serialization condition. Accepts a string with values `'always'`, `'unless-none'`, `'json'`, 

102 and `'json-unless-none'`. 

103 check_fields: Whether to check that the fields actually exist on the model. 

104 """ 

105 

106 decorator_repr: ClassVar[str] = '@field_serializer' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

107 fields: tuple[str, ...] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

108 mode: Literal['plain', 'wrap'] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

109 return_type: Any 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

110 when_used: core_schema.WhenUsed 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

111 check_fields: bool | None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

112 

113 

114@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

115class ModelSerializerDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

116 """A container for data from `@model_serializer` so that we can access it 

117 while building the pydantic-core schema. 

118 

119 Attributes: 

120 decorator_repr: A class variable representing the decorator string, '@model_serializer'. 

121 mode: The proposed serializer mode. 

122 return_type: The type of the serializer's return value. 

123 when_used: The serialization condition. Accepts a string with values `'always'`, `'unless-none'`, `'json'`, 

124 and `'json-unless-none'`. 

125 """ 

126 

127 decorator_repr: ClassVar[str] = '@model_serializer' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

128 mode: Literal['plain', 'wrap'] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

129 return_type: Any 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

130 when_used: core_schema.WhenUsed 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

131 

132 

133@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

134class ModelValidatorDecoratorInfo: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

135 """A container for data from `@model_validator` so that we can access it 

136 while building the pydantic-core schema. 

137 

138 Attributes: 

139 decorator_repr: A class variable representing the decorator string, '@model_validator'. 

140 mode: The proposed serializer mode. 

141 """ 

142 

143 decorator_repr: ClassVar[str] = '@model_validator' 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

144 mode: Literal['wrap', 'before', 'after'] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

145 

146 

147DecoratorInfo: TypeAlias = """Union[ 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

148 ValidatorDecoratorInfo, 

149 FieldValidatorDecoratorInfo, 

150 RootValidatorDecoratorInfo, 

151 FieldSerializerDecoratorInfo, 

152 ModelSerializerDecoratorInfo, 

153 ModelValidatorDecoratorInfo, 

154 ComputedFieldInfo, 

155]""" 

156 

157ReturnType = TypeVar('ReturnType') 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

158DecoratedType: TypeAlias = ( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

159 'Union[classmethod[Any, Any, ReturnType], staticmethod[Any, ReturnType], Callable[..., ReturnType], property]' 

160) 

161 

162 

163@dataclass # can't use slots here since we set attributes on `__post_init__` 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

164class PydanticDescriptorProxy(Generic[ReturnType]): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

165 """Wrap a classmethod, staticmethod, property or unbound function 

166 and act as a descriptor that allows us to detect decorated items 

167 from the class' attributes. 

168 

169 This class' __get__ returns the wrapped item's __get__ result, 

170 which makes it transparent for classmethods and staticmethods. 

171 

172 Attributes: 

173 wrapped: The decorator that has to be wrapped. 

174 decorator_info: The decorator info. 

175 shim: A wrapper function to wrap V1 style function. 

176 """ 

177 

178 wrapped: DecoratedType[ReturnType] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

179 decorator_info: DecoratorInfo 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

180 shim: Callable[[Callable[..., Any]], Callable[..., Any]] | None = None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

181 

182 def __post_init__(self): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

183 for attr in 'setter', 'deleter': 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

184 if hasattr(self.wrapped, attr): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

185 f = partial(self._call_wrapped_attr, name=attr) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

186 setattr(self, attr, f) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

187 

188 def _call_wrapped_attr(self, func: Callable[[Any], None], *, name: str) -> PydanticDescriptorProxy[ReturnType]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

189 self.wrapped = getattr(self.wrapped, name)(func) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

190 if isinstance(self.wrapped, property): 190 ↛ 196line 190 didn't jump to line 196 because the condition on line 190 was always true41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

191 # update ComputedFieldInfo.wrapped_property 

192 from ..fields import ComputedFieldInfo 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

193 

194 if isinstance(self.decorator_info, ComputedFieldInfo): 194 ↛ 196line 194 didn't jump to line 196 because the condition on line 194 was always true41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

195 self.decorator_info.wrapped_property = self.wrapped 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

196 return self 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

197 

198 def __get__(self, obj: object | None, obj_type: type[object] | None = None) -> PydanticDescriptorProxy[ReturnType]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

199 try: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

200 return self.wrapped.__get__(obj, obj_type) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

201 except AttributeError: 26 ctx1stabcdefzyuvghijklwxmnopqr

202 # not a descriptor, e.g. a partial object 

203 return self.wrapped # type: ignore[return-value] 26 ctx1stabcdefzyuvghijklwxmnopqr

204 

205 def __set_name__(self, instance: Any, name: str) -> None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

206 if hasattr(self.wrapped, '__set_name__'): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

207 self.wrapped.__set_name__(instance, name) # pyright: ignore[reportFunctionMemberAccess] 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

208 

209 def __getattr__(self, name: str, /) -> Any: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

210 """Forward checks for __isabstractmethod__ and such.""" 

211 return getattr(self.wrapped, name) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

212 

213 

214DecoratorInfoType = TypeVar('DecoratorInfoType', bound=DecoratorInfo) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

215 

216 

217@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

218class Decorator(Generic[DecoratorInfoType]): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

219 """A generic container class to join together the decorator metadata 

220 (metadata from decorator itself, which we have when the 

221 decorator is called but not when we are building the core-schema) 

222 and the bound function (which we have after the class itself is created). 

223 

224 Attributes: 

225 cls_ref: The class ref. 

226 cls_var_name: The decorated function name. 

227 func: The decorated function. 

228 shim: A wrapper function to wrap V1 style function. 

229 info: The decorator info. 

230 """ 

231 

232 cls_ref: str 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

233 cls_var_name: str 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

234 func: Callable[..., Any] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

235 shim: Callable[[Any], Any] | None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

236 info: DecoratorInfoType 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

237 

238 @staticmethod 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

239 def build( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

240 cls_: Any, 

241 *, 

242 cls_var_name: str, 

243 shim: Callable[[Any], Any] | None, 

244 info: DecoratorInfoType, 

245 ) -> Decorator[DecoratorInfoType]: 

246 """Build a new decorator. 

247 

248 Args: 

249 cls_: The class. 

250 cls_var_name: The decorated function name. 

251 shim: A wrapper function to wrap V1 style function. 

252 info: The decorator info. 

253 

254 Returns: 

255 The new decorator instance. 

256 """ 

257 func = get_attribute_from_bases(cls_, cls_var_name) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

258 if shim is not None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

259 func = shim(func) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

260 func = unwrap_wrapped_function(func, unwrap_partial=False) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

261 if not callable(func): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

262 # This branch will get hit for classmethod properties 

263 attribute = get_attribute_from_base_dicts(cls_, cls_var_name) # prevents the binding call to `__get__` 26 ctx1stabcdefzyuvghijklwxmnopqr

264 if isinstance(attribute, PydanticDescriptorProxy): 264 ↛ 266line 264 didn't jump to line 266 because the condition on line 264 was always true26 ctx1stabcdefzyuvghijklwxmnopqr

265 func = unwrap_wrapped_function(attribute.wrapped) 26 ctx1stabcdefzyuvghijklwxmnopqr

266 return Decorator( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

267 cls_ref=get_type_ref(cls_), 

268 cls_var_name=cls_var_name, 

269 func=func, 

270 shim=shim, 

271 info=info, 

272 ) 

273 

274 def bind_to_cls(self, cls: Any) -> Decorator[DecoratorInfoType]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

275 """Bind the decorator to a class. 

276 

277 Args: 

278 cls: the class. 

279 

280 Returns: 

281 The new decorator instance. 

282 """ 

283 return self.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

284 cls, 

285 cls_var_name=self.cls_var_name, 

286 shim=self.shim, 

287 info=self.info, 

288 ) 

289 

290 

291def get_bases(tp: type[Any]) -> tuple[type[Any], ...]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

292 """Get the base classes of a class or typeddict. 

293 

294 Args: 

295 tp: The type or class to get the bases. 

296 

297 Returns: 

298 The base classes. 

299 """ 

300 if is_typeddict(tp): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

301 return tp.__orig_bases__ # type: ignore 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

302 try: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

303 return tp.__bases__ 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

304 except AttributeError: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

305 return () 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

306 

307 

308def mro(tp: type[Any]) -> tuple[type[Any], ...]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

309 """Calculate the Method Resolution Order of bases using the C3 algorithm. 

310 

311 See https://www.python.org/download/releases/2.3/mro/ 

312 """ 

313 # try to use the existing mro, for performance mainly 

314 # but also because it helps verify the implementation below 

315 if not is_typeddict(tp): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

316 try: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

317 return tp.__mro__ 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

318 except AttributeError: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

319 # GenericAlias and some other cases 

320 pass 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

321 

322 bases = get_bases(tp) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

323 return (tp,) + mro_for_bases(bases) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

324 

325 

326def mro_for_bases(bases: tuple[type[Any], ...]) -> tuple[type[Any], ...]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

327 def merge_seqs(seqs: list[deque[type[Any]]]) -> Iterable[type[Any]]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

328 while True: 35 ctx1abcdefABCDEyghijklFGHIJPmnopqrKLMNO

329 non_empty = [seq for seq in seqs if seq] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

330 if not non_empty: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

331 # Nothing left to process, we're done. 

332 return 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

333 candidate: type[Any] | None = None 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

334 for seq in non_empty: # Find merge candidates among seq heads. 334 ↛ 342line 334 didn't jump to line 342 because the loop on line 334 didn't complete42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

335 candidate = seq[0] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

336 not_head = [s for s in non_empty if candidate in islice(s, 1, None)] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

337 if not_head: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

338 # Reject the candidate. 

339 candidate = None 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

340 else: 

341 break 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

342 if not candidate: 342 ↛ 343line 342 didn't jump to line 343 because the condition on line 342 was never true42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

343 raise TypeError('Inconsistent hierarchy, no C3 MRO is possible') 

344 yield candidate 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

345 for seq in non_empty: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

346 # Remove candidate. 

347 if seq[0] == candidate: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

348 seq.popleft() 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

349 

350 seqs = [deque(mro(base)) for base in bases] + [deque(bases)] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

351 return tuple(merge_seqs(seqs)) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

352 

353 

354_sentinel = object() 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

355 

356 

357def get_attribute_from_bases(tp: type[Any] | tuple[type[Any], ...], name: str) -> Any: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

358 """Get the attribute from the next class in the MRO that has it, 

359 aiming to simulate calling the method on the actual class. 

360 

361 The reason for iterating over the mro instead of just getting 

362 the attribute (which would do that for us) is to support TypedDict, 

363 which lacks a real __mro__, but can have a virtual one constructed 

364 from its bases (as done here). 

365 

366 Args: 

367 tp: The type or class to search for the attribute. If a tuple, this is treated as a set of base classes. 

368 name: The name of the attribute to retrieve. 

369 

370 Returns: 

371 Any: The attribute value, if found. 

372 

373 Raises: 

374 AttributeError: If the attribute is not found in any class in the MRO. 

375 """ 

376 if isinstance(tp, tuple): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

377 for base in mro_for_bases(tp): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

378 attribute = base.__dict__.get(name, _sentinel) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

379 if attribute is not _sentinel: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

380 attribute_get = getattr(attribute, '__get__', None) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

381 if attribute_get is not None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

382 return attribute_get(None, tp) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

383 return attribute 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

384 raise AttributeError(f'{name} not found in {tp}') 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

385 else: 

386 try: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

387 return getattr(tp, name) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

388 except AttributeError: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

389 return get_attribute_from_bases(mro(tp), name) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

390 

391 

392def get_attribute_from_base_dicts(tp: type[Any], name: str) -> Any: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

393 """Get an attribute out of the `__dict__` following the MRO. 

394 This prevents the call to `__get__` on the descriptor, and allows 

395 us to get the original function for classmethod properties. 

396 

397 Args: 

398 tp: The type or class to search for the attribute. 

399 name: The name of the attribute to retrieve. 

400 

401 Returns: 

402 Any: The attribute value, if found. 

403 

404 Raises: 

405 KeyError: If the attribute is not found in any class's `__dict__` in the MRO. 

406 """ 

407 for base in reversed(mro(tp)): 407 ↛ 410line 407 didn't jump to line 410 because the loop on line 407 didn't complete26 ctx1stabcdefzyuvghijklwxmnopqr

408 if name in base.__dict__: 26 ctx1stabcdefzyuvghijklwxmnopqr

409 return base.__dict__[name] 26 ctx1stabcdefzyuvghijklwxmnopqr

410 return tp.__dict__[name] # raise the error 

411 

412 

413@dataclass(**slots_true) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

414class DecoratorInfos: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

415 """Mapping of name in the class namespace to decorator info. 

416 

417 note that the name in the class namespace is the function or attribute name 

418 not the field name! 

419 """ 

420 

421 validators: dict[str, Decorator[ValidatorDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

422 field_validators: dict[str, Decorator[FieldValidatorDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

423 root_validators: dict[str, Decorator[RootValidatorDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

424 field_serializers: dict[str, Decorator[FieldSerializerDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

425 model_serializers: dict[str, Decorator[ModelSerializerDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

426 model_validators: dict[str, Decorator[ModelValidatorDecoratorInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

427 computed_fields: dict[str, Decorator[ComputedFieldInfo]] = field(default_factory=dict) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

428 

429 @staticmethod 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

430 def build(model_dc: type[Any]) -> DecoratorInfos: # noqa: C901 (ignore complexity) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

431 """We want to collect all DecFunc instances that exist as 

432 attributes in the namespace of the class (a BaseModel or dataclass) 

433 that called us 

434 But we want to collect these in the order of the bases 

435 So instead of getting them all from the leaf class (the class that called us), 

436 we traverse the bases from root (the oldest ancestor class) to leaf 

437 and collect all of the instances as we go, taking care to replace 

438 any duplicate ones with the last one we see to mimic how function overriding 

439 works with inheritance. 

440 If we do replace any functions we put the replacement into the position 

441 the replaced function was in; that is, we maintain the order. 

442 """ 

443 # reminder: dicts are ordered and replacement does not alter the order 

444 res = DecoratorInfos() 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

445 for base in reversed(mro(model_dc)[1:]): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

446 existing: DecoratorInfos | None = base.__dict__.get('__pydantic_decorators__') 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

447 if existing is None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

448 existing = DecoratorInfos.build(base) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

449 res.validators.update({k: v.bind_to_cls(model_dc) for k, v in existing.validators.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

450 res.field_validators.update({k: v.bind_to_cls(model_dc) for k, v in existing.field_validators.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

451 res.root_validators.update({k: v.bind_to_cls(model_dc) for k, v in existing.root_validators.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

452 res.field_serializers.update({k: v.bind_to_cls(model_dc) for k, v in existing.field_serializers.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

453 res.model_serializers.update({k: v.bind_to_cls(model_dc) for k, v in existing.model_serializers.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

454 res.model_validators.update({k: v.bind_to_cls(model_dc) for k, v in existing.model_validators.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

455 res.computed_fields.update({k: v.bind_to_cls(model_dc) for k, v in existing.computed_fields.items()}) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

456 

457 to_replace: list[tuple[str, Any]] = [] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

458 

459 for var_name, var_value in vars(model_dc).items(): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

460 if isinstance(var_value, PydanticDescriptorProxy): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

461 info = var_value.decorator_info 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

462 if isinstance(info, ValidatorDecoratorInfo): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

463 res.validators[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

464 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

465 ) 

466 elif isinstance(info, FieldValidatorDecoratorInfo): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

467 res.field_validators[var_name] = Decorator.build( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

468 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

469 ) 

470 elif isinstance(info, RootValidatorDecoratorInfo): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

471 res.root_validators[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

472 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

473 ) 

474 elif isinstance(info, FieldSerializerDecoratorInfo): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

475 # check whether a serializer function is already registered for fields 

476 for field_serializer_decorator in res.field_serializers.values(): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

477 # check that each field has at most one serializer function. 

478 # serializer functions for the same field in subclasses are allowed, 

479 # and are treated as overrides 

480 if field_serializer_decorator.cls_var_name == var_name: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

481 continue 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

482 for f in info.fields: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

483 if f in field_serializer_decorator.info.fields: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

484 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

485 'Multiple field serializer functions were defined ' 

486 f'for field {f!r}, this is not allowed.', 

487 code='multiple-field-serializers', 

488 ) 

489 res.field_serializers[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

490 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

491 ) 

492 elif isinstance(info, ModelValidatorDecoratorInfo): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

493 res.model_validators[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

494 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

495 ) 

496 elif isinstance(info, ModelSerializerDecoratorInfo): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

497 res.model_serializers[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

498 model_dc, cls_var_name=var_name, shim=var_value.shim, info=info 

499 ) 

500 else: 

501 from ..fields import ComputedFieldInfo 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

502 

503 isinstance(var_value, ComputedFieldInfo) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

504 res.computed_fields[var_name] = Decorator.build( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

505 model_dc, cls_var_name=var_name, shim=None, info=info 

506 ) 

507 to_replace.append((var_name, var_value.wrapped)) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

508 if to_replace: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

509 # If we can save `__pydantic_decorators__` on the class we'll be able to check for it above 

510 # so then we don't need to re-process the type, which means we can discard our descriptor wrappers 

511 # and replace them with the thing they are wrapping (see the other setattr call below) 

512 # which allows validator class methods to also function as regular class methods 

513 model_dc.__pydantic_decorators__ = res 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

514 for name, value in to_replace: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

515 setattr(model_dc, name, value) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

516 return res 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

517 

518 def update_from_config(self, config_wrapper: ConfigWrapper) -> None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

519 """Update the decorator infos from the configuration of the class they are attached to.""" 

520 for name, computed_field_dec in self.computed_fields.items(): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

521 computed_field_dec.info._update_from_config(config_wrapper, name) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

522 

523 

524def inspect_validator(validator: Callable[..., Any], mode: FieldValidatorModes) -> bool: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

525 """Look at a field or model validator function and determine whether it takes an info argument. 

526 

527 An error is raised if the function has an invalid signature. 

528 

529 Args: 

530 validator: The validator function to inspect. 

531 mode: The proposed validator mode. 

532 

533 Returns: 

534 Whether the validator takes an info argument. 

535 """ 

536 try: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

537 sig = signature(validator) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

538 except (ValueError, TypeError): 39 ctx1stabcdefABCDEuvghijklFGHIJwxmnopqrKLMNO

539 # `inspect.signature` might not be able to infer a signature, e.g. with C objects. 

540 # In this case, we assume no info argument is present: 

541 return False 39 ctx1stabcdefABCDEuvghijklFGHIJwxmnopqrKLMNO

542 n_positional = count_positional_required_params(sig) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

543 if mode == 'wrap': 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

544 if n_positional == 3: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

545 return True 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

546 elif n_positional == 2: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

547 return False 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

548 else: 

549 assert mode in {'before', 'after', 'plain'}, f"invalid mode: {mode!r}, expected 'before', 'after' or 'plain" 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

550 if n_positional == 2: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

551 return True 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

552 elif n_positional == 1: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

553 return False 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

554 

555 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

556 f'Unrecognized field_validator function signature for {validator} with `mode={mode}`:{sig}', 

557 code='validator-signature', 

558 ) 

559 

560 

561def inspect_field_serializer(serializer: Callable[..., Any], mode: Literal['plain', 'wrap']) -> tuple[bool, bool]: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

562 """Look at a field serializer function and determine if it is a field serializer, 

563 and whether it takes an info argument. 

564 

565 An error is raised if the function has an invalid signature. 

566 

567 Args: 

568 serializer: The serializer function to inspect. 

569 mode: The serializer mode, either 'plain' or 'wrap'. 

570 

571 Returns: 

572 Tuple of (is_field_serializer, info_arg). 

573 """ 

574 try: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

575 sig = signature(serializer) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

576 except (ValueError, TypeError): 

577 # `inspect.signature` might not be able to infer a signature, e.g. with C objects. 

578 # In this case, we assume no info argument is present and this is not a method: 

579 return (False, False) 

580 

581 first = next(iter(sig.parameters.values()), None) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

582 is_field_serializer = first is not None and first.name == 'self' 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

583 

584 n_positional = count_positional_required_params(sig) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

585 if is_field_serializer: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

586 # -1 to correct for self parameter 

587 info_arg = _serializer_info_arg(mode, n_positional - 1) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

588 else: 

589 info_arg = _serializer_info_arg(mode, n_positional) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

590 

591 if info_arg is None: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

592 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

593 f'Unrecognized field_serializer function signature for {serializer} with `mode={mode}`:{sig}', 

594 code='field-serializer-signature', 

595 ) 

596 

597 return is_field_serializer, info_arg 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

598 

599 

600def inspect_annotated_serializer(serializer: Callable[..., Any], mode: Literal['plain', 'wrap']) -> bool: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

601 """Look at a serializer function used via `Annotated` and determine whether it takes an info argument. 

602 

603 An error is raised if the function has an invalid signature. 

604 

605 Args: 

606 serializer: The serializer function to check. 

607 mode: The serializer mode, either 'plain' or 'wrap'. 

608 

609 Returns: 

610 info_arg 

611 """ 

612 try: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

613 sig = signature(serializer) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

614 except (ValueError, TypeError): 39 ctx1stabcdefABCDEuvghijklFGHIJwxmnopqrKLMNO

615 # `inspect.signature` might not be able to infer a signature, e.g. with C objects. 

616 # In this case, we assume no info argument is present: 

617 return False 39 ctx1stabcdefABCDEuvghijklFGHIJwxmnopqrKLMNO

618 info_arg = _serializer_info_arg(mode, count_positional_required_params(sig)) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

619 if info_arg is None: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

620 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

621 f'Unrecognized field_serializer function signature for {serializer} with `mode={mode}`:{sig}', 

622 code='field-serializer-signature', 

623 ) 

624 else: 

625 return info_arg 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

626 

627 

628def inspect_model_serializer(serializer: Callable[..., Any], mode: Literal['plain', 'wrap']) -> bool: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

629 """Look at a model serializer function and determine whether it takes an info argument. 

630 

631 An error is raised if the function has an invalid signature. 

632 

633 Args: 

634 serializer: The serializer function to check. 

635 mode: The serializer mode, either 'plain' or 'wrap'. 

636 

637 Returns: 

638 `info_arg` - whether the function expects an info argument. 

639 """ 

640 if isinstance(serializer, (staticmethod, classmethod)) or not is_instance_method_from_sig(serializer): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

641 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

642 '`@model_serializer` must be applied to instance methods', code='model-serializer-instance-method' 

643 ) 

644 

645 sig = signature(serializer) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

646 info_arg = _serializer_info_arg(mode, count_positional_required_params(sig)) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

647 if info_arg is None: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

648 raise PydanticUserError( 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

649 f'Unrecognized model_serializer function signature for {serializer} with `mode={mode}`:{sig}', 

650 code='model-serializer-signature', 

651 ) 

652 else: 

653 return info_arg 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

654 

655 

656def _serializer_info_arg(mode: Literal['plain', 'wrap'], n_positional: int) -> bool | None: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

657 if mode == 'plain': 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

658 if n_positional == 1: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

659 # (input_value: Any, /) -> Any 

660 return False 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

661 elif n_positional == 2: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

662 # (model: Any, input_value: Any, /) -> Any 

663 return True 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

664 else: 

665 assert mode == 'wrap', f"invalid mode: {mode!r}, expected 'plain' or 'wrap'" 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

666 if n_positional == 2: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

667 # (input_value: Any, serializer: SerializerFunctionWrapHandler, /) -> Any 

668 return False 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

669 elif n_positional == 3: 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

670 # (input_value: Any, serializer: SerializerFunctionWrapHandler, info: SerializationInfo, /) -> Any 

671 return True 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

672 

673 return None 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

674 

675 

676AnyDecoratorCallable: TypeAlias = ( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

677 'Union[classmethod[Any, Any, Any], staticmethod[Any, Any], partialmethod[Any], Callable[..., Any]]' 

678) 

679 

680 

681def is_instance_method_from_sig(function: AnyDecoratorCallable) -> bool: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

682 """Whether the function is an instance method. 

683 

684 It will consider a function as instance method if the first parameter of 

685 function is `self`. 

686 

687 Args: 

688 function: The function to check. 

689 

690 Returns: 

691 `True` if the function is an instance method, `False` otherwise. 

692 """ 

693 sig = signature(unwrap_wrapped_function(function)) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

694 first = next(iter(sig.parameters.values()), None) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

695 if first and first.name == 'self': 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

696 return True 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

697 return False 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

698 

699 

700def ensure_classmethod_based_on_signature(function: AnyDecoratorCallable) -> Any: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

701 """Apply the `@classmethod` decorator on the function. 

702 

703 Args: 

704 function: The function to apply the decorator on. 

705 

706 Return: 

707 The `@classmethod` decorator applied function. 

708 """ 

709 if not isinstance( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

710 unwrap_wrapped_function(function, unwrap_class_static_method=False), classmethod 

711 ) and _is_classmethod_from_sig(function): 

712 return classmethod(function) # type: ignore[arg-type] 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

713 return function 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

714 

715 

716def _is_classmethod_from_sig(function: AnyDecoratorCallable) -> bool: 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

717 sig = signature(unwrap_wrapped_function(function)) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

718 first = next(iter(sig.parameters.values()), None) 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

719 if first and first.name == 'cls': 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

720 return True 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

721 return False 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

722 

723 

724def unwrap_wrapped_function( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

725 func: Any, 

726 *, 

727 unwrap_partial: bool = True, 

728 unwrap_class_static_method: bool = True, 

729) -> Any: 

730 """Recursively unwraps a wrapped function until the underlying function is reached. 

731 This handles property, functools.partial, functools.partialmethod, staticmethod, and classmethod. 

732 

733 Args: 

734 func: The function to unwrap. 

735 unwrap_partial: If True (default), unwrap partial and partialmethod decorators. 

736 unwrap_class_static_method: If True (default), also unwrap classmethod and staticmethod 

737 decorators. If False, only unwrap partial and partialmethod decorators. 

738 

739 Returns: 

740 The underlying function of the wrapped function. 

741 """ 

742 # Define the types we want to check against as a single tuple. 

743 unwrap_types = ( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

744 (property, cached_property) 

745 + ((partial, partialmethod) if unwrap_partial else ()) 

746 + ((staticmethod, classmethod) if unwrap_class_static_method else ()) 

747 ) 

748 

749 while isinstance(func, unwrap_types): 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

750 if unwrap_class_static_method and isinstance(func, (classmethod, staticmethod)): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

751 func = func.__func__ 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

752 elif isinstance(func, (partial, partialmethod)): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

753 func = func.func 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

754 elif isinstance(func, property): 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

755 func = func.fget # arbitrary choice, convenient for computed fields 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

756 else: 

757 # Make coverage happy as it can only get here in the last possible case 

758 assert isinstance(func, cached_property) 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

759 func = func.func # type: ignore 41 ctx1stabcdefABCDEzyuvghijklFGHIJwxmnopqrKLMNO

760 

761 return func 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

762 

763 

764_function_like = ( 42 ctx1stabcdefABCDEzyuvghijklFGHIJPwxmnopqrKLMNO

765 partial, 

766 partialmethod, 

767 types.FunctionType, 

768 types.BuiltinFunctionType, 

769 types.MethodType, 

770 types.WrapperDescriptorType, 

771 types.MethodWrapperType, 

772 types.MemberDescriptorType, 

773) 

774 

775