def trace[**P, T](func: Callable[P, T]) -> Callable[P, T]:
 @functools.wraps(func)
 def func_with_log(*args: P.args, **kwargs: P.kwargs) -> T:
 return func(*args, **kwargs)
 return func_with_log
class Foo:
 @trace
 def foo(self, a: int) -> str:
 return "foo"
Foo().foo(1)

class MyCallable[**P, T](Protocol):
 def __call__(self, *args: P.args, **kwargs: P.kwargs) -> T: ...
def trace[**P, T](func: MyCallable[P, T]) -> MyCallable[P, T]:
 @functools.wraps(func)
 def func_with_log(*args: P.args, **kwargs: P.kwargs) -> T:
 return func(*args, **kwargs)
 return func_with_log
class Foo:
 @trace
 def foo(self, a: int) -> str:
 return "foo"
Foo().foo(1)

error: Missing positional argument "a" in call to "__call__" of "MyCallable" [call-arg]
error: Argument 1 to "__call__" of "MyCallable" has incompatible type "int"; expected "Foo" [arg-type]

class MyCallable[**P, T](Protocol):
 def __call__(self_, self: Any, *args: P.args, **kwargs: P.kwargs) -> T: ...
 def __get__(self_, *args: Any, **kwargs: Any) -> MyCallableBound[P, T]: ...
class MyCallableBound[**P, T](Protocol):
 def __call__(self, *args: P.args, **kwargs: P.kwargs) -> T: ...
def trace[**P, T](func: MyCallable[P, T]) -> MyCallable[P, T]:
 @functools.wraps(func)
 def func_with_log(self: Any, *args: P.args, **kwargs: P.kwargs) -> T:
 return func(self, *args, **kwargs)
 return func_with_log
class Foo:
 @trace
 def foo(self, a: int) -> str:
 return "foo"
Foo().foo(1)

error: Incompatible return value type (got "_Wrapped[[Any, **P], T, [Any, **P], T]", expected "MyCallable[P, T]") [return-value]
note: "_Wrapped" is missing following "MyCallable" protocol member:
note: __get__

import contextlib
from typing import Callable, TypeVar
WRAPPER_ASSIGNMENTS = ('__module__', '__name__', '__qualname__', '__doc__',
 '__annotate__', '__type_params__')
WRAPPER_UPDATES = ('__dict__',)
def wraps[T](func: T) -> Callable[[T], T]:
 def decorator(new_func: T) -> T:
 for attr in WRAPPER_ASSIGNMENTS:
 with contextlib.suppress(AttributeError):
 setattr(new_func, attr, getattr(func, attr))
 for attr in WRAPPER_UPDATES:
 getattr(new_func, attr).update(getattr(func, attr, {}))
 setattr(new_func, "__wrapped__", func)
 return new_func
 return decorator