Tobias/func.py
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

last changes

Browse Source
This commit is contained in:
Tobias Weise 2025-07-14 08:56:44 +02:00
parent da946981b1
commit 3c3580fc07
1 changed files with 27 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
func_py/func.py
View File

@ -40,7 +40,7 @@ class function:
sig_g = signature(g) sig_g = signature(g)
g_req_params = [v for k, v in sig_g.parameters.items() if v.default == _empty] g_req_params = [v for k, v in sig_g.parameters.items() if v.default == _empty]
g_typed_req_params = [p if p.annotation == _empty else Parameter(name=p.name, annotation=Any, kind=Parameter.POSITIONAL_ONLY) for p in g_req_params] g_typed_req_params = [p if p.annotation == _empty else Parameter(name=p.name, annotation=Any, kind=Parameter.POSITIONAL_OR_KEYWORD) for p in g_req_params]
def h(*args, **kwargs): def h(*args, **kwargs):
return f(g(*args, **kwargs)) return f(g(*args, **kwargs))
@ -49,18 +49,33 @@ class function:
h.__signature__ = sig_g.replace(parameters=g_typed_req_params, return_annotation=Any if f_ret_type == _empty else f_ret_type) h.__signature__ = sig_g.replace(parameters=g_typed_req_params, return_annotation=Any if f_ret_type == _empty else f_ret_type)
return function(h) return function(h)
#TODO: complexity prop
#TODO: getter for inverse function
#TODO: implement fly weight pattern #TODO: implement fly weight pattern
def __init__(self, f): def __init__(self, f):
self.original_func = f self.original_func = f
self.signature = signature(f) #TODO: change to support pattern matching: sig ((int, str), {"name": str}) self.signature = signature(f) #TODO: change to support pattern matching: sig ((int, str), {"name": str})
self.__signature__ = signature(f)
sig = signature(f)
req_params = {k: v for k, v in sig.parameters.items() if v.default == _empty}
params = []
for name, p in req_params.items():
if p.annotation == _empty:
params.append( Parameter(name=name, annotation=Any, kind=Parameter.POSITIONAL_OR_KEYWORD) )
else:
params.append(p)
self.__signature__ = sig.replace(parameters=params, return_annotation=Any if sig.return_annotation == _empty else sig.return_annotation)
self.cache = {} self.cache = {}
self.is_pure = True self.is_pure = True
def __str__(self): def __str__(self):
return str(signature(self.original_func)).replace(", /)", ")") return str(self.__signature__)
def __repr__(self): def __repr__(self):
return str(self) return str(self)
@ -77,14 +92,17 @@ class function:
@property @property
def required_args(self): def required_args(self):
return [(k, v) for k, v in signature(self.original_func).parameters.items() if v.default == _empty] return [(name, param) for name, param in self.__signature__.parameters.items() if param.default == _empty]
def __call__(self, *args, **kwargs): def __call__(self, *args, **kwargs):
if len(args) < len(self.required_args): provided_req_kwargs = {k for k, p in self.required_args if k in kwargs}
sig = signature(self.original_func) num_given_req_args = len(args) + len(provided_req_kwargs)
f = partial(self.original_func, *args)
req_params = [v for k, v in sig.parameters.items() if v.default == _empty] if num_given_req_args < len(self.required_args):
remaining_params = req_params[len(args):] sig = self.__signature__
f = partial(self.original_func, *args, **kwargs)
remaining_params = [v for k, v in sig.parameters.items() if v.default == _empty and k not in provided_req_kwargs][len(args):]
f.__signature__ = sig.replace(parameters=remaining_params, return_annotation=Any if sig.return_annotation == _empty else sig.return_annotation) f.__signature__ = sig.replace(parameters=remaining_params, return_annotation=Any if sig.return_annotation == _empty else sig.return_annotation)
return function(f) return function(f)