flask-Humanify 0.1.2__tar.gz → 0.1.3.1__tar.gz

{flask_humanify-0.1.2/flask_Humanify.egg-info → flask_humanify-0.1.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flask-Humanify
-Version: 0.1.2
+Version: 0.1.3.1
 Summary: Protect against bots and DDoS attacks
 Author-email: TN3W <tn3w@protonmail.com>
 License-Expression: Apache-2.0
@@ -42,19 +42,32 @@ from flask_Humanify import Humanify
 app = Flask(__name__)
 humanify = Humanify(app)
 
+# Register the middleware to deny access to bots
+humanify.register_middleware(action="deny_access")
+
 @app.route("/")
 def index():
     """
     A route that is protected against bots and DDoS attacks.
     """
-    if humanify.is_bot:
-        return humanify.redirect_to_access_denied()
     return "Hello, Human!"
 
 if __name__ == "__main__":
     app.run()
 ```
 
+Not using the middleware:
+```python
+@app.route("/")
+def index():
+    """
+    A route that is protected against bots and DDoS attacks.
+    """
+    if humanify.is_bot:
+        return humanify.deny_access()
+    return "Hello, Human!"
+```
+
 ## Usage
 
 ### Installation

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1}/README.md

@@ -11,19 +11,32 @@ from flask_Humanify import Humanify
 app = Flask(__name__)
 humanify = Humanify(app)
 
+# Register the middleware to deny access to bots
+humanify.register_middleware(action="deny_access")
+
 @app.route("/")
 def index():
     """
     A route that is protected against bots and DDoS attacks.
     """
-    if humanify.is_bot:
-        return humanify.redirect_to_access_denied()
     return "Hello, Human!"
 
 if __name__ == "__main__":
     app.run()
 ```
 
+Not using the middleware:
+```python
+@app.route("/")
+def index():
+    """
+    A route that is protected against bots and DDoS attacks.
+    """
+    if humanify.is_bot:
+        return humanify.deny_access()
+    return "Hello, Human!"
+```
+
 ## Usage
 
 ### Installation

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1/flask_Humanify.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flask-Humanify
-Version: 0.1.2
+Version: 0.1.3.1
 Summary: Protect against bots and DDoS attacks
 Author-email: TN3W <tn3w@protonmail.com>
 License-Expression: Apache-2.0
@@ -42,19 +42,32 @@ from flask_Humanify import Humanify
 app = Flask(__name__)
 humanify = Humanify(app)
 
+# Register the middleware to deny access to bots
+humanify.register_middleware(action="deny_access")
+
 @app.route("/")
 def index():
     """
     A route that is protected against bots and DDoS attacks.
     """
-    if humanify.is_bot:
-        return humanify.redirect_to_access_denied()
     return "Hello, Human!"
 
 if __name__ == "__main__":
     app.run()
 ```
 
+Not using the middleware:
+```python
+@app.route("/")
+def index():
+    """
+    A route that is protected against bots and DDoS attacks.
+    """
+    if humanify.is_bot:
+        return humanify.deny_access()
+    return "Hello, Human!"
+```
+
 ## Usage
 
 ### Installation

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1}/flask_humanify/__init__.py

@@ -4,7 +4,7 @@ Flask-Humanify
 A Flask extension that protects against bots and DDoS attacks.
 """
 
-__version__ = "0.1.2"
+__version__ = "0.1.3.1"
 
 from . import utils
 from .humanify import Humanify

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1}/flask_humanify/features/rate_limiter.py

@@ -51,7 +51,7 @@ class RateLimiter:
         Before request hook.
         """
         ip = get_client_ip(request)
-        if request.endpoint == "humanify.rate_limited":
+        if request.endpoint in ["humanify.rate_limited", "humanify.access_denied"]:
             return
         if self.is_rate_limited(ip or "127.0.0.1"):
             return redirect(

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1}/flask_humanify/humanify.py

@@ -2,7 +2,8 @@ from dataclasses import dataclass
 import logging
 from typing import List, Optional
 
-from flask import Blueprint, request, render_template, redirect, url_for, Response
+from werkzeug.wrappers import Response
+from flask import Blueprint, request, render_template, redirect, url_for, current_app
 from .ipset import IPSetClient, ensure_server_running
 from .utils import get_client_ip, get_return_url
 
@@ -120,6 +121,25 @@ class Humanify:
                 {"Cache-Control": "public, max-age=15552000"},
             )
 
+    def register_middleware(self, action: str = "deny_access"):
+        """
+        Register the middleware.
+        """
+
+        self.app = self.app or current_app
+
+        @self.app.before_request
+        def before_request():
+            """
+            Before request hook.
+            """
+            if request.endpoint in ["humanify.rate_limited", "humanify.access_denied"]:
+                return
+
+            if self.is_bot:
+                if action == "deny_access":
+                    return self.deny_access()
+
     @property
     def is_bot(self) -> HumanifyResult:
         """
@@ -131,7 +151,7 @@ class Humanify:
         ip_groups = self.ipset_client.lookup_ip(ip)
         return HumanifyResult.from_ip_groups(ip, ip_groups)
 
-    def redirect_to_access_denied(self) -> Response:
+    def deny_access(self) -> Response:
         """
         Redirect to the access denied page.
         """

{flask_humanify-0.1.2 → flask_humanify-0.1.3.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flask-Humanify"
-version = "0.1.2"
+version = "0.1.3.1"
 description = "Protect against bots and DDoS attacks"
 readme = "README.md"
 authors = [