aiwaf 0.1.9.0__tar.gz → 0.1.9.0.2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.9.0
+Version: 0.1.9.0.2
 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.2}/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.2/aiwaf/__init__.py

@@ -0,0 +1,6 @@
+default_app_config = "aiwaf.apps.AiwafConfig"
+
+__version__ = "0.1.9.0.2"
+
+# 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 → aiwaf-0.1.9.0.2}/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.2}/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.2}/aiwaf/trainer.py

@@ -17,7 +17,7 @@ from .utils import is_exempt_path
 from .storage import get_blacklist_store, get_exemption_store, get_keyword_store
 
 # ─────────── Configuration ───────────
-LOG_PATH   = settings.AIWAF_ACCESS_LOG
+LOG_PATH   = getattr(settings, 'AIWAF_ACCESS_LOG', None)
 MODEL_PATH = os.path.join(os.path.dirname(__file__), "resources", "model.pkl")
 
 STATIC_KW  = [".php", "xmlrpc", "wp-", ".env", ".git", ".bak", "conflg", "shell", "filemanager"]

{aiwaf-0.1.9.0 → aiwaf-0.1.9.0.2}/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.2
 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.2}/pyproject.toml

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

{aiwaf-0.1.9.0 → aiwaf-0.1.9.0.2}/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.2",
     description="AI‑driven, self‑learning Web Application Firewall for Django",
     long_description=long_description,
     long_description_content_type="text/markdown",

aiwaf-0.1.9.0/aiwaf/__init__.py

@@ -1,19 +0,0 @@
-default_app_config = "aiwaf.apps.AiwafConfig"
-
-__version__ = "0.1.9.0"
-
-# 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")