aiwaf 0.1.8.8__py3-none-any.whl → 0.1.9.0__py3-none-any.whl

aiwaf/__init__.py

@@ -1 +1,19 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
+
+__version__ = "0.1.9.0"
+
+# Import main middleware classes for easier access
+try:
+    from .middleware import (
+        IPAndKeywordBlockMiddleware,
+        RateLimitMiddleware, 
+        AIAnomalyMiddleware,
+        HoneypotTimingMiddleware,
+        UUIDTamperMiddleware
+    )
+except ImportError as e:
+    # Handle import errors gracefully during package installation
+    import sys
+    if 'runserver' in sys.argv or 'migrate' in sys.argv or 'shell' in sys.argv:
+        print(f"Warning: Could not import middleware classes: {e}")
+        print("Tip: Run 'python manage.py aiwaf_diagnose' to troubleshoot")

aiwaf/management/commands/aiwaf_diagnose.py

@@ -0,0 +1,127 @@
+from django.core.management.base import BaseCommand
+from django.conf import settings
+import sys
+
+class Command(BaseCommand):
+    help = 'Diagnose AI-WAF installation and middleware setup'
+
+    def handle(self, *args, **options):
+        self.stdout.write(self.style.HTTP_INFO("🔍 AI-WAF Installation Diagnostics"))
+        self.stdout.write("")
+        
+        # Check AI-WAF import
+        try:
+            import aiwaf
+            version = getattr(aiwaf, '__version__', 'Unknown')
+            self.stdout.write(self.style.SUCCESS(f"✅ AI-WAF imported successfully (version: {version})"))
+        except ImportError as e:
+            self.stdout.write(self.style.ERROR(f"❌ AI-WAF import failed: {e}"))
+            return
+        
+        # Check if aiwaf is in INSTALLED_APPS
+        installed_apps = getattr(settings, 'INSTALLED_APPS', [])
+        if 'aiwaf' in installed_apps:
+            self.stdout.write(self.style.SUCCESS("✅ 'aiwaf' found in INSTALLED_APPS"))
+        else:
+            self.stdout.write(self.style.ERROR("❌ 'aiwaf' NOT found in INSTALLED_APPS"))
+            self.stdout.write(self.style.WARNING("   Add 'aiwaf' to your INSTALLED_APPS in settings.py"))
+        
+        # Check middleware availability
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("🧱 Middleware Availability Check:"))
+        
+        middleware_classes = [
+            'IPAndKeywordBlockMiddleware',
+            'RateLimitMiddleware', 
+            'AIAnomalyMiddleware',
+            'HoneypotTimingMiddleware',
+            'UUIDTamperMiddleware'
+        ]
+        
+        try:
+            import aiwaf.middleware as mw
+            available_classes = dir(mw)
+            
+            for middleware_class in middleware_classes:
+                if middleware_class in available_classes:
+                    self.stdout.write(self.style.SUCCESS(f"  ✅ {middleware_class}"))
+                else:
+                    self.stdout.write(self.style.ERROR(f"  ❌ {middleware_class} (missing)"))
+        except ImportError as e:
+            self.stdout.write(self.style.ERROR(f"❌ Could not import aiwaf.middleware: {e}"))
+        
+        # Check configured middleware
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("⚙️ Configured Middleware:"))
+        
+        middleware_setting = getattr(settings, 'MIDDLEWARE', [])
+        aiwaf_middleware = [mw for mw in middleware_setting if 'aiwaf' in mw.lower()]
+        
+        if aiwaf_middleware:
+            for mw in aiwaf_middleware:
+                # Test if middleware can be imported
+                try:
+                    from django.utils.module_loading import import_string
+                    import_string(mw)
+                    self.stdout.write(self.style.SUCCESS(f"  ✅ {mw}"))
+                except ImportError as e:
+                    self.stdout.write(self.style.ERROR(f"  ❌ {mw} - Import Error: {e}"))
+                except Exception as e:
+                    self.stdout.write(self.style.WARNING(f"  ⚠️  {mw} - Warning: {e}"))
+        else:
+            self.stdout.write(self.style.WARNING("  No AI-WAF middleware found in MIDDLEWARE setting"))
+        
+        # Check storage configuration
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("💾 Storage Configuration:"))
+        
+        storage_mode = getattr(settings, 'AIWAF_STORAGE_MODE', 'models')
+        self.stdout.write(f"  Storage Mode: {storage_mode}")
+        
+        if storage_mode == 'csv':
+            csv_dir = getattr(settings, 'AIWAF_CSV_DATA_DIR', 'aiwaf_data')
+            self.stdout.write(f"  CSV Directory: {csv_dir}")
+        
+        # Check access log
+        access_log = getattr(settings, 'AIWAF_ACCESS_LOG', None)
+        if access_log:
+            import os
+            if os.path.exists(access_log):
+                self.stdout.write(self.style.SUCCESS(f"  ✅ Access log found: {access_log}"))
+            else:
+                self.stdout.write(self.style.WARNING(f"  ⚠️  Access log not found: {access_log}"))
+        else:
+            self.stdout.write(self.style.WARNING("  ⚠️  AIWAF_ACCESS_LOG not configured"))
+        
+        # Check middleware logging
+        middleware_logging = getattr(settings, 'AIWAF_MIDDLEWARE_LOGGING', False)
+        if middleware_logging:
+            self.stdout.write(self.style.SUCCESS("  ✅ Middleware logging enabled"))
+        else:
+            self.stdout.write(self.style.WARNING("  ⚠️  Middleware logging disabled"))
+        
+        # Recommendations
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("💡 Recommendations:"))
+        
+        if 'aiwaf' not in installed_apps:
+            self.stdout.write("  1. Add 'aiwaf' to INSTALLED_APPS")
+            
+        if not aiwaf_middleware:
+            self.stdout.write("  2. Add AI-WAF middleware to MIDDLEWARE setting")
+            
+        if not access_log and not middleware_logging:
+            self.stdout.write("  3. Configure AIWAF_ACCESS_LOG or enable AIWAF_MIDDLEWARE_LOGGING")
+        
+        # Quick fix commands
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("🚀 Quick Fix Commands:"))
+        self.stdout.write("")
+        self.stdout.write("# Update AI-WAF to latest version:")
+        self.stdout.write("pip install --upgrade aiwaf")
+        self.stdout.write("")
+        self.stdout.write("# Run migrations (if using models mode):")
+        self.stdout.write("python manage.py migrate")
+        self.stdout.write("")
+        self.stdout.write("# Test AI-WAF commands:")
+        self.stdout.write("python manage.py add_ipexemption 127.0.0.1 --reason Testing")

