Merge pull request #34 from python-ellar/ellar_070

Ellar 0.7.0 support

Author: eadwinCode

ellar_throttler/__init__.py

@@ -1,6 +1,6 @@
 """A rate limiting module for Ellar"""
 
-__version__ = "0.1.0"
+__version__ = "0.1.2"
 
 from .decorators import skip_throttle, throttle
 from .exception import ThrottledException

ellar_throttler/throttler_guard.py

@@ -2,10 +2,10 @@
 import typing as t
 
 from ellar.common import GuardCanActivate, IExecutionContext, Response
-from ellar.common.utils import get_name
 from ellar.core.connection import HTTPConnection
 from ellar.core.services import Reflector
 from ellar.di import injectable
+from ellar.utils import get_name
 
 from .constants import THROTTLER_LIMIT, THROTTLER_SKIP, THROTTLER_TTL
 from .exception import ThrottledException
@@ -60,8 +60,10 @@ def get_request_response(
         return connection_host.get_client(), connection_host.get_response()
 
     def get_tracker(self, connection: HTTPConnection) -> str:
-        assert connection.client
-        return connection.client.host
+        if connection.client:
+            return connection.client.host
+
+        return t.cast(str, connection.scope["server"][0])
 
     def generate_key(self, context: IExecutionContext, suffix: str) -> str:
         prefix = f"{get_name(context.get_class())}-{get_name(context.get_handler())}"

pyproject.toml

@@ -40,7 +40,7 @@ classifiers = [
 ]
 
 dependencies = [
-    "ellar >= 0.5.8"
+    "ellar >= 0.7.0"
 ]