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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
from __future__ import annotations
from collections.abc import Callable
from typing import (
Concatenate,
Generic,
ParamSpec,
TypeVar,
cast,
)
from ._base import CRU
_P = ParamSpec("_P")
_T = TypeVar("_T")
_O = TypeVar("_O")
_R = TypeVar("_R")
class CruDecorator:
class ConvertResult(Generic[_T, _O]):
def __init__(
self,
converter: Callable[[_T], _O],
) -> None:
self.converter = converter
def __call__(self, origin: Callable[_P, _T]) -> Callable[_P, _O]:
converter = self.converter
def real_impl(*args: _P.args, **kwargs: _P.kwargs) -> _O:
return converter(origin(*args, **kwargs))
return real_impl
class ImplementedBy(Generic[_T, _O, _P, _R]):
def __init__(
self,
impl: Callable[Concatenate[_O, _P], _R],
converter: Callable[[_T], _O],
) -> None:
self.impl = impl
self.converter = converter
def __call__(
self, _origin: Callable[[_T], None]
) -> Callable[Concatenate[_T, _P], _R]:
converter = self.converter
impl = self.impl
def real_impl(_self: _T, *args: _P.args, **kwargs: _P.kwargs) -> _R:
return cast(Callable[Concatenate[_O, _P], _R], impl)(
converter(_self), *args, **kwargs
)
return real_impl
@staticmethod
def create_factory(converter: Callable[[_T], _O]) -> Callable[
[Callable[Concatenate[_O, _P], _R]],
CruDecorator.ImplementedBy[_T, _O, _P, _R],
]:
def create(
m: Callable[Concatenate[_O, _P], _R],
) -> CruDecorator.ImplementedBy[_T, _O, _P, _R]:
return CruDecorator.ImplementedBy(m, converter)
return create
class ImplementedByNoSelf(Generic[_P, _R]):
def __init__(self, impl: Callable[_P, _R]) -> None:
self.impl = impl
def __call__(
self, _origin: Callable[[_T], None]
) -> Callable[Concatenate[_T, _P], _R]:
impl = self.impl
def real_impl(_self: _T, *args: _P.args, **kwargs: _P.kwargs) -> _R:
return cast(Callable[_P, _R], impl)(*args, **kwargs)
return real_impl
@staticmethod
def create_factory() -> (
Callable[[Callable[_P, _R]], CruDecorator.ImplementedByNoSelf[_P, _R]]
):
def create(
m: Callable[_P, _R],
) -> CruDecorator.ImplementedByNoSelf[_P, _R]:
return CruDecorator.ImplementedByNoSelf(m)
return create
CRU.add_objects(CruDecorator)
|
