aiwaf 0.1.9.0.3__tar.gz → 0.1.9.0.4__tar.gz

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/PKG-INFO

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

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/aiwaf/__init__.py

@@ -1,6 +1,6 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.0.3"
+__version__ = "0.1.9.0.4"
 
 # 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.4/aiwaf/management/commands/debug_csv.py

@@ -0,0 +1,155 @@
+from django.core.management.base import BaseCommand
+import os
+import csv
+
+class Command(BaseCommand):
+    help = 'Debug and fix AI-WAF CSV functionality'
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '--test-ip',
+            type=str,
+            help='Test IP address to add to exemption list',
+            default='127.0.0.1'
+        )
+        parser.add_argument(
+            '--fix',
+            action='store_true',
+            help='Attempt to fix identified issues',
+        )
+
+    def handle(self, *args, **options):
+        self.stdout.write(self.style.HTTP_INFO("🔍 AI-WAF CSV Debug & Fix"))
+        self.stdout.write("")
+        
+        # Check storage mode
+        from django.conf import settings
+        storage_mode = getattr(settings, 'AIWAF_STORAGE_MODE', 'models')
+        csv_dir = getattr(settings, 'AIWAF_CSV_DATA_DIR', 'aiwaf_data')
+        
+        self.stdout.write(f"Storage Mode: {storage_mode}")
+        self.stdout.write(f"CSV Directory: {csv_dir}")
+        self.stdout.write("")
+        
+        # Check middleware logging
+        middleware_logging = getattr(settings, 'AIWAF_MIDDLEWARE_LOGGING', False)
+        middleware_log = getattr(settings, 'AIWAF_MIDDLEWARE_LOG', 'aiwaf_requests.log')
+        
+        self.stdout.write(f"Middleware Logging: {middleware_logging}")
+        self.stdout.write(f"Middleware Log File: {middleware_log}")
+        self.stdout.write("")
+        
+        # Check if CSV directory exists
+        if os.path.exists(csv_dir):
+            self.stdout.write(self.style.SUCCESS(f"✅ CSV directory exists: {csv_dir}"))
+        else:
+            self.stdout.write(self.style.ERROR(f"❌ CSV directory missing: {csv_dir}"))
+            if options['fix']:
+                os.makedirs(csv_dir, exist_ok=True)
+                self.stdout.write(self.style.SUCCESS(f"✅ Created CSV directory: {csv_dir}"))
+        
+        # Check CSV files
+        csv_files = ['blacklist.csv', 'exemptions.csv', 'keywords.csv']
+        for filename in csv_files:
+            filepath = os.path.join(csv_dir, filename)
+            if os.path.exists(filepath):
+                # Count entries
+                try:
+                    with open(filepath, 'r', newline='', encoding='utf-8') as f:
+                        reader = csv.reader(f)
+                        rows = list(reader)
+                        entry_count = len(rows) - 1 if rows else 0  # Subtract header
+                        self.stdout.write(self.style.SUCCESS(f"✅ {filename}: {entry_count} entries"))
+                except Exception as e:
+                    self.stdout.write(self.style.ERROR(f"❌ {filename}: Error reading - {e}"))
+            else:
+                self.stdout.write(self.style.WARNING(f"⚠️  {filename}: Not found"))
+        
+        self.stdout.write("")
+        
+        # Test storage functionality
+        self.stdout.write(self.style.HTTP_INFO("🧪 Testing Storage Functions"))
+        
+        try:
+            from aiwaf.storage import get_exemption_store, get_blacklist_store, get_keyword_store
+            
+            # Test exemption store
+            exemption_store = get_exemption_store()
+            self.stdout.write(f"Exemption Store: {exemption_store.__name__}")
+            
+            # Test blacklist store
+            blacklist_store = get_blacklist_store()
+            self.stdout.write(f"Blacklist Store: {blacklist_store.__name__}")
+            
+            # Test keyword store
+            keyword_store = get_keyword_store()
+            self.stdout.write(f"Keyword Store: {keyword_store.__name__}")
+            
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f"❌ Storage import failed: {e}"))
+            return
+        
+        self.stdout.write("")
+        
+        # Test exemption functionality
+        test_ip = options['test_ip']
+        self.stdout.write(f"🧪 Testing exemption with IP: {test_ip}")
+        
+        try:
+            # Check if already exempted
+            is_exempted_before = exemption_store.is_exempted(test_ip)
+            self.stdout.write(f"Before: IP {test_ip} exempted = {is_exempted_before}")
+            
+            # Add to exemption
+            exemption_store.add_ip(test_ip, "Test exemption from debug command")
+            self.stdout.write(f"✅ Added {test_ip} to exemption list")
+            
+            # Check if now exempted
+            is_exempted_after = exemption_store.is_exempted(test_ip)
+            self.stdout.write(f"After: IP {test_ip} exempted = {is_exempted_after}")
+            
+            if is_exempted_after:
+                self.stdout.write(self.style.SUCCESS("✅ Exemption functionality working!"))
+            else:
+                self.stdout.write(self.style.ERROR("❌ Exemption functionality not working!"))
+                
+            # List all exemptions
+            all_exemptions = exemption_store.get_all()
+            self.stdout.write(f"Total exemptions: {len(all_exemptions)}")
+            
+            for exemption in all_exemptions:
+                self.stdout.write(f"  - {exemption.get('ip_address', exemption)}")
+                
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f"❌ Exemption test failed: {e}"))
+        
+        self.stdout.write("")
+        
+        # Check middleware logger file
+        csv_log_file = middleware_log.replace('.log', '.csv')
+        if os.path.exists(csv_log_file):
+            try:
+                with open(csv_log_file, 'r', newline='', encoding='utf-8') as f:
+                    reader = csv.reader(f)
+                    rows = list(reader)
+                    entry_count = len(rows) - 1 if rows else 0
+                    self.stdout.write(self.style.SUCCESS(f"✅ Middleware CSV log: {entry_count} entries"))
+            except Exception as e:
+                self.stdout.write(self.style.ERROR(f"❌ Middleware CSV log error: {e}"))
+        else:
+            self.stdout.write(self.style.WARNING(f"⚠️  Middleware CSV log not found: {csv_log_file}"))
+            self.stdout.write("   Make some requests to generate log entries")
+        
+        # Recommendations
+        self.stdout.write("")
+        self.stdout.write(self.style.HTTP_INFO("💡 Recommendations:"))
+        
+        if storage_mode != 'csv':
+            self.stdout.write("1. Set AIWAF_STORAGE_MODE = 'csv' in settings.py")
+            
+        if not middleware_logging:
+            self.stdout.write("2. Set AIWAF_MIDDLEWARE_LOGGING = True in settings.py")
+            
+        self.stdout.write("3. Add AIWAFLoggerMiddleware to MIDDLEWARE in settings.py")
+        self.stdout.write("4. Make some requests to generate log data")
+        self.stdout.write("5. Run 'python manage.py detect_and_train' to train with data")

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/aiwaf/middleware_logger.py

