aiwaf 0.1.9.0.8__tar.gz → 0.1.9.1.0__tar.gz

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/PKG-INFO

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

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/aiwaf/__init__.py

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

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/aiwaf/storage.py

@@ -111,6 +111,16 @@ 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 add_ip(ip, reason="Automated block"):
+        """Add IP to blacklist (alias for block_ip)"""
+        ModelBlacklistStore.block_ip(ip, reason)
+
     @staticmethod
     def get_all_blocked_ips():
         """Get all blocked IPs"""
@@ -184,7 +194,7 @@ class ModelExemptionStore:
 
 class ModelKeywordStore:
     @staticmethod
-    def add_keyword(keyword):
+    def add_keyword(keyword, count=1):
         """Add a keyword to the dynamic keyword list"""
         _import_models()
         if DynamicKeyword is None:
@@ -192,11 +202,25 @@ class ModelKeywordStore:
         try:
             obj, created = DynamicKeyword.objects.get_or_create(keyword=keyword)
             if not created:
-                obj.count += 1
+                obj.count += count
+                obj.save()
+            else:
+                obj.count = count
                 obj.save()
         except Exception as e:
             print(f"Error adding keyword {keyword}: {e}")
 
+    @staticmethod
+    def remove_keyword(keyword):
+        """Remove a keyword from the dynamic keyword list"""
+        _import_models()
+        if DynamicKeyword is None:
+            return
+        try:
+            DynamicKeyword.objects.filter(keyword=keyword).delete()
+        except Exception as e:
+            print(f"Error removing keyword {keyword}: {e}")
+
     @staticmethod
     def get_top_keywords(n=10):
         """Get top N keywords by count"""

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/aiwaf.egg-info/PKG-INFO

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

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "aiwaf"
-version = "0.1.9.0.8"
+version = "0.1.9.1.0"
 description = "AI-powered Web Application Firewall"
 readme = "README.md"
 requires-python = ">=3.8"

{aiwaf-0.1.9.0.8 → aiwaf-0.1.9.1.0}/setup.py

@@ -9,7 +9,7 @@ long_description = (HERE / "README.md").read_text(encoding="utf-8")
 
 setup(
     name="aiwaf",
-    version="0.1.9.0.8",
+    version="0.1.9.1.0",
     description="AI‑driven, self‑learning Web Application Firewall for Django",
     long_description=long_description,
     long_description_content_type="text/markdown",