aiwaf 0.1.9.2.1__py3-none-any.whl → 0.1.9.2.2__py3-none-any.whl

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/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/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.dist-info → aiwaf-0.1.9.2.2.dist-info}/METADATA

@@ -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.dist-info → aiwaf-0.1.9.2.2.dist-info}/RECORD

@@ -1,4 +1,4 @@
-aiwaf/__init__.py,sha256=SLcMD_OTXr3DXtHpuCKxFvNl_pjrg-J5KLcJ-Swutuo,220
+aiwaf/__init__.py,sha256=EwzM3mRDf7i5IVX0-pTMpVFRf51lpIIbh6ZzkzOw10M,220
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=LYCeKFB-7e_C6Bg2WeFJWFIIQlrfRMPuGp30ivrnhQY,1196
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
@@ -6,13 +6,14 @@ aiwaf/middleware.py,sha256=8EC4AKfUjHhmVSKpquimkMUebBekr92pqyVF97wlbx0,27408
 aiwaf/middleware_logger.py,sha256=LWZVDAnjh6CGESirA8eMbhGgJKB7lVDGRQqVroH95Lo,4742
 aiwaf/models.py,sha256=vQxgY19BDVMjoO903UNrTZC1pNoLltMU6wbyWPoAEns,2719
 aiwaf/storage.py,sha256=5ImrZMRn3u7HNsPH0fDjWhDrD2tgG2IHVnOXtLz0fk4,10253
-aiwaf/trainer.py,sha256=47HP81kTaJCOfyONUm18r-FVc1YeRvcliO_akpX3BqI,18613
+aiwaf/trainer.py,sha256=U-X79nFhSTEbVexFHo3IXFf1HgvXrFnQ__WqTar0o4M,19118
 aiwaf/utils.py,sha256=BJk5vJCYdGPl_4QQiknjhCbkzv5HZCXgFcBJDMJpHok,3390
 aiwaf/management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiwaf/management/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aiwaf/management/commands/add_exemption.py,sha256=U_ByfJw1EstAZ8DaSoRb97IGwYzXs0DBJkVAqeN4Wak,1128
 aiwaf/management/commands/add_ipexemption.py,sha256=sSf3d9hGK9RqqlBYkCrnrd8KZWGT-derSpoWnEY4H60,952
 aiwaf/management/commands/aiwaf_diagnose.py,sha256=nXFRhq66N4QC3e4scYJ2sUngJce-0yDxtBO3R2BllRM,6134
+aiwaf/management/commands/aiwaf_list.py,sha256=tZK3FugApmPxxvmoB4-nLY9fpZJgiRtD137Bre5hEp8,3839
 aiwaf/management/commands/aiwaf_logging.py,sha256=FCIqULn2tii2vD9VxL7vk3PV4k4vr7kaA00KyaCExYY,7692
 aiwaf/management/commands/aiwaf_reset.py,sha256=pcF0zOYDSqjpCwDtk2HYJZLgr76td8OFRENtl20c1dQ,7472
 aiwaf/management/commands/check_dependencies.py,sha256=GOZl00pDwW2cJjDvIaCeB3yWxmeYcJDRTIpmOTLvy2c,37204
@@ -28,8 +29,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.2.1.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.9.2.1.dist-info/METADATA,sha256=OgVYn0PPKBDcGCVlhYEFa7uc9XU4Rn-0ZS-W2CE9a1Q,26824
-aiwaf-0.1.9.2.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.9.2.1.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.9.2.1.dist-info/RECORD,,
+aiwaf-0.1.9.2.2.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.2.2.dist-info/METADATA,sha256=NFu9QZWsGPcmAJaHeJroSXZL_PstDr33NbSffV94bLQ,26824
+aiwaf-0.1.9.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.2.2.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.2.2.dist-info/RECORD,,