flask-Humanify 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

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"
 
 from . import utils
 from .humanify import Humanify

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/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,22 @@ 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 self.is_bot:
+                if action == "deny_access":
+                    return self.deny_access()
+
     @property
     def is_bot(self) -> HumanifyResult:
         """
@@ -131,7 +148,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.dist-info → flask_humanify-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flask-Humanify
-Version: 0.1.2
+Version: 0.1.3
 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.dist-info → flask_humanify-0.1.3.dist-info}/RECORD

@@ -1,14 +1,14 @@
-flask_humanify/__init__.py,sha256=L67JR_FDgC9iNnBrnRchNPZe_z7tXbwV_xJ4NEQKGFw,269
-flask_humanify/humanify.py,sha256=JJeTcJlMte6zj2QRlat2876iJzFg40YNoMEX5kWepiU,3772
+flask_humanify/__init__.py,sha256=LxR6tCZClN04nVGXLyET5HGZ7mv0vOQrROj8Dvwe-M0,269
+flask_humanify/humanify.py,sha256=3tZx3RunXLVKGYJ6LukhsTuc-LFurMVf8pAQpMf0kUI,4213
 flask_humanify/ipset.py,sha256=ZZZRtgtddkZr2q1zuI6803hJQ8OodVHNdyZvZGqpmMI,10866
 flask_humanify/utils.py,sha256=CJ4FPhNJ75FuWcAn_ZuZkqRa9HR3jVvYOu9NZaN4L1o,2461
 flask_humanify/datasets/ipset.json,sha256=YNPqwI109lYkfvZeOPsoDH_dKJxOCs0G2nvx_s2mvqU,30601191
-flask_humanify/features/rate_limiter.py,sha256=kl3l1SHtDOPzOb1TLZizIRp-mJo5d30uW62ocmBm_fM,2175
+flask_humanify/features/rate_limiter.py,sha256=YhAh4xAZTiqdMbS028jA_v6FeZEB1v6JtbOVubjGZao,2203
 flask_humanify/templates/access_denied.html,sha256=Y7EzM53LRBMJBniczTWYvinHuPPFQVKrqvCmKaqLHak,3165
 flask_humanify/templates/oneclick_captcha.html,sha256=CnK4qTOdjIrlnFB8lytuK8pEcpj1dL5uYzj0ZNBkx-E,6241
 flask_humanify/templates/rate_limited.html,sha256=LSS1-c9F3YLxXF-jiyyNN6vIMoIAdmd6ObO1PDcjIxI,3110
-flask_humanify-0.1.2.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-flask_humanify-0.1.2.dist-info/METADATA,sha256=9oxqmwHRjGz9RVxSwxuZWdXcjYdPtaC2j17rOkOWVyE,2787
-flask_humanify-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-flask_humanify-0.1.2.dist-info/top_level.txt,sha256=9-c6uhxwCpPE3BJYge1Y9Z_bYmWitI0fY5RgqMiFWr0,15
-flask_humanify-0.1.2.dist-info/RECORD,,
+flask_humanify-0.1.3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+flask_humanify-0.1.3.dist-info/METADATA,sha256=fFcFNB3xEAp7HtSQjFvW9kH_bQpo1GGZUPhZ1NJLFoo,3048
+flask_humanify-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+flask_humanify-0.1.3.dist-info/top_level.txt,sha256=9-c6uhxwCpPE3BJYge1Y9Z_bYmWitI0fY5RgqMiFWr0,15
+flask_humanify-0.1.3.dist-info/RECORD,,