@@ -28,7 +28,11 @@ class AIWAFLoggerMiddleware(MiddlewareMixin):
     def _ensure_csv_header(self):
         """Ensure CSV file has proper header row"""
         if not os.path.exists(self.csv_file):
-            os.makedirs(os.path.dirname(self.csv_file), exist_ok=True) if os.path.dirname(self.csv_file) else None
+            # Create directory if it doesn't exist
+            csv_dir = os.path.dirname(self.csv_file)
+            if csv_dir and not os.path.exists(csv_dir):
+                os.makedirs(csv_dir, exist_ok=True)
+            
             with open(self.csv_file, 'w', newline='', encoding='utf-8') as f:
                 writer = csv.writer(f)
                 writer.writerow([
@@ -73,6 +77,11 @@ class AIWAFLoggerMiddleware(MiddlewareMixin):
     def _log_to_csv(self, data):
         """Write log entry to CSV file"""
         try:
+            # Ensure directory exists before writing
+            csv_dir = os.path.dirname(self.csv_file)
+            if csv_dir and not os.path.exists(csv_dir):
+                os.makedirs(csv_dir, exist_ok=True)
+                
             with open(self.csv_file, 'a', newline='', encoding='utf-8') as f:
                 writer = csv.writer(f)
                 writer.writerow([

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/aiwaf/storage.py

@@ -167,28 +167,39 @@ class CsvExemptionStore:
     @staticmethod
     def add_ip(ip_address, reason=""):
         ensure_csv_directory()
-        # Check if IP already exists
+        
+        # Check if IP already exists to avoid duplicates
         if CsvExemptionStore.is_exempted(ip_address):
             return
         
         # Add new entry
         new_file = not os.path.exists(EXEMPTION_CSV)
-        with open(EXEMPTION_CSV, "a", newline="", encoding="utf-8") as f:
-            writer = csv.writer(f)
-            if new_file:
-                writer.writerow(["ip_address", "reason", "created_at"])
-            writer.writerow([ip_address, reason, timezone.now().isoformat()])
+        try:
+            with open(EXEMPTION_CSV, "a", newline="", encoding="utf-8") as f:
+                writer = csv.writer(f)
+                if new_file:
+                    writer.writerow(["ip_address", "reason", "created_at"])
+                writer.writerow([ip_address, reason, timezone.now().isoformat()])
+        except Exception as e:
+            print(f"Error writing to exemption CSV: {e}")
+            print(f"File path: {EXEMPTION_CSV}")
+            print(f"Directory exists: {os.path.exists(CSV_DATA_DIR)}")
+            raise
     
     @staticmethod
     def is_exempted(ip_address):
         if not os.path.exists(EXEMPTION_CSV):
             return False
         
-        with open(EXEMPTION_CSV, "r", newline="", encoding="utf-8") as f:
-            reader = csv.DictReader(f)
-            for row in reader:
-                if row["ip_address"] == ip_address:
-                    return True
+        try:
+            with open(EXEMPTION_CSV, "r", newline="", encoding="utf-8") as f:
+                reader = csv.DictReader(f)
+                for row in reader:
+                    if row["ip_address"] == ip_address:
+                        return True
+        except Exception as e:
+            print(f"Error reading exemption CSV: {e}")
+            return False
         return False
     
     @staticmethod

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/aiwaf.egg-info/PKG-INFO

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

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/aiwaf.egg-info/SOURCES.txt

@@ -23,6 +23,7 @@ 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/debug_csv.py
 aiwaf/management/commands/detect_and_train.py
 aiwaf/management/commands/regenerate_model.py
 aiwaf/resources/model.pkl

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/pyproject.toml

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

{aiwaf-0.1.9.0.3 → aiwaf-0.1.9.0.4}/setup.py

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