Use AttrName enum when attribute names are needed

alexdelorenzo · Jan 29, 2022 · 84d2ab4 · 84d2ab4
1 parent 620d30f
commit 84d2ab4
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/limiter/limiter.py b/limiter/limiter.py
@@ -2,7 +2,7 @@
 
 from typing import (
   AsyncContextManager, ContextManager, Awaitable, TypedDict,
-  Final, cast
+  cast
 )
 from contextlib import (
   AbstractContextManager, AbstractAsyncContextManager,
@@ -27,9 +27,6 @@
 )
 
 
-LIM_KEY: Final[str] = 'limiter'
-
-
 class Attrs(TypedDict):
   consume: Tokens
   bucket: BucketName
@@ -128,7 +125,7 @@ def __call__(
   @property
   def attrs(self) -> Attrs:
     attrs = asdict(self)
-    attrs.pop(LIM_KEY, None)
+    attrs.pop(AttrName.limiter, None)
 
     return attrs