Simplify exempt decorator using functools.partial

alisaifee · Jan 5, 2025 · 807d8a1 · 807d8a1
1 parent 2b223b5
commit 807d8a1
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/flask_limiter/extension.py b/flask_limiter/extension.py
@@ -6,6 +6,7 @@
 
 import dataclasses
 import datetime
+import functools
 import itertools
 import logging
 import time
@@ -39,7 +40,6 @@
     Sequence,
     Set,
     Tuple,
-    TypeVar,
     Union,
     cast,
 )
@@ -730,13 +730,8 @@ def route(...):
         elif obj:
             self.limit_manager.add_route_exemption(get_qualified_name(obj), flags)
         else:
-            _R = TypeVar("_R")
-            _WO = TypeVar("_WO", Callable[..., _R], flask.Blueprint)
+            return functools.partial(self.exempt, flags=flags)
 
-            def wrapper(obj: _WO) -> _WO:
-                return self.exempt(obj, flags=flags)
-
-            return wrapper
         return obj
 
     def request_filter(self, fn: Callable[[], bool]) -> Callable[[], bool]: