Tweak typing hints

limits/strategies.py

@@ -4,7 +4,10 @@
 
 from abc import ABCMeta, abstractmethod
 import weakref
+from typing import Dict
 from typing import Tuple
+from typing import Type
+from typing import Union
 
 from .limits import RateLimitItem
 from .storage import Storage
@@ -183,7 +186,13 @@ def hit(self, item: RateLimitItem, *identifiers) -> bool:
         )
 
 
-STRATEGIES = {
+KnownStrategy = Union[
+    Type[FixedWindowRateLimiter],
+    Type[FixedWindowElasticExpiryRateLimiter],
+    Type[MovingWindowRateLimiter],
+]
+
+STRATEGIES: Dict[str, KnownStrategy] = {
     "fixed-window": FixedWindowRateLimiter,
     "fixed-window-elastic-expiry": FixedWindowElasticExpiryRateLimiter,
     "moving-window": MovingWindowRateLimiter,

setup.py

@@ -14,9 +14,11 @@
 THIS_DIR = os.path.abspath(os.path.dirname(__file__))
 REQUIREMENTS = [
     k
+
     for k in open(os.path.join(THIS_DIR, "requirements", "main.txt"))
     .read()
     .splitlines()
+
     if k.strip()
 ]
 
@@ -35,4 +37,7 @@
     long_description=open("README.rst").read(),
     packages=find_packages(exclude=["google.*", "google", "tests*"]),
     python_requires=">3.7",
+    package_data={
+        "limits": ["py.typed"],
+    },
 )