aiwaf/storage.py

@@ -3,7 +3,18 @@ import numpy as np
 import pandas as pd
 from django.conf import settings
 from django.utils import timezone
-from .models import FeatureSample, BlacklistEntry, IPExemption, DynamicKeyword
+
+# Only import models if aiwaf is in INSTALLED_APPS
+try:
+    from django.apps import apps
+    if apps.is_installed('aiwaf'):
+        from .models import FeatureSample, BlacklistEntry, IPExemption, DynamicKeyword
+    else:
+        # Create dummy classes to avoid import errors
+        FeatureSample = BlacklistEntry = IPExemption = DynamicKeyword = None
+except (ImportError, RuntimeError):
+    # Handle cases where Django isn't fully initialized yet
+    FeatureSample = BlacklistEntry = IPExemption = DynamicKeyword = None
 
 # Configuration
 STORAGE_MODE = getattr(settings, "AIWAF_STORAGE_MODE", "models")  # "models" or "csv"
@@ -52,22 +63,25 @@ class CsvFeatureStore:
 class DbFeatureStore:
     @staticmethod
     def persist_rows(rows):
-        objs = []
-        for ip,pl,kw,rt,si,bc,t404,label in rows:
-            objs.append(FeatureSample(
-                ip=ip, path_len=pl, kw_hits=kw,
-                resp_time=rt, status_idx=si,
-                burst_count=bc, total_404=t404,
-                label=label
-            ))
-        FeatureSample.objects.bulk_create(objs, ignore_conflicts=True)
+        if FeatureSample is not None:
+            objs = []
+            for ip,pl,kw,rt,si,bc,t404,label in rows:
+                objs.append(FeatureSample(
+                    ip=ip, path_len=pl, kw_hits=kw,
+                    resp_time=rt, status_idx=si,
+                    burst_count=bc, total_404=t404,
+                    label=label
+                ))
+            FeatureSample.objects.bulk_create(objs, ignore_conflicts=True)
 
     @staticmethod
     def load_matrix():
