aiwaf 0.1.9.1.4__py3-none-any.whl → 0.1.9.1.6__py3-none-any.whl

aiwaf/__init__.py

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

aiwaf/management/commands/check_dependencies.py

@@ -40,6 +40,11 @@ class Command(BaseCommand):
             action='store_true',
             help='Show what would be upgraded without actually upgrading'
         )
+        parser.add_argument(
+            '--update-requirements',
+            action='store_true',
+            help='Update requirements.txt with upgraded package versions'
+        )
 
     def handle(self, *args, **options):
         self.stdout.write(self.style.SUCCESS('🔍 Checking project dependencies...\n'))
@@ -61,7 +66,10 @@ class Command(BaseCommand):
                 self.check_compatibility(results)
             
             if options['upgrade']:
-                self.perform_safe_upgrade(results, options['dry_run'])
+                upgraded_packages = self.perform_safe_upgrade(results, options['dry_run'])
+                
+                if options['update_requirements'] and upgraded_packages and not options['dry_run']:
+                    self.update_requirements_file(upgraded_packages)
             
             if options['check_security']:
                 self.check_security_vulnerabilities(dependencies)
@@ -444,6 +452,7 @@ class Command(BaseCommand):
         # Get packages that can be safely upgraded
         safe_upgrades = []
         blocked_upgrades = []
+        upgraded_packages = []  # Track successfully upgraded packages
         
         for pkg in results:
             if pkg['status'] == 'outdated' and pkg['name'].lower() != 'aiwaf':
@@ -479,7 +488,7 @@ class Command(BaseCommand):
         
         if not safe_upgrades:
             self.stdout.write(self.style.NOTICE("ℹ️  No safe upgrades available at this time."))
-            return
+            return []
         
         # Execute upgrades
         if dry_run:
@@ -489,7 +498,11 @@ class Command(BaseCommand):
                 else u['package']['name'] 
                 for u in safe_upgrades
             ])
-            self.stdout.write(f"Command that would be executed:\n   {upgrade_cmd}")
+            self.stdout.write(f"Commands that would be executed:")
+            self.stdout.write(f"   {upgrade_cmd}")
+            self.stdout.write(f"   pip cache purge  # Clear pip cache after upgrades")
+            return []
+            return []
         else:
             self.stdout.write(self.style.HTTP_INFO("\n🚀 Executing safe upgrades..."))
             success_count = 0
@@ -510,6 +523,11 @@ class Command(BaseCommand):
                     if result.returncode == 0:
                         self.stdout.write(self.style.SUCCESS(f"   ✅ {pkg_name} upgraded successfully"))
                         success_count += 1
+                        upgraded_packages.append({
+                            'name': pkg_name,
+                            'old_version': upgrade['package']['installed'],
+                            'new_version': target_version
+                        })
                     else:
                         self.stdout.write(self.style.ERROR(f"   ❌ Failed to upgrade {pkg_name}: {result.stderr}"))
                 
@@ -521,10 +539,26 @@ class Command(BaseCommand):
             self.stdout.write(f"\n🎉 Upgrade complete: {success_count}/{len(safe_upgrades)} packages upgraded successfully")
             
             if success_count > 0:
+                # Clear pip cache after successful upgrades
+                self.stdout.write(self.style.HTTP_INFO("\n🧹 Clearing pip cache..."))
+                try:
+                    cache_result = subprocess.run(['pip', 'cache', 'purge'], 
+                                                capture_output=True, text=True, timeout=60)
+                    if cache_result.returncode == 0:
+                        self.stdout.write(self.style.SUCCESS("   ✅ Pip cache cleared successfully"))
+                    else:
+                        self.stdout.write(self.style.WARNING(f"   ⚠️  Cache clear warning: {cache_result.stderr}"))
+                except subprocess.TimeoutExpired:
+                    self.stdout.write(self.style.WARNING("   ⚠️  Pip cache clear timed out"))
+                except Exception as e:
+                    self.stdout.write(self.style.WARNING(f"   ⚠️  Could not clear pip cache: {e}"))
+                
                 self.stdout.write(self.style.HTTP_INFO("\n💡 Recommendations after upgrade:"))
                 self.stdout.write("   1. Run tests to ensure everything works correctly")
                 self.stdout.write("   2. Run 'python manage.py check_dependencies' again to verify")
                 self.stdout.write("   3. Consider running 'python manage.py detect_and_train' to retrain with new packages")
+            
+            return upgraded_packages
 
     def get_aiwaf_compatibility_constraints(self):
         """Get AIWAF's compatibility constraints to ensure stability"""
@@ -692,6 +726,91 @@ class Command(BaseCommand):
         
         return {'safe': True, 'reason': 'No compatibility issues detected'}
 
