aiwaf 0.1.9.1.7__py3-none-any.whl → 0.1.9.1.9__py3-none-any.whl

aiwaf/__init__.py

@@ -1,6 +1,6 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.1.7"
+__version__ = "0.1.9.1.9"
 
 # 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_reset.py

@@ -1,5 +1,6 @@
 from django.core.management.base import BaseCommand
 from aiwaf.storage import get_blacklist_store, get_exemption_store
+import sys
 
 class Command(BaseCommand):
     help = 'Reset AI-WAF by clearing all blacklist and exemption (whitelist) entries'
@@ -26,12 +27,29 @@ class Command(BaseCommand):
         exemptions_only = options['exemptions_only']
         confirm = options['confirm']
         
-        blacklist_store = get_blacklist_store()
-        exemption_store = get_exemption_store()
+        try:
+            blacklist_store = get_blacklist_store()
+            exemption_store = get_exemption_store()
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f'Error initializing stores: {e}'))
+            return
         
-        # Count current entries
-        blacklist_count = len(blacklist_store.get_all())
-        exemption_count = len(exemption_store.get_all())
+        # Count current entries safely
+        try:
+            blacklist_entries = blacklist_store.get_all()
+            blacklist_count = len(blacklist_entries)
+        except Exception as e:
+            self.stdout.write(self.style.WARNING(f'Warning: Could not count blacklist entries: {e}'))
+            blacklist_count = 0
+            blacklist_entries = []
+        
+        try:
+            exemption_entries = exemption_store.get_all()
+            exemption_count = len(exemption_entries)
+        except Exception as e:
+            self.stdout.write(self.style.WARNING(f'Warning: Could not count exemption entries: {e}'))
+            exemption_count = 0
+            exemption_entries = []
         
         if blacklist_only and exemptions_only:
             self.stdout.write(self.style.ERROR('Cannot use both --blacklist-only and --exemptions-only flags'))
@@ -55,29 +73,47 @@ class Command(BaseCommand):
         self.stdout.write(f"AI-WAF Reset: {action}")
         
         if not confirm:
-            response = input("Are you sure you want to proceed? [y/N]: ")
-            if response.lower() not in ['y', 'yes']:
-                self.stdout.write(self.style.WARNING('Operation cancelled'))
+            try:
+                response = input("Are you sure you want to proceed? [y/N]: ")
+                if response.lower() not in ['y', 'yes']:
+                    self.stdout.write(self.style.WARNING('Operation cancelled'))
+                    return
+            except (EOFError, KeyboardInterrupt):
+                self.stdout.write(self.style.WARNING('\nOperation cancelled'))
                 return
         
         # Perform the reset
-        deleted_counts = {'blacklist': 0, 'exemptions': 0}
+        deleted_counts = {'blacklist': 0, 'exemptions': 0, 'errors': []}
         
         if clear_blacklist:
             # Clear blacklist entries
-            blacklist_entries = blacklist_store.get_all()
-            for entry in blacklist_entries:
-                blacklist_store.remove_ip(entry['ip_address'])
-            deleted_counts['blacklist'] = len(blacklist_entries)
+            try:
+                for entry in blacklist_entries:
+                    try:
+                        blacklist_store.remove_ip(entry['ip_address'])
+                        deleted_counts['blacklist'] += 1
+                    except Exception as e:
+                        deleted_counts['errors'].append(f"Error removing blacklist IP {entry.get('ip_address', 'unknown')}: {e}")
+            except Exception as e:
+                deleted_counts['errors'].append(f"Error clearing blacklist: {e}")
         
         if clear_exemptions:
             # Clear exemption entries
-            exemption_entries = exemption_store.get_all()
-            for entry in exemption_entries:
-                exemption_store.remove_ip(entry['ip_address'])
-            deleted_counts['exemptions'] = len(exemption_entries)
+            try:
+                for entry in exemption_entries:
+                    try:
+                        exemption_store.remove_ip(entry['ip_address'])
+                        deleted_counts['exemptions'] += 1
+                    except Exception as e:
+                        deleted_counts['errors'].append(f"Error removing exemption IP {entry.get('ip_address', 'unknown')}: {e}")
+            except Exception as e:
+                deleted_counts['errors'].append(f"Error clearing exemptions: {e}")
         
         # Report results