-        qs = FeatureSample.objects.all().values_list(
-            "path_len","kw_hits","resp_time","status_idx","burst_count","total_404"
-        )
-        return np.array(list(qs), dtype=float)
+        if FeatureSample is not None:
+            qs = FeatureSample.objects.all().values_list(
+                "path_len","kw_hits","resp_time","status_idx","burst_count","total_404"
+            )
+            return np.array(list(qs), dtype=float)
+        return np.empty((0,6))
 
 def get_store():
     if getattr(settings, "AIWAF_FEATURE_STORE", "csv") == "db":
@@ -266,8 +280,12 @@ def get_blacklist_store():
     if STORAGE_MODE == "csv":
         return CsvBlacklistStore
     else:
-        # Return a wrapper for Django models
-        return ModelBlacklistStore
+        # Return a wrapper for Django models (only if models are available)
+        if BlacklistEntry is not None:
+            return ModelBlacklistStore
+        else:
+            # Fallback to CSV if models aren't available
+            return CsvBlacklistStore
 
 
 def get_exemption_store():
@@ -275,7 +293,10 @@ def get_exemption_store():
     if STORAGE_MODE == "csv":
         return CsvExemptionStore
     else:
-        return ModelExemptionStore
+        if IPExemption is not None:
+            return ModelExemptionStore
+        else:
+            return CsvExemptionStore
 
 
 def get_keyword_store():
@@ -283,7 +304,10 @@ def get_keyword_store():
     if STORAGE_MODE == "csv":
         return CsvKeywordStore
     else:
-        return ModelKeywordStore
+        if DynamicKeyword is not None:
+            return ModelKeywordStore
+        else:
+            return CsvKeywordStore
 
 
 # ============= Django Model Wrappers =============
@@ -293,19 +317,25 @@ class ModelBlacklistStore:
     
     @staticmethod
     def add_ip(ip_address, reason):
-        BlacklistEntry.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
+        if BlacklistEntry is not None:
+            BlacklistEntry.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
     
     @staticmethod
     def is_blocked(ip_address):
-        return BlacklistEntry.objects.filter(ip_address=ip_address).exists()
+        if BlacklistEntry is not None:
+            return BlacklistEntry.objects.filter(ip_address=ip_address).exists()
+        return False
     
     @staticmethod
     def get_all():
-        return list(BlacklistEntry.objects.values("ip_address", "reason", "created_at"))
+        if BlacklistEntry is not None:
+            return list(BlacklistEntry.objects.values("ip_address", "reason", "created_at"))
+        return []
     
     @staticmethod
     def remove_ip(ip_address):
-        BlacklistEntry.objects.filter(ip_address=ip_address).delete()
+        if BlacklistEntry is not None:
+            BlacklistEntry.objects.filter(ip_address=ip_address).delete()
 
 
 class ModelExemptionStore:
@@ -313,19 +343,25 @@ class ModelExemptionStore:
     
     @staticmethod
     def add_ip(ip_address, reason=""):
-        IPExemption.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
+        if IPExemption is not None:
+            IPExemption.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
     
     @staticmethod
     def is_exempted(ip_address):