+    def update_requirements_file(self, upgraded_packages):
+        """Update requirements.txt with new package versions"""
+        self.stdout.write(self.style.HTTP_INFO("\n📝 Updating requirements.txt..."))
+        
+        requirements_path = os.path.join(settings.BASE_DIR, 'requirements.txt')
+        
+        if not os.path.exists(requirements_path):
+            self.stdout.write(self.style.WARNING(f"   ⚠️  requirements.txt not found at {requirements_path}"))
+            self.stdout.write("   💡 You can create one manually with updated versions")
+            return
+        
+        try:
+            # Read current requirements.txt
+            with open(requirements_path, 'r') as f:
+                lines = f.readlines()
+            
+            # Create backup
+            backup_path = requirements_path + '.backup'
+            with open(backup_path, 'w') as f:
+                f.writelines(lines)
+            self.stdout.write(f"   📋 Backup created: {backup_path}")
+            
+            # Update lines with new versions
+            updated_lines = []
+            updated_count = 0
+            
+            for line in lines:
+                original_line = line.strip()
+                updated_line = line
+                
+                if original_line and not original_line.startswith('#'):
+                    # Check if this line contains an upgraded package
+                    for pkg in upgraded_packages:
+                        pkg_name = pkg['name']
+                        new_version = pkg['new_version']
+                        
+                        # Check various formats: package>=version, package==version, etc.
+                        if self.line_contains_package(original_line, pkg_name):
+                            # Update the line with new version
+                            updated_line = self.update_package_line(original_line, pkg_name, new_version)
+                            if updated_line != original_line:
+                                updated_count += 1
+                                self.stdout.write(f"   📦 {pkg_name}: {original_line} → {updated_line.strip()}")
+                            break
+                
+                updated_lines.append(updated_line)
+            
+            # Write updated requirements.txt
+            with open(requirements_path, 'w') as f:
+                f.writelines(updated_lines)
+            
+            if updated_count > 0:
+                self.stdout.write(self.style.SUCCESS(f"\n   ✅ Updated {updated_count} packages in requirements.txt"))
+                self.stdout.write(f"   💾 Original backed up as: {backup_path}")
+            else:
+                self.stdout.write(self.style.NOTICE("   ℹ️  No package lines found to update in requirements.txt"))
+                
+        except Exception as e:
+            self.stdout.write(self.style.ERROR(f"   ❌ Error updating requirements.txt: {e}"))
+
+    def line_contains_package(self, line, package_name):
+        """Check if a requirements line contains the specified package"""
+        # Remove comments and whitespace
+        line = line.split('#')[0].strip()
+        if not line:
+            return False
+        
+        # Check if line starts with package name followed by version specifier
+        return (line.lower().startswith(package_name.lower()) and 
+                len(line) > len(package_name) and 
+                line[len(package_name)] in ['=', '<', '>', '!', '~', ' ', '\t'])
+
+    def update_package_line(self, line, package_name, new_version):
+        """Update a requirements line with new package version"""
+        # Split line into package part and comment part
+        parts = line.split('#', 1)
+        package_part = parts[0].strip()
+        comment_part = f" #{parts[1]}" if len(parts) > 1 else ""
+        
+        # Extract package name and update with new version
+        # Use >= to allow for future compatible versions
+        updated_package = f"{package_name}>={new_version}"
+        
+        return updated_package + comment_part + '\n'
+
     def check_security_vulnerabilities(self, dependencies):
         """Check for known security vulnerabilities using safety"""
         self.stdout.write(self.style.HTTP_INFO("\n🔒 Checking for security vulnerabilities..."))