+        if deleted_counts['errors']:
+            for error in deleted_counts['errors']:
+                self.stdout.write(self.style.WARNING(f"⚠️  {error}"))
+        
         if clear_blacklist and clear_exemptions:
             self.stdout.write(
                 self.style.SUCCESS(
@@ -93,3 +129,8 @@ class Command(BaseCommand):
             self.stdout.write(
                 self.style.SUCCESS(f"✅ Exemptions cleared: Deleted {deleted_counts['exemptions']} entries")
             )
+        
+        if deleted_counts['errors']:
+            self.stdout.write(
+                self.style.WARNING(f"⚠️  Completed with {len(deleted_counts['errors'])} errors (see above)")
+            )

aiwaf/management/commands/clear_blacklist.py

@@ -0,0 +1,66 @@
+from django.core.management.base import BaseCommand
+from aiwaf.storage import get_blacklist_store
+
+class Command(BaseCommand):
+    help = 'Clear all blacklist entries (fast method)'
+
+    def add_arguments(self, parser):
+        parser.add_argument(
+            '--confirm',
+            action='store_true',
+            help='Skip confirmation prompt'
+        )
+
+    def handle(self, *args, **options):
+        confirm = options['confirm']
+        
+        try:
+            blacklist_store = get_blacklist_store()
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f'Error initializing blacklist store: {e}'))
+            return
+        
+        # Count current entries safely
+        try:
+            blacklist_entries = blacklist_store.get_all()
+            blacklist_count = len(blacklist_entries)
+        except Exception as e:
+            self.stdout.write(self.style.WARNING(f'Warning: Could not count blacklist entries: {e}'))
+            # Try to get count using clear_all method which returns count
+            blacklist_count = "unknown number of"
+        
+        # Show what will be cleared
+        self.stdout.write(f"Clear Blacklist: Will remove {blacklist_count} blacklist entries")
+        
+        if not confirm:
+            try:
+                response = input("Are you sure you want to proceed? [y/N]: ")
+                if response.lower() not in ['y', 'yes']:
+                    self.stdout.write(self.style.WARNING('Operation cancelled'))
+                    return
+            except (EOFError, KeyboardInterrupt):
+                self.stdout.write(self.style.WARNING('\nOperation cancelled'))
+                return
+        
+        # Perform the reset using clear_all for better performance
+        try:
+            if hasattr(blacklist_store, 'clear_all'):
+                deleted_count = blacklist_store.clear_all()
+                self.stdout.write(
+                    self.style.SUCCESS(f"✅ Blacklist cleared: Deleted {deleted_count} entries")
+                )
+            else:
+                # Fallback to individual deletion
+                deleted_count = 0
+                for entry in blacklist_entries:
+                    try:
+                        blacklist_store.remove_ip(entry['ip_address'])
+                        deleted_count += 1
+                    except Exception as e:
+                        self.stdout.write(self.style.WARNING(f"⚠️  Error removing IP {entry.get('ip_address', 'unknown')}: {e}"))
+                
+                self.stdout.write(
+                    self.style.SUCCESS(f"✅ Blacklist cleared: Deleted {deleted_count} entries")
+                )
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f'Error clearing blacklist: {e}'))

aiwaf/storage.py

@@ -132,6 +132,31 @@ class ModelBlacklistStore:
         except Exception:
             return []
 
+    @staticmethod
+    def get_all():
+        """Get all blacklist entries as dictionaries"""
+        _import_models()
+        if BlacklistEntry is None:
+            return []
+        try:
+            return list(BlacklistEntry.objects.values('ip_address', 'reason', 'created_at'))
+        except Exception:
+            return []
+
+    @staticmethod
+    def clear_all():
+        """Clear all blacklist entries"""
+        _import_models()
+        if BlacklistEntry is None:
+            return 0
+        try:
+            count = BlacklistEntry.objects.count()
+            BlacklistEntry.objects.all().delete()
+            return count
+        except Exception as e:
+            print(f"Error clearing all blacklist entries: {e}")
+            return 0
+
 class ModelExemptionStore:
     @staticmethod
     def is_exempted(ip):
@@ -170,6 +195,16 @@ class ModelExemptionStore:
         except Exception as e:
             print(f"Error removing exemption for IP {ip}: {e}")
 
