aiwaf 0.1.9.0__tar.gz → 0.1.9.0.1__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.9.0
+Version: 0.1.9.0.1
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -370,6 +370,7 @@ MIDDLEWARE = [
    ```
 
 **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`

{aiwaf-0.1.9.0 → aiwaf-0.1.9.0.1}/README.md

@@ -349,6 +349,7 @@ MIDDLEWARE = [
    ```
 
 **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`

{aiwaf-0.1.9.0 → aiwaf-0.1.9.0.1}/aiwaf/__init__.py

@@ -1,6 +1,11 @@
 default_app_config = "aiwaf.apps.AiwafConfig"
 
-__version__ = "0.1.9.0"
+__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:

{aiwaf-0.1.9.0 → aiwaf-0.1.9.0.1}/aiwaf/management/commands/aiwaf_diagnose.py

@@ -14,6 +14,14 @@ class Command(BaseCommand):
             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

{aiwaf-0.1.9.0 → 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.9.0 → aiwaf-0.1.9.0.1}/aiwaf.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.9.0
+Version: 0.1.9.0.1
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -370,6 +370,7 @@ MIDDLEWARE = [
    ```
 
 **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`

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

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

{aiwaf-0.1.9.0 → 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.9.0",
+    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",