{aiwaf-0.1.9.1.4.dist-info → aiwaf-0.1.9.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aiwaf
-Version: 0.1.9.1.4
+Version: 0.1.9.1.6
 Summary: AI-powered Web Application Firewall
 Home-page: https://github.com/aayushgauba/aiwaf
 Author: Aayush Gauba
@@ -232,8 +232,11 @@ python manage.py check_dependencies --upgrade --dry-run
 # Actually upgrade packages safely
 python manage.py check_dependencies --upgrade
 
+# Upgrade packages and update requirements.txt
+python manage.py check_dependencies --upgrade --update-requirements
+
 # Full workflow: check, upgrade, and scan for vulnerabilities
-python manage.py check_dependencies --upgrade --check-security
+python manage.py check_dependencies --upgrade --check-security --update-requirements
 ```
 
 **Core Features:**
@@ -245,6 +248,8 @@ python manage.py check_dependencies --upgrade --check-security
 - ✅ **Safe package upgrades** (maintains AIWAF stability)
 - ✅ **Dry run mode** for testing upgrade plans
 - ✅ **AIWAF compatibility validation**
+- ✅ **Automatic requirements.txt updates** after successful upgrades
+- ✅ **Pip cache clearing** to prevent cache-related issues
 
 **Safe Upgrade System:**
 
@@ -299,6 +304,18 @@ The upgrade system is designed to maintain AIWAF stability while keeping your pa
 ❌ numpy               1.21.0       ✗ 2.0.1
    🚨 NumPy 2.0+ may cause compatibility issues (max safe: 1.99.99)
 
+🎉 Upgrade complete: 2/2 packages upgraded successfully
+
+🧹 Clearing pip cache...
+   ✅ Pip cache cleared successfully
+
+📝 Updating requirements.txt...
+   📋 Backup created: requirements.txt.backup
+   📦 pandas: pandas>=1.3 → pandas>=1.5.3
+   📦 joblib: joblib>=1.1 → joblib>=1.4.2
+   ✅ Updated 2 packages in requirements.txt
+   💾 Original backed up as: requirements.txt.backup
+
 �💡 To update outdated packages, run:
    pip install --upgrade pandas==1.5.3 joblib
 ```
@@ -309,6 +326,8 @@ The upgrade system is designed to maintain AIWAF stability while keeping your pa
 - 📊 **Conservative Constraints**: Avoids known problematic versions
 - 🧪 **Dry Run Mode**: Test upgrade plans before execution
 - ⚠️ **Clear Blocking Reasons**: Explains why upgrades are blocked
+- 📝 **Requirements.txt Updates**: Automatically updates dependency files
+- 🧹 **Cache Management**: Clears pip cache after successful upgrades
 
 **Recommended Upgrade Workflow:**
 
@@ -324,7 +343,7 @@ The upgrade system is designed to maintain AIWAF stability while keeping your pa
 
 3. **Execute safe upgrades:**
    ```bash
-   python manage.py check_dependencies --upgrade
+   python manage.py check_dependencies --upgrade --update-requirements
    ```
 
 4. **Verify after upgrade:**
@@ -347,6 +366,7 @@ The system uses a multi-layer decision process:
 - **Layer 3**: Analyze cross-package dependencies  
 - **Layer 4**: Select highest safe version within constraints
 - **Layer 5**: Execute with error handling and rollback capability
+- **Layer 6**: Clear pip cache and update requirements.txt after success
 
 This will ensure the IP is never blocked by AI‑WAF. You can also manage exemptions via the Django admin interface.
 

{aiwaf-0.1.9.1.4.dist-info → aiwaf-0.1.9.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-aiwaf/__init__.py,sha256=vkCOYsWFm0FB3Cys7T_e9fuFbGyXKYgqS4JU9yTnz0I,220
+aiwaf/__init__.py,sha256=a8AUvwMuCDX-PCu3FtSH2GKroDDvM1zG6q6KbYyuo6k,220
 aiwaf/apps.py,sha256=nCez-Ptlv2kaEk5HenA8b1pATz1VfhrHP1344gwcY1A,142
 aiwaf/blacklist_manager.py,sha256=LYCeKFB-7e_C6Bg2WeFJWFIIQlrfRMPuGp30ivrnhQY,1196
 aiwaf/decorators.py,sha256=IUKOdM_gdroffImRZep1g1wT6gNqD10zGwcp28hsJCs,825
@@ -15,7 +15,7 @@ aiwaf/management/commands/add_ipexemption.py,sha256=sSf3d9hGK9RqqlBYkCrnrd8KZWGT
 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/check_dependencies.py,sha256=dLi3prJjbhzCQzjQND3QLLGzqLJ96XiSLfK8jM0dUpo,31264
+aiwaf/management/commands/check_dependencies.py,sha256=GOZl00pDwW2cJjDvIaCeB3yWxmeYcJDRTIpmOTLvy2c,37204
 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 +27,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.4.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
-aiwaf-0.1.9.1.4.dist-info/METADATA,sha256=peKauMlND1kyqP_pYsLTrumQHXoO5Q65VaylYL5_SlM,21231
-aiwaf-0.1.9.1.4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aiwaf-0.1.9.1.4.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
-aiwaf-0.1.9.1.4.dist-info/RECORD,,
+aiwaf-0.1.9.1.6.dist-info/licenses/LICENSE,sha256=Ir8PX4dxgAcdB0wqNPIkw84fzIIRKE75NoUil9RX0QU,1069
+aiwaf-0.1.9.1.6.dist-info/METADATA,sha256=d9OfrFnIC8H5G1l8_hASPjqdF4Fe3nRk9I6YEBN5RnI,22145
+aiwaf-0.1.9.1.6.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aiwaf-0.1.9.1.6.dist-info/top_level.txt,sha256=kU6EyjobT6UPCxuWpI_BvcHDG0I2tMgKaPlWzVxe2xI,6
+aiwaf-0.1.9.1.6.dist-info/RECORD,,