1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
|
from __future__ import annotations
import contextlib
import inspect
from typing import Any, Callable
from polyfactory import PostGenerated
class post_generated: # noqa: N801
"""Descriptor class for wrapping a classmethod into a ``PostGenerated`` field."""
__slots__ = ("cache", "method")
def __init__(self, method: Callable | classmethod) -> None:
if not isinstance(method, classmethod):
msg = "post_generated decorator can only be used on classmethods"
raise TypeError(msg)
self.method = method
self.cache: dict[type, PostGenerated] = {}
def __get__(self, obj: Any, objtype: type) -> PostGenerated:
with contextlib.suppress(KeyError):
return self.cache[objtype]
fn = self.method.__func__ # pyright: ignore[reportFunctionMemberAccess]
fn_args = inspect.getfullargspec(fn).args[1:]
def new_fn(name: str, values: dict[str, Any]) -> Any: # noqa: ARG001 - investigate @guacs
return fn(objtype, **{arg: values[arg] for arg in fn_args if arg in values})
return self.cache.setdefault(objtype, PostGenerated(new_fn))
|
