aiwaf 0.1.9.2.1__tar.gz → 0.1.9.2.2__tar.gz

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/PKG-INFO

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

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/aiwaf/__init__.py

@@ -1,6 +1,6 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.2.1"
+__version__ = "0.1.9.2.2"
 
 # 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.2.2/aiwaf/management/commands/aiwaf_list.py

@@ -0,0 +1,81 @@
+from django.core.management.base import BaseCommand
+from django.utils import timezone
+from aiwaf.storage import get_blacklist_store, get_exemption_store, get_keyword_store
+from datetime import timedelta
+import json
+
+def _sort(items, order):
+    reverse = (order == "newest")
+    return sorted(items, key=lambda x: x.get("created_at") or timezone.make_aware(timezone.datetime.min),
+                  reverse=reverse)
+
+def _filter_since(items, seconds):
+    if not seconds: return items
+    cutoff = timezone.now() - timedelta(seconds=seconds)
+    return [it for it in items if it.get("created_at") and it["created_at"] >= cutoff]
+
+def _print_table(rows, headers):
+    widths = [len(h) for h in headers]
+    for r in rows:
+        for i, cell in enumerate(r):
+            widths[i] = max(widths[i], len(str(cell)))
+    print(" | ".join(h.ljust(widths[i]) for i, h in enumerate(headers)))
+    print("-+-".join("-" * w for w in widths))
+    for r in rows:
+        print(" | ".join(str(cell).ljust(widths[i]) for i, cell in enumerate(r)))
+
+class Command(BaseCommand):
+    help = "Lister les données AIWAF (IPs bloquées, exemptions, mots-clés dynamiques)."
+
+    def add_arguments(self, parser):
+        grp = parser.add_mutually_exclusive_group()
+        grp.add_argument("--ips", action="store_true", help="Lister les IPs bloquées (défaut).")
+        grp.add_argument("--exemptions", action="store_true", help="Lister les IPs exemptées.")
+        grp.add_argument("--keywords", action="store_true", help="Lister les mots-clés dynamiques.")
+        grp.add_argument("--all", action="store_true", help="Tout lister.")
+        parser.add_argument("--format", choices=["table", "json"], default="table")
+        parser.add_argument("--limit", type=int, default=100)
+        parser.add_argument("--order", choices=["newest", "oldest"], default="newest")
+        parser.add_argument("--since", type=int, help="Fenêtre en secondes (ex: 86400 = 24h).")
+
+    def handle(self, *args, **o):
+        if not any([o["exemptions"], o["keywords"], o["all"]]):  # défaut = ips
+            o["ips"] = True
+        payload = {}
+
+        if o["all"] or o["ips"]:
+            data = get_blacklist_store().get_all()
+            data = _filter_since(data, o.get("since"))
+            data = _sort(data, o["order"])[:o["limit"]]
+            payload["ips"] = data
+
+        if o["all"] or o["exemptions"]:
+            data = get_exemption_store().get_all()
+            data = _filter_since(data, o.get("since"))
+            data = _sort(data, o["order"])[:o["limit"]]
+            payload["exemptions"] = data
+
+        if o["all"] or o["keywords"]:
+            kws = get_keyword_store().get_top_keywords(o["limit"])
+            payload["keywords"] = [{"keyword": k} for k in kws]
+
+        if o["format"] == "json":
+            def _default(v):
+                try: return v.isoformat()
+                except Exception: return str(v)
+            self.stdout.write(json.dumps(payload, ensure_ascii=False, indent=2, default=_default))
+        else:
+            if "ips" in payload:
+                print("\n== IPs bloquées ==")
+                rows = [[r.get("ip_address",""), r.get("reason",""), r.get("created_at","")]
+                        for r in payload["ips"]]
+                _print_table(rows, ["ip_address", "reason", "created_at"])
+            if "exemptions" in payload:
+                print("\n== Exemptions ==")
+                rows = [[r.get("ip_address",""), r.get("reason",""), r.get("created_at","")]
+                        for r in payload["exemptions"]]
+                _print_table(rows, ["ip_address", "reason", "created_at"])
+            if "keywords" in payload:
+                print("\n== Mots-clés dynamiques ==")
+                rows = [[r["keyword"]] for r in payload["keywords"]]
+                _print_table(rows, ["keyword"])

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/aiwaf/trainer.py

@@ -34,19 +34,34 @@ def path_exists_in_django(path: str) -> bool:
     from django.urls import get_resolver
     from django.urls.resolvers import URLResolver
 
-    candidate = path.split("?")[0].lstrip("/")
+    candidate = path.split("?")[0].strip("/")  # Remove query params and normalize slashes
+    
+    # Try exact resolution first - this is the most reliable method
     try:
         get_resolver().resolve(f"/{candidate}")
         return True
     except:
         pass
-
-    root = get_resolver()
-    for p in root.url_patterns:
-        if isinstance(p, URLResolver):
-            prefix = p.pattern.describe().strip("^/")
-            if prefix and candidate.startswith(prefix):
-                return True
+    
+    # Also try with trailing slash if it doesn't have one
+    if not candidate.endswith("/"):
+        try:
+            get_resolver().resolve(f"/{candidate}/")
+            return True
+        except:
+            pass
+    
+    # Try without trailing slash if it has one
+    if candidate.endswith("/"):
+        try:
+            get_resolver().resolve(f"/{candidate.rstrip('/')}")
+            return True
+        except:
+            pass
+
+    # If direct resolution fails, be conservative
+    # Only do basic prefix matching for known include patterns
+    # but don't assume sub-paths exist just because the prefix exists
     return False
 
 

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/aiwaf.egg-info/PKG-INFO

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

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/aiwaf.egg-info/SOURCES.txt

@@ -22,6 +22,7 @@ aiwaf/management/commands/__init__.py
 aiwaf/management/commands/add_exemption.py
 aiwaf/management/commands/add_ipexemption.py
 aiwaf/management/commands/aiwaf_diagnose.py
+aiwaf/management/commands/aiwaf_list.py
 aiwaf/management/commands/aiwaf_logging.py
 aiwaf/management/commands/aiwaf_reset.py
 aiwaf/management/commands/check_dependencies.py

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/pyproject.toml

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

{aiwaf-0.1.9.2.1 → aiwaf-0.1.9.2.2}/setup.py

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