aiwaf 0.1.9.0.6__py3-none-any.whl → 0.1.9.0.7__py3-none-any.whl

aiwaf/__init__.py

@@ -1,6 +1,6 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.0.6"
+__version__ = "0.1.9.0.7"
 
 # Note: Middleware classes are available from aiwaf.middleware
 # Import them only when needed to avoid circular imports during Django app loading

aiwaf/management/commands/add_ipexemption.py

@@ -17,7 +17,7 @@ class Command(BaseCommand):
         if store.is_exempted(ip):
             self.stdout.write(self.style.WARNING(f'IP {ip} is already exempted.'))
         else:
-            store.add_ip(ip, reason)
+            store.add_exemption(ip, reason)
             self.stdout.write(self.style.SUCCESS(f'IP {ip} added to exemption list.'))
             if reason:
                 self.stdout.write(self.style.SUCCESS(f'Reason: {reason}'))

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_exemption(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/management/commands/test_exemption.py

@@ -63,7 +63,7 @@ class Command(BaseCommand):
             
             # Create the exemption
             try:
-                exemption_store.add_ip(test_ip, "Test exemption from debug")
+                exemption_store.add_exemption(test_ip, "Test exemption from debug")
                 self.stdout.write(self.style.SUCCESS("✅ Created test exemption"))
             except Exception as e:
                 self.stdout.write(self.style.ERROR(f"❌ Failed to create exemption: {e}"))

{aiwaf-0.1.9.0.6.dist-info → aiwaf-0.1.9.0.7.dist-info}/METADATA

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

{aiwaf-0.1.9.0.6.dist-info → aiwaf-0.1.9.0.7.dist-info}/RECORD

@@ -1,4 +1,4 @@
-aiwaf/__init__.py,sha256=CaCYDiHa_an8eS3dgJs0d6bCNQN7KUMc2sJBLEXNt8I,220
+aiwaf/__init__.py,sha256=5uH_XGymeTFxegNihwozJE8d-Y70jvwpKhXvzZ9HkQI,220
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=LYCeKFB-7e_C6Bg2WeFJWFIIQlrfRMPuGp30ivrnhQY,1196
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
@@ -11,22 +11,23 @@ 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=srgdVPDJtF7G9GGIqaZ7L3qTuNheoS_uwlhlRO4W2bc,945
+aiwaf/management/commands/add_ipexemption.py,sha256=sSf3d9hGK9RqqlBYkCrnrd8KZWGT-derSpoWnEY4H60,952
 aiwaf/management/commands/aiwaf_diagnose.py,sha256=nXFRhq66N4QC3e4scYJ2sUngJce-0yDxtBO3R2BllRM,6134
 aiwaf/management/commands/aiwaf_logging.py,sha256=FCIqULn2tii2vD9VxL7vk3PV4k4vr7kaA00KyaCExYY,7692
 aiwaf/management/commands/aiwaf_reset.py,sha256=0FIBqpZS8xgFFvAKJ-0zAC_-QNQwRkOHpXb8N-OdFr8,3740
 aiwaf/management/commands/clear_cache.py,sha256=cdnuTgxkhKLqT_6k6yTcEBlREovNRQxAE51ceXlGYMA,647
+aiwaf/management/commands/debug_csv.py,sha256=Lddqp37mIn0zdvHf4GbuNTWYyJ5h8bumDcGmFSAioi0,6801
 aiwaf/management/commands/detect_and_train.py,sha256=-o-LZ7QZ5GeJPCekryox1DGXKMmFEkwwrcDsiM166K0,269
 aiwaf/management/commands/diagnose_blocking.py,sha256=HKb_FdN4b6QdyqNDf54B08I5jyWfrv9Mh-SFBrr3LbU,4140
 aiwaf/management/commands/regenerate_model.py,sha256=SUy7TCTTDJy4kRZNAbTIVBxSmljUaAC6ms0JTfSO6BE,3445
 aiwaf/management/commands/setup_models.py,sha256=JzuxwAqO3e-8L4PdFlXkyEQmOA8EGCXBfaOwfCNv1Gg,1678
-aiwaf/management/commands/test_exemption.py,sha256=qX7GMnpGFhRzC8cTtJMVN7pvjXziEWJ934cyOLN3cqs,5511
+aiwaf/management/commands/test_exemption.py,sha256=ENmWFMJE8iQyzJGAPdw_5PUPknLajE8JjwHgH8DCFsE,5518
 aiwaf/management/commands/test_exemption_fix.py,sha256=ngyGaHUCmQQ6y--6j4q1viZJtR-RvI526yDqvEaEXPs,2553
 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.0.6.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.9.0.6.dist-info/METADATA,sha256=xhkKJlK7aW0mObUkrIboi4E4GTP1aLpyoo_I0QUJs4E,13763
-aiwaf-0.1.9.0.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.9.0.6.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.9.0.6.dist-info/RECORD,,
+aiwaf-0.1.9.0.7.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.0.7.dist-info/METADATA,sha256=rFNbBTlRRki2Prz5URgHw8yYmWTTSsxxBTeE5evUabE,13763
+aiwaf-0.1.9.0.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.0.7.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.0.7.dist-info/RECORD,,