Update 2025-04-24_11:44:19

venv/lib/python3.11/site-packages/limits/storage/registry.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+from abc import ABCMeta
+
+SCHEMES: dict[str, StorageRegistry] = {}
+
+
+class StorageRegistry(ABCMeta):
+    def __new__(
+        mcs, name: str, bases: tuple[type, ...], dct: dict[str, str | list[str]]
+    ) -> StorageRegistry:
+        storage_scheme = dct.get("STORAGE_SCHEME", None)
+        cls = super().__new__(mcs, name, bases, dct)
+
+        if storage_scheme:
+            if isinstance(storage_scheme, str):  # noqa
+                schemes = [storage_scheme]
+            else:
+                schemes = storage_scheme
+
+            for scheme in schemes:
+                SCHEMES[scheme] = cls
+
+        return cls