fix View support

JGLTechnologies · Jul 13, 2023 · c457098 · c457098
1 parent 7fce396
commit c457098
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 6 deletions.
diff --git a/aiohttp_ratelimiter.egg-info/PKG-INFO b/aiohttp_ratelimiter.egg-info/PKG-INFO
@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aiohttp-ratelimiter
-Version: 4.1.1
+Version: 4.1.2
 Summary: A simple rate limiter for aiohttp.web
 Home-page: https://jgltechnologies.com/aiohttplimiter
 Author: George Luca
@@ -144,7 +144,7 @@ Views Example
 @routes.view("/")
 class Home(View):
     @limiter.limit("1/second")
-    def get(self: web.Request):
+    def get(self):
         return web.Response(text="hello")
 ```
 

diff --git a/aiohttplimiter/limiter.py b/aiohttplimiter/limiter.py
@@ -8,12 +8,14 @@
 from limits import parse
 
 
-def default_keyfunc(request: Union[Request, View]) -> str:
+def default_keyfunc(ctx: Union[Request, View]) -> str:
     """
     Returns the user's IP
     """
-    if isinstance(request, View):
-        request = request.request
+    if isinstance(ctx, View):
+        request = ctx.request
+    else:
+        request = ctx
     ip = request.headers.get(
         "X-Forwarded-For") or request.remote or "127.0.0.1"
     ip = ip.split(",")[0]

diff --git a/dist/aiohttp-ratelimiter-4.1.1.tar.gz b/dist/aiohttp-ratelimiter-4.1.1.tar.gz
diff --git a/setup.py b/setup.py
@@ -10,7 +10,7 @@ def get_long_description():
         return file.read()
 
 
-VERSION = "4.1.1"
+VERSION = "4.1.2"
 
 classifiers = [
     "Development Status :: 5 - Production/Stable",