fixing code style

Author: eadwinCode

ellar_throttler/module.py

@@ -38,7 +38,7 @@ def setup(
                     use_value=ThrottlerModuleOptions(
                         limit=limit,
                         ttl=ttl,
-                        skip_if=skip_if,  # type:ignore[arg-type]
+                        skip_if=skip_if,
                     ),
                 ),
             ],

ellar_throttler/throttler_guard.py

@@ -34,10 +34,8 @@ async def can_activate(self, context: IExecutionContext) -> bool:
         # Return early if the current route should be skipped.
         # or self.options.skipIf?.(context)
 
-        if (
-            self.reflector.get_all_and_override(THROTTLER_SKIP, handler, class_ref)
-            or self.options.skip_if
-            and self.options.skip_if(context)
+        if self.reflector.get_all_and_override(THROTTLER_SKIP, handler, class_ref) or (
+            self.options.skip_if and self.options.skip_if(context)
         ):
             return True
 

ellar_throttler/throttler_module_options.py

@@ -1,11 +1,10 @@
-import dataclasses
 import typing as t
 
 from ellar.core import IExecutionContext
+from ellar.serializer import Serializer
 
 
-@dataclasses.dataclass
-class ThrottlerModuleOptions:
+class ThrottlerModuleOptions(Serializer):
     # The amount of requests that are allowed within the ttl's time window.
     limit: int
 
@@ -14,4 +13,4 @@ class ThrottlerModuleOptions:
 
     # A factory method to determine if throttling should be skipped.
     # This can be based on the incoming context.
-    skip_if: t.Callable[[IExecutionContext], bool] = None  # type:ignore[assignment]
+    skip_if: t.Optional[t.Callable[[IExecutionContext], bool]] = None

ellar_throttler/throttler_storage_options.py

@@ -1,7 +1,6 @@
-import dataclasses
+from ellar.serializer import Serializer
 
 
-@dataclasses.dataclass
-class ThrottlerStorageOption:
+class ThrottlerStorageOption(Serializer):
     total_hits: int
     expires_at: float

ellar_throttler/throttler_storage_record.py

@@ -1,8 +1,7 @@
-import dataclasses
+from ellar.serializer import Serializer
 
 
-@dataclasses.dataclass
-class ThrottlerStorageRecord:
+class ThrottlerStorageRecord(Serializer):
     #  Amount of requests done by a specific user (partially based on IP).
     total_hits: int
     #  Amount of seconds when the `total_hits` should expire.