aiwaf 0.1.9.0.7__py3-none-any.whl → 0.1.9.0.9__py3-none-any.whl

aiwaf/__init__.py

@@ -1,6 +1,6 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.0.7"
+__version__ = "0.1.9.0.9"
 
 # Note: Middleware classes are available from aiwaf.middleware
 # Import them only when needed to avoid circular imports during Django app loading

aiwaf/middleware.py

@@ -135,12 +135,12 @@ class IPAndKeywordBlockMiddleware:
 
 
 class RateLimitMiddleware:
-    WINDOW = 10  # seconds
-    MAX = 20     # soft limit
-    FLOOD = 40   # hard limit
-
     def __init__(self, get_response):
         self.get_response = get_response
+        # Make rate limiting configurable via Django settings
+        self.WINDOW = getattr(settings, "AIWAF_RATE_WINDOW", 10)  # seconds
+        self.MAX = getattr(settings, "AIWAF_RATE_MAX", 20)        # soft limit
+        self.FLOOD = getattr(settings, "AIWAF_RATE_FLOOD", 40)    # hard limit
 
     def __call__(self, request):
         if is_exempt(request):

aiwaf/storage.py

@@ -111,6 +111,11 @@ class ModelBlacklistStore:
         except Exception as e:
             print(f"Error unblocking IP {ip}: {e}")
 
+    @staticmethod
+    def remove_ip(ip):
+        """Remove IP from blacklist (alias for unblock_ip)"""
+        ModelBlacklistStore.unblock_ip(ip)
+
     @staticmethod
     def get_all_blocked_ips():
         """Get all blocked IPs"""
@@ -171,6 +176,17 @@ class ModelExemptionStore:
         except Exception:
             return []
 
+    @staticmethod
+    def get_all():
+        """Get all exempted IP entries as dictionaries"""
+        _import_models()
+        if IPExemption is None:
+            return []
+        try:
+            return list(IPExemption.objects.values('ip_address', 'reason', 'created_at'))
+        except Exception:
+            return []
+
 class ModelKeywordStore:
     @staticmethod
     def add_keyword(keyword):

{aiwaf-0.1.9.0.7.dist-info → aiwaf-0.1.9.0.9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.9.0.7
+Version: 0.1.9.0.9
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba

{aiwaf-0.1.9.0.7.dist-info → aiwaf-0.1.9.0.9.dist-info}/RECORD

@@ -1,11 +1,11 @@
-aiwaf/__init__.py,sha256=5uH_XGymeTFxegNihwozJE8d-Y70jvwpKhXvzZ9HkQI,220
+aiwaf/__init__.py,sha256=06AIH-8CTV0EiCTYU4yai9x4cfBcJ12BONhmB9MJ25s,220
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=LYCeKFB-7e_C6Bg2WeFJWFIIQlrfRMPuGp30ivrnhQY,1196
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
-aiwaf/middleware.py,sha256=uKGelYd7SLB37BMcR7SQ_83NIPy6n2Y17ZvTOFpAOWM,12262
+aiwaf/middleware.py,sha256=4Ox0pUdB7rMT1Sw5XHO6-udQrfqyF9VGdkkkgLioRJ0,12470
 aiwaf/middleware_logger.py,sha256=LWZVDAnjh6CGESirA8eMbhGgJKB7lVDGRQqVroH95Lo,4742
 aiwaf/models.py,sha256=vQxgY19BDVMjoO903UNrTZC1pNoLltMU6wbyWPoAEns,2719
-aiwaf/storage.py,sha256=z2QjAySa95FN3SPJL2GBL11tE8JYrAH6mB7BMSZ6GEM,7310
+aiwaf/storage.py,sha256=M_MT0z7rwqAxRTfxlFUxESVJiq6-x9tHw0JSX-EOOo0,7783
 aiwaf/trainer.py,sha256=dOjdQ5zqEuwzffgTu7WZMokFcW5b0-KTDSjMeHBcIhw,9203
 aiwaf/utils.py,sha256=BJk5vJCYdGPl_4QQiknjhCbkzv5HZCXgFcBJDMJpHok,3390
 aiwaf/management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -26,8 +26,8 @@ aiwaf/management/commands/test_exemption_fix.py,sha256=ngyGaHUCmQQ6y--6j4q1viZJt
 aiwaf/resources/model.pkl,sha256=5t6h9BX8yoh2xct85MXOO60jdlWyg1APskUOW0jZE1Y,1288265
 aiwaf/templatetags/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiwaf/templatetags/aiwaf_tags.py,sha256=XXfb7Tl4DjU3Sc40GbqdaqOEtKTUKELBEk58u83wBNw,357
-aiwaf-0.1.9.0.7.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.9.0.7.dist-info/METADATA,sha256=rFNbBTlRRki2Prz5URgHw8yYmWTTSsxxBTeE5evUabE,13763
-aiwaf-0.1.9.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.9.0.7.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.9.0.7.dist-info/RECORD,,
+aiwaf-0.1.9.0.9.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.0.9.dist-info/METADATA,sha256=Ee1HG67zz35gFuvRfctd76FLm3eEdccUuZxAw9BJlnU,13763
+aiwaf-0.1.9.0.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.0.9.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.0.9.dist-info/RECORD,,