+    @staticmethod
+    def remove_ip(ip):
+        """Remove IP from exemption list (alias for remove_exemption)"""
+        ModelExemptionStore.remove_exemption(ip)
+
+    @staticmethod
+    def add_ip(ip, reason="Manual exemption"):
+        """Add IP to exemption list (alias for add_exemption)"""
+        ModelExemptionStore.add_exemption(ip, reason)
+
     @staticmethod
     def get_all_exempted_ips():
         """Get all exempted IPs"""
@@ -192,6 +227,20 @@ class ModelExemptionStore:
         except Exception:
             return []
 
+    @staticmethod
+    def clear_all():
+        """Clear all exemption entries"""
+        _import_models()
+        if IPExemption is None:
+            return 0
+        try:
+            count = IPExemption.objects.count()
+            IPExemption.objects.all().delete()
+            return count
+        except Exception as e:
+            print(f"Error clearing all exemption entries: {e}")
+            return 0
+
 class ModelKeywordStore:
     @staticmethod
     def add_keyword(keyword, count=1):

{aiwaf-0.1.9.1.7.dist-info → aiwaf-0.1.9.1.9.dist-info}/METADATA

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

{aiwaf-0.1.9.1.7.dist-info → aiwaf-0.1.9.1.9.dist-info}/RECORD

@@ -1,11 +1,11 @@
-aiwaf/__init__.py,sha256=hhKVupfj0NOn_nRM1Ukd6aUP9mL5OkjKLE-tZ3xGdIE,220
+aiwaf/__init__.py,sha256=BGGn_OwueGmxbbWRV-PwE7HGpzB5Ol61jhYI6z4tHug,220
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=LYCeKFB-7e_C6Bg2WeFJWFIIQlrfRMPuGp30ivrnhQY,1196
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
 aiwaf/middleware.py,sha256=EMAQA_Gnz0jv4nevlognT921ZeBEro13J_DSv_mQ3Dw,15482
 aiwaf/middleware_logger.py,sha256=LWZVDAnjh6CGESirA8eMbhGgJKB7lVDGRQqVroH95Lo,4742
 aiwaf/models.py,sha256=vQxgY19BDVMjoO903UNrTZC1pNoLltMU6wbyWPoAEns,2719
-aiwaf/storage.py,sha256=HYSnis7S8ETsos_NxWkd05OoiHXMhIWQy8FcFTqO4vk,8408
+aiwaf/storage.py,sha256=vswojWT8KEH5h24TQ9wwYCsxRUOjaAKudtFJnFxNHKk,9914
 aiwaf/trainer.py,sha256=1RPjWVOdGQ3qSrjFopw8HKu7THVTMvF4nNYouij6i_A,10685
 aiwaf/utils.py,sha256=BJk5vJCYdGPl_4QQiknjhCbkzv5HZCXgFcBJDMJpHok,3390
 aiwaf/management/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -14,8 +14,9 @@ aiwaf/management/commands/add_exemption.py,sha256=U_ByfJw1EstAZ8DaSoRb97IGwYzXs0
 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/aiwaf_reset.py,sha256=wG7EcdPqkxmjF2ivQOmZ7swuvHVJ_OVLgOEijGLvmFs,5586
 aiwaf/management/commands/check_dependencies.py,sha256=GOZl00pDwW2cJjDvIaCeB3yWxmeYcJDRTIpmOTLvy2c,37204
+aiwaf/management/commands/clear_blacklist.py,sha256=Tisedg0EVlc3E01mA3hBZQorwMzc5j1cns-oYshja0g,2770
 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
@@ -27,8 +28,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.1.7.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.9.1.7.dist-info/METADATA,sha256=P3lYeDWOKfjHCBWcaEoFpN5lJJkj0iP_HWIJjAI-hz0,22145
-aiwaf-0.1.9.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.9.1.7.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.9.1.7.dist-info/RECORD,,
+aiwaf-0.1.9.1.9.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.1.9.dist-info/METADATA,sha256=YeyuawG8pPFTBrOOBp8MayiGxCdyywAFvKKMY8dIk-M,22145
+aiwaf-0.1.9.1.9.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.1.9.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.1.9.dist-info/RECORD,,