-        return IPExemption.objects.filter(ip_address=ip_address).exists()
+        if IPExemption is not None:
+            return IPExemption.objects.filter(ip_address=ip_address).exists()
+        return False
     
     @staticmethod
     def get_all():
-        return list(IPExemption.objects.values("ip_address", "reason", "created_at"))
+        if IPExemption is not None:
+            return list(IPExemption.objects.values("ip_address", "reason", "created_at"))
+        return []
     
     @staticmethod
     def remove_ip(ip_address):
-        IPExemption.objects.filter(ip_address=ip_address).delete()
+        if IPExemption is not None:
+            IPExemption.objects.filter(ip_address=ip_address).delete()
 
 
 class ModelKeywordStore:
@@ -333,19 +369,24 @@ class ModelKeywordStore:
     
     @staticmethod
     def add_keyword(keyword, count=1):
-        obj, created = DynamicKeyword.objects.get_or_create(keyword=keyword, defaults={"count": count})
-        if not created:
-            obj.count += count
-            obj.save()
+        if DynamicKeyword is not None:
+            obj, created = DynamicKeyword.objects.get_or_create(keyword=keyword, defaults={"count": count})
+            if not created:
+                obj.count += count
+                obj.save()
     
     @staticmethod
     def get_top_keywords(limit=10):
-        return list(DynamicKeyword.objects.order_by("-count").values_list("keyword", flat=True)[:limit])
+        if DynamicKeyword is not None:
+            return list(DynamicKeyword.objects.order_by("-count").values_list("keyword", flat=True)[:limit])
+        return []
     
     @staticmethod
     def remove_keyword(keyword):
-        DynamicKeyword.objects.filter(keyword=keyword).delete()
+        if DynamicKeyword is not None:
+            DynamicKeyword.objects.filter(keyword=keyword).delete()
     
     @staticmethod
     def clear_all():
-        DynamicKeyword.objects.all().delete()
+        if DynamicKeyword is not None:
+            DynamicKeyword.objects.all().delete()

