From d9190be417f9fd68417de5029b5910f17d81722b Mon Sep 17 00:00:00 2001 From: Crutcher Dunnavant Date: Tue, 21 Mar 2023 11:19:10 -0700 Subject: [PATCH 1/2] Extend type forwardig to other context managers Signed-off-by: Crutcher Dunnavant --- prometheus_client/context_managers.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/prometheus_client/context_managers.py b/prometheus_client/context_managers.py index ef4db178..179d7f2e 100644 --- a/prometheus_client/context_managers.py +++ b/prometheus_client/context_managers.py @@ -46,7 +46,7 @@ def __enter__(self): def __exit__(self, typ, value, traceback): self._gauge.dec() - def __call__(self, f): + def __call__(self, f: "F") -> "F": def wrapped(func, *args, **kwargs): with self: return func(*args, **kwargs) @@ -75,7 +75,7 @@ def __exit__(self, typ, value, traceback): def labels(self, *args, **kw): self._metric = self._metric.labels(*args, **kw) - def __call__(self, f): + def __call__(self, f : "F") -> "F": def wrapped(func, *args, **kwargs): # Obtaining new instance of timer every time # ensures thread safety and reentrancy. From 1897e2bbb559e3f92c9083478dc339d1db226bb7 Mon Sep 17 00:00:00 2001 From: Crutcher Dunnavant Date: Tue, 21 Mar 2023 11:21:38 -0700 Subject: [PATCH 2/2] Fix lint. Signed-of-by: Crutcher Dunnavant Signed-off-by: Crutcher Dunnavant --- prometheus_client/context_managers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/prometheus_client/context_managers.py b/prometheus_client/context_managers.py index 179d7f2e..964a9304 100644 --- a/prometheus_client/context_managers.py +++ b/prometheus_client/context_managers.py @@ -75,7 +75,7 @@ def __exit__(self, typ, value, traceback): def labels(self, *args, **kw): self._metric = self._metric.labels(*args, **kw) - def __call__(self, f : "F") -> "F": + def __call__(self, f: "F") -> "F": def wrapped(func, *args, **kwargs): # Obtaining new instance of timer every time # ensures thread safety and reentrancy.