aiwaf 0.1.8.9__tar.gz → 0.1.9.0.1__tar.gz

{aiwaf-0.1.8.9 → aiwaf-0.1.9.0.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.8.9
+Version: 0.1.9.0.1
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -324,6 +324,58 @@ 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:**
+- **AppRegistryNotReady Error**: Fixed in v0.1.9.0.1 - update with `pip install --upgrade aiwaf`
+- 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.9 → aiwaf-0.1.9.0.1}/README.md

@@ -303,6 +303,58 @@ 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:**
+- **AppRegistryNotReady Error**: Fixed in v0.1.9.0.1 - update with `pip install --upgrade aiwaf`
+- 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.9.0.1/aiwaf/__init__.py

@@ -0,0 +1,24 @@
+default_app_config = "aiwaf.apps.AiwafConfig"
+
+__version__ = "0.1.9.0.1"
+
+# Note: Middleware classes are available from aiwaf.middleware
+# Import them only when needed to avoid circular imports during Django app loading= "aiwaf.apps.AiwafConfig"
+
+__version__ = "0.1.9.0.1"
+
+# 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-0.1.9.0.1/aiwaf/management/commands/aiwaf_diagnose.py

@@ -0,0 +1,135 @@
+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})"))
+            
+            # Test critical imports that caused AppRegistryNotReady
+            try:
+                from aiwaf import storage, utils, trainer
+                self.stdout.write(self.style.SUCCESS("✅ Critical modules (storage, utils, trainer) imported successfully"))
+            except Exception as e:
+                self.stdout.write(self.style.ERROR(f"❌ Critical module import failed: {e}"))
+                
+        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-0.1.8.9 → aiwaf-0.1.9.0.1}/aiwaf/storage.py

@@ -4,17 +4,23 @@ import pandas as pd
 from django.conf import settings
 from django.utils import timezone
 
-# 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
+# Defer model imports to avoid AppRegistryNotReady during Django app loading
+FeatureSample = BlacklistEntry = IPExemption = DynamicKeyword = None
+
+def _import_models():
+    """Import Django models only when needed and apps are ready."""
+    global FeatureSample, BlacklistEntry, IPExemption, DynamicKeyword
+    
+    if FeatureSample is not None:
+        return  # Already imported
+    
+    try:
+        from django.apps import apps
+        if apps.ready and apps.is_installed('aiwaf'):
+            from .models import FeatureSample, BlacklistEntry, IPExemption, DynamicKeyword
+    except (ImportError, RuntimeError, Exception):
+        # Keep models as None if can't import
+        pass
 
 # Configuration
 STORAGE_MODE = getattr(settings, "AIWAF_STORAGE_MODE", "models")  # "models" or "csv"
@@ -63,6 +69,7 @@ class CsvFeatureStore:
 class DbFeatureStore:
     @staticmethod
     def persist_rows(rows):
+        _import_models()
         if FeatureSample is not None:
             objs = []
             for ip,pl,kw,rt,si,bc,t404,label in rows:
@@ -76,6 +83,7 @@ class DbFeatureStore:
 
     @staticmethod
     def load_matrix():
+        _import_models()
         if FeatureSample is not None:
             qs = FeatureSample.objects.all().values_list(
                 "path_len","kw_hits","resp_time","status_idx","burst_count","total_404"
@@ -317,23 +325,27 @@ class ModelBlacklistStore:
     
     @staticmethod
     def add_ip(ip_address, reason):
+        _import_models()
         if BlacklistEntry is not None:
             BlacklistEntry.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
     
     @staticmethod
     def is_blocked(ip_address):
+        _import_models()
         if BlacklistEntry is not None:
             return BlacklistEntry.objects.filter(ip_address=ip_address).exists()
         return False
     
     @staticmethod
     def get_all():
+        _import_models()
         if BlacklistEntry is not None:
             return list(BlacklistEntry.objects.values("ip_address", "reason", "created_at"))
         return []
     
     @staticmethod
     def remove_ip(ip_address):
+        _import_models()
         if BlacklistEntry is not None:
             BlacklistEntry.objects.filter(ip_address=ip_address).delete()
 
@@ -343,23 +355,27 @@ class ModelExemptionStore:
     
     @staticmethod
     def add_ip(ip_address, reason=""):
+        _import_models()
         if IPExemption is not None:
             IPExemption.objects.get_or_create(ip_address=ip_address, defaults={"reason": reason})
     
     @staticmethod
     def is_exempted(ip_address):
+        _import_models()
         if IPExemption is not None:
             return IPExemption.objects.filter(ip_address=ip_address).exists()
         return False
     
     @staticmethod
     def get_all():
+        _import_models()
         if IPExemption is not None:
             return list(IPExemption.objects.values("ip_address", "reason", "created_at"))
         return []
     
     @staticmethod
     def remove_ip(ip_address):
+        _import_models()
         if IPExemption is not None:
             IPExemption.objects.filter(ip_address=ip_address).delete()
 
@@ -369,6 +385,7 @@ class ModelKeywordStore:
     
     @staticmethod
     def add_keyword(keyword, count=1):
+        _import_models()
         if DynamicKeyword is not None:
             obj, created = DynamicKeyword.objects.get_or_create(keyword=keyword, defaults={"count": count})
             if not created:
@@ -377,16 +394,19 @@ class ModelKeywordStore:
     
     @staticmethod
     def get_top_keywords(limit=10):
+        _import_models()
         if DynamicKeyword is not None:
             return list(DynamicKeyword.objects.order_by("-count").values_list("keyword", flat=True)[:limit])
         return []
     
     @staticmethod
     def remove_keyword(keyword):
+        _import_models()
         if DynamicKeyword is not None:
             DynamicKeyword.objects.filter(keyword=keyword).delete()
     
     @staticmethod
     def clear_all():
+        _import_models()
         if DynamicKeyword is not None:
             DynamicKeyword.objects.all().delete()

{aiwaf-0.1.8.9 → aiwaf-0.1.9.0.1}/aiwaf.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.8.9
+Version: 0.1.9.0.1
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -324,6 +324,58 @@ 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:**
+- **AppRegistryNotReady Error**: Fixed in v0.1.9.0.1 - update with `pip install --upgrade aiwaf`
+- 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.9 → aiwaf-0.1.9.0.1}/aiwaf.egg-info/SOURCES.txt

@@ -20,6 +20,7 @@ aiwaf.egg-info/top_level.txt
 aiwaf/management/__init__.py
 aiwaf/management/commands/__init__.py
 aiwaf/management/commands/add_ipexemption.py
+aiwaf/management/commands/aiwaf_diagnose.py
 aiwaf/management/commands/aiwaf_logging.py
 aiwaf/management/commands/aiwaf_reset.py
 aiwaf/management/commands/detect_and_train.py

{aiwaf-0.1.8.9 → aiwaf-0.1.9.0.1}/pyproject.toml

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

{aiwaf-0.1.8.9 → aiwaf-0.1.9.0.1}/setup.py

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

aiwaf-0.1.8.9/aiwaf/__init__.py

@@ -1 +0,0 @@
-default_app_config = "aiwaf.apps.AiwafConfig"