{aiwaf-0.1.8.8.dist-info → aiwaf-0.1.9.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.8.8
+Version: 0.1.9.0
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -27,6 +27,18 @@ Dynamic: requires-python
 
 ---
 
+## 🚀 Quick Installation
+
+```bash
+pip install aiwaf
+```
+
+**⚠️ Important:** Add `'aiwaf'` to your Django `INSTALLED_APPS` to avoid setup errors.
+
+**📋 Complete Setup Guide:** See [INSTALLATION.md](INSTALLATION.md) for detailed installation instructions and troubleshooting.
+
+---
+
 ## System Requirements
 
 No GPU needed—AI-WAF runs entirely on CPU with just Python 3.8+, Django 3.2+, a single vCPU and ~512 MB RAM for small sites; for moderate production traffic you can bump to 2–4 vCPUs and 2–4 GB RAM, offload the daily detect-and-train job to a worker, and rotate logs to keep memory use bounded.
@@ -312,6 +324,57 @@ MIDDLEWARE = [
 
 > **⚠️ Order matters!** AI-WAF protection middleware should come early. The logger middleware should come near the end to capture final response data.
 
+### **Troubleshooting Middleware Errors**
+
+**Error: `Module "aiwaf.middleware" does not define a "UUIDTamperMiddleware" attribute/class`**
+
+**Solutions:**
+1. **Update AI-WAF to latest version:**
+   ```bash
+   pip install --upgrade aiwaf
+   ```
+
+2. **Run diagnostic commands:**
+   ```bash
+   # Quick debug script (from AI-WAF directory)
+   python debug_aiwaf.py
+   
+   # Django management command  
+   python manage.py aiwaf_diagnose
+   ```
+
+3. **Check available middleware classes:**
+   ```python
+   # In Django shell: python manage.py shell
+   import aiwaf.middleware
+   print(dir(aiwaf.middleware))
+   ```
+
+4. **Verify AI-WAF is in INSTALLED_APPS:**
+   ```python
+   # In settings.py
+   INSTALLED_APPS = [
+       # ... other apps ...
+       'aiwaf',  # Must be included
+   ]
+   ```
+
+5. **Use minimal middleware setup if needed:**
+   ```python
+   MIDDLEWARE = [
+       # ... your existing middleware ...
+       "aiwaf.middleware.IPAndKeywordBlockMiddleware",  # Core protection
+       "aiwaf.middleware.RateLimitMiddleware",          # Rate limiting  
+       "aiwaf.middleware.AIAnomalyMiddleware",          # AI detection
+   ]
+   ```
+
+**Common Issues:**
+- Missing Django: `pip install Django`
+- Old AI-WAF version: `pip install --upgrade aiwaf`
+- Missing migrations: `python manage.py migrate`
+- Import errors: Check `INSTALLED_APPS` includes `'aiwaf'`
+
 ---
 
 ##  Running Detection & Training

{aiwaf-0.1.8.8.dist-info → aiwaf-0.1.9.0.dist-info}/RECORD

@@ -1,24 +1,25 @@
-aiwaf/__init__.py,sha256=nQFpJ1YpX48snzLjEQCf8zD2YNh8v0b_kPTrXx8uBYc,46
+aiwaf/__init__.py,sha256=XzdZPG3k3A6cA_FfVC397hcb_di2dW2LV6iULGLbZ5w,653
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=92ltIrFfv8WOC4CXwvNVZYfivkRZHGNg3E2QAbHQipQ,550
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
 aiwaf/middleware.py,sha256=1JPrc0npI_a5bnB-thN0ME1ehfTbWBl1j9wTndZwRdQ,9505
 aiwaf/middleware_logger.py,sha256=uTYTvIc4Mv1pjY50aXaqQ5cWAO9qqquijAyVMs1KWlM,6517
 aiwaf/models.py,sha256=XaG1pd_oZu3y-fw66u4wblGlWcUY9gvsTNKGD0kQk7Y,1672
-aiwaf/storage.py,sha256=Z0KWArfLmOHnvUcL5aVx8W_aHMr-qoEW8FVGrM23BvA,11639
+aiwaf/storage.py,sha256=oI9ZuTCrLwLSOE3I3_tDTFs6jiile7Vr4e7jacf2B6Y,13218
 aiwaf/trainer.py,sha256=bgVoBewnNVMJdgxcNchfhsPOnFXxStoBOqNhFYnpsqs,9244
 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_ipexemption.py,sha256=srgdVPDJtF7G9GGIqaZ7L3qTuNheoS_uwlhlRO4W2bc,945
+aiwaf/management/commands/aiwaf_diagnose.py,sha256=W2OQbPJ2xXTD4OTJw2VnqJTo4QR3N9acPst3SKXy0E0,5705
 aiwaf/management/commands/aiwaf_logging.py,sha256=FCIqULn2tii2vD9VxL7vk3PV4k4vr7kaA00KyaCExYY,7692
 aiwaf/management/commands/aiwaf_reset.py,sha256=0FIBqpZS8xgFFvAKJ-0zAC_-QNQwRkOHpXb8N-OdFr8,3740
 aiwaf/management/commands/detect_and_train.py,sha256=-o-LZ7QZ5GeJPCekryox1DGXKMmFEkwwrcDsiM166K0,269
 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.8.8.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.8.8.dist-info/METADATA,sha256=851Url25O97G0KGi3gHFF-zYSaI91BHiK7CbKJrLbk0,11261
-aiwaf-0.1.8.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.8.8.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.8.8.dist-info/RECORD,,
+aiwaf-0.1.9.0.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.0.dist-info/METADATA,sha256=xZRQCE6zKCkNtE54rrWaxBllGdslHApII1CCC1JQE0I,12903
+aiwaf-0.1.9.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.0.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.0.dist-info/RECORD,,