Use functools.wraps instead of decorator

Signed-off-by: Yury Pliner <yury.pliner@gmail.com>
Signed-off-by: Iurii Pliner <yury.pliner@gmail.com>

Author: Pliner

prometheus_client/context_managers.py

@@ -1,12 +1,11 @@
+import functools
 from timeit import default_timer
 from types import TracebackType
 from typing import (
     Any, Callable, Literal, Optional, Tuple, Type, TYPE_CHECKING, TypeVar,
     Union,
 )
 
-from .decorator import decorate
-
 if TYPE_CHECKING:
     from . import Counter
     F = TypeVar("F", bound=Callable[..., Any])
@@ -26,11 +25,11 @@ def __exit__(self, typ: Optional[Type[BaseException]], value: Optional[BaseExcep
         return False
 
     def __call__(self, f: "F") -> "F":
-        def wrapped(func, *args, **kwargs):
+        @functools.wraps(f)
+        def wrapped(*args: Any, **kwargs: Any) -> Any:
             with self:
-                return func(*args, **kwargs)
-
-        return decorate(f, wrapped)
+                return f(*args, **kwargs)
+        return wrapped
 
 
 class InprogressTracker:
@@ -44,11 +43,11 @@ def __exit__(self, typ, value, traceback):
         self._gauge.dec()
 
     def __call__(self, f: "F") -> "F":
-        def wrapped(func, *args, **kwargs):
+        @functools.wraps(f)
+        def wrapped(*args: Any, **kwargs: Any) -> Any:
             with self:
-                return func(*args, **kwargs)
-
-        return decorate(f, wrapped)
+                return f(*args, **kwargs)
+        return wrapped
 
 
 class Timer:
@@ -73,10 +72,10 @@ def labels(self, *args, **kw):
         self._metric = self._metric.labels(*args, **kw)
 
     def __call__(self, f: "F") -> "F":
-        def wrapped(func, *args, **kwargs):
+        @functools.wraps(f)
+        def wrapped(*args: Any, **kwargs: Any) -> Any:
             # Obtaining new instance of timer every time
             # ensures thread safety and reentrancy.
             with self._new_timer():
-                return func(*args, **kwargs)
-
-        return decorate(f, wrapped)
+                return f(*args, **kwargs)
+        return wrapped