sqlmap-ai 2.0.5__tar.gz → 2.0.7__tar.gz

{sqlmap_ai-2.0.5/sqlmap_ai.egg-info → sqlmap_ai-2.0.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlmap-ai
-Version: 2.0.5
+Version: 2.0.7
 Summary: AI-powered SQL injection testing tool with multiple AI providers
 Home-page: https://github.com/atiilla/sqlmap-ai
 Author: Atilla

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "sqlmap-ai"
-version = "2.0.5"
+version = "2.0.7"
 description = "AI-powered SQL injection testing tool with multiple AI providers"
 readme = "README.md"
 license = "MIT"

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/setup.py

@@ -1,8 +1,4 @@
 #!/usr/bin/env python3
-"""
-Setup script for SQLMap AI
-Provides backward compatibility and additional installation features
-"""
 
 import os
 import sys
@@ -19,7 +15,7 @@ def read_readme():
 
 # Read requirements from pyproject.toml
 def get_requirements():
-    """Get requirements from pyproject.toml"""
+    
     try:
         import tomllib
     except ImportError:
@@ -33,7 +29,7 @@ def get_requirements():
     return []
 
 def install_sqlmap():
-    """Install SQLMap if not already installed"""
+    
     try:
         # Check if sqlmap is already installed
         result = subprocess.run(
@@ -43,7 +39,7 @@ def install_sqlmap():
             timeout=10
         )
         if result.returncode == 0:
-            print("✅ SQLMap is already installed")
+            print("✓ SQLMap is already installed")
             return True
     except (subprocess.TimeoutExpired, FileNotFoundError, subprocess.SubprocessError):
         pass
@@ -56,19 +52,19 @@ def install_sqlmap():
             capture_output=True,
             text=True
         )
-        print("✅ SQLMap installed successfully")
+        print("✓ SQLMap installed successfully")
         return True
     except subprocess.CalledProcessError as e:
-        print(f"❌ Failed to install SQLMap: {e}")
+        print(f"X Failed to install SQLMap: {e}")
         print("Please install SQLMap manually: pip install sqlmap")
         return False
 
 def create_env_template():
-    """Create .env template if it doesn't exist"""
+    
     # Use current working directory instead of project root
     env_file = Path.cwd() / ".env"
     if env_file.exists():
-        print("✅ .env file already exists")
+        print("✓ .env file already exists")
         return True
     
     print("📝 Creating .env template...")
@@ -108,15 +104,15 @@ DEFAULT_TIMEOUT=300
     try:
         with open(env_file, 'w') as f:
             f.write(env_content)
-        print("✅ .env template created")
+        print("✓ .env template created")
         print("  Please edit .env file and add your API keys")
         return True
     except Exception as e:
-        print(f"❌ Failed to create .env template: {e}")
+        print(f"X Failed to create .env template: {e}")
         return False
 
 def setup_directories():
-    """Create necessary directories"""
+    
     print("📁 Setting up directories...")
     
     directories = [
@@ -128,15 +124,15 @@ def setup_directories():
     try:
         for directory in directories:
             directory.mkdir(exist_ok=True)
-        print("✅ Directories created")
+        print("✓ Directories created")
         return True
     except Exception as e:
-        print(f"❌ Failed to create directories: {e}")
+        print(f"X Failed to create directories: {e}")
         return False
 
 def run_post_install():
-    """Run post-installation tasks"""
-    print("\n🚀 Running post-installation tasks...")
+    
+    print("\nRunning post-installation tasks...")
     
     success = True
     
@@ -153,13 +149,13 @@ def run_post_install():
         success = False
     
     if success:
-        print("\n🎉 Installation completed successfully!")
+        print("\nInstallation completed successfully!")
         print("\nNext steps:")
         print("1. Edit .env file and add your API keys")
         print("2. Run: sqlmap-ai --config-wizard")
         print("3. Run: sqlmap-ai --help")
     else:
-        print("\n⚠️  Installation completed with issues")
+        print("\n⚠️ Installation completed with issues")
         print("Please resolve the issues above before using SQLMap AI")
     
     return success
@@ -171,12 +167,12 @@ if __name__ == "__main__":
         # Run setup first
         setup(
             name="sqlmap-ai",
-            version="2.0.0",
+            version="2.0.7",
             description="AI-powered SQL injection testing tool with multiple AI providers",
             long_description=read_readme(),
             long_description_content_type="text/markdown",
             author="Atilla",
-            author_email="atiilla@example.com",
+            author_email="attilla@tuta.io",
             url="https://github.com/atiilla/sqlmap-ai",
             packages=find_packages(include=["sqlmap_ai*", "utils*"]),
             include_package_data=True,
@@ -239,7 +235,7 @@ if __name__ == "__main__":
             long_description=read_readme(),
             long_description_content_type="text/markdown",
             author="Atilla",
-            author_email="atiilla@example.com",
+            author_email="attilla@tuta.io",
             url="https://github.com/atiilla/sqlmap-ai",
             packages=find_packages(include=["sqlmap_ai*", "utils*"]),
             include_package_data=True,

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/sqlmap_ai/__init__.py

@@ -1,10 +1,6 @@
-"""
-SQLMap AI - AI-powered SQL injection testing tool
-"""
-
-__version__ = "2.0.0"
+__version__ = "2.0.7"
 __author__ = "Atilla"
-__email__ = "atiilla@example.com"
+__email__ = "atilla@tuta.io"
 __description__ = "AI-powered SQL injection testing tool with multiple AI providers"
 
 # Import main components

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/sqlmap_ai/adaptive_testing.py

@@ -24,7 +24,7 @@ class AdaptiveTestingEngine:
         self.test_parameter = test_parameter  # Specific parameter(s) to test
 
     def _add_param_option(self, options: List[str]) -> List[str]:
-        """Add -p parameter option if specified"""
+        
         if self.test_parameter:
             options.append(f"-p {self.test_parameter}")
         return options

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/sqlmap_ai/advanced_reporting.py

@@ -1,8 +1,3 @@
-"""
-Advanced reporting system with HTML, PDF, and interactive visualizations.
-Provides comprehensive vulnerability analysis and remediation guidance.
-"""
-
 import json
 import time
 import os
@@ -36,7 +31,7 @@ except ImportError:
 
 @dataclass
 class VulnerabilityDetails:
-    """Detailed vulnerability information"""
+    
     parameter: str
     injection_type: str
     payload: str
@@ -50,7 +45,7 @@ class VulnerabilityDetails:
 
 @dataclass
 class ScanStatistics:
-    """Scan performance and statistics"""
+    
     total_requests: int
     successful_injections: int
     false_positives: int
@@ -62,7 +57,7 @@ class ScanStatistics:
 
 @dataclass
 class RemediationGuidance:
-    """Remediation recommendations"""
+    
     immediate_actions: List[str]
     long_term_fixes: List[str]
     secure_coding_practices: List[str]
@@ -71,7 +66,7 @@ class RemediationGuidance:
 
 
 class VulnerabilityAssessment:
-    """Vulnerability risk assessment engine"""
+    
     
     def __init__(self):
         self.risk_factors = {
@@ -106,7 +101,7 @@ class VulnerabilityAssessment:
         }
     
     def assess_vulnerability(self, vuln_data: Dict[str, Any]) -> VulnerabilityDetails:
-        """Assess vulnerability risk and impact"""
+        
         
         # Determine injection type
         injection_type = self._determine_injection_type(vuln_data)
@@ -134,7 +129,7 @@ class VulnerabilityAssessment:
         )
     
     def _determine_injection_type(self, vuln_data: Dict[str, Any]) -> str:
-        """Determine the type of SQL injection"""
+        
         techniques = vuln_data.get('techniques', [])
         
         if 'time-based blind' in str(techniques).lower():
@@ -151,7 +146,7 @@ class VulnerabilityAssessment:
             return "Generic SQL Injection"
     
     def _calculate_risk_score(self, vuln_data: Dict[str, Any]) -> int:
-        """Calculate numerical risk score"""
+        
         base_score = 60  # Base vulnerability score
         
         # Database privileges factor
@@ -180,7 +175,7 @@ class VulnerabilityAssessment:
         return min(base_score, 100)
     
     def _get_risk_level(self, score: int) -> str:
-        """Convert risk score to risk level"""
+        
         if score >= 90:
             return "CRITICAL"
         elif score >= 70:
@@ -193,7 +188,7 @@ class VulnerabilityAssessment:
             return "INFO"
     
     def _assess_exploitation_complexity(self, vuln_data: Dict[str, Any]) -> str:
-        """Assess how difficult the vulnerability is to exploit"""
+        
         factors = 0
         
         # WAF presence increases complexity
@@ -226,7 +221,7 @@ class VulnerabilityAssessment:
             return "Trivial"
     
     def _calculate_remediation_priority(self, risk_score: int, complexity: str) -> str:
-        """Calculate remediation priority"""
+        
         if risk_score >= 90:
             return "IMMEDIATE"
         elif risk_score >= 70:
@@ -240,7 +235,7 @@ class VulnerabilityAssessment:
 
 
 class AdvancedReportGenerator:
-    """Advanced report generator with multiple output formats"""
+    
     
     def __init__(self):
         self.vulnerability_assessor = VulnerabilityAssessment()
@@ -249,7 +244,7 @@ class AdvancedReportGenerator:
         self._create_templates()
     
     def _create_templates(self):
-        """Create HTML templates for reports"""
+        
         if not HAS_JINJA2:
             return
         
@@ -395,7 +390,7 @@ class AdvancedReportGenerator:
         output_format: str = "html",
         output_path: Optional[str] = None
     ) -> str:
-        """Generate comprehensive vulnerability report"""
+        
         
         # Process scan data
         vulnerabilities = self._process_vulnerabilities(scan_data)
@@ -411,7 +406,7 @@ class AdvancedReportGenerator:
             )
     
     def _process_vulnerabilities(self, scan_data: Dict[str, Any]) -> List[VulnerabilityDetails]:
-        """Process scan data to extract vulnerability details"""
+        
         vulnerabilities = []
         
         scan_history = scan_data.get('scan_history', [])
@@ -441,7 +436,7 @@ class AdvancedReportGenerator:
         scan_data: Dict[str, Any], 
         vulnerabilities: List[VulnerabilityDetails]
     ) -> Dict[str, Any]:
-        """Generate scan summary statistics"""
+        
         
         # Calculate overall risk
         if vulnerabilities:
@@ -482,7 +477,7 @@ class AdvancedReportGenerator:
         }
     
     def _generate_remediation_guidance(self, vulnerabilities: List[VulnerabilityDetails]) -> RemediationGuidance:
-        """Generate comprehensive remediation guidance"""
+        
         
         immediate_actions = [
             "Immediately patch or disable vulnerable endpoints",
@@ -542,7 +537,7 @@ class AdvancedReportGenerator:
         remediation: RemediationGuidance,
         output_path: Optional[str] = None
     ) -> str:
-        """Generate HTML report"""
+        
         
         if not HAS_JINJA2:
             return self._generate_simple_html_report(scan_data, vulnerabilities, scan_summary)
@@ -587,7 +582,7 @@ class AdvancedReportGenerator:
         vulnerabilities: List[VulnerabilityDetails],
         scan_summary: Dict[str, Any]
     ) -> str:
-        """Generate simple HTML report without Jinja2"""
+        
         
         html_content = f"""
         <!DOCTYPE html>
@@ -653,7 +648,7 @@ class AdvancedReportGenerator:
         remediation: RemediationGuidance,
         output_path: Optional[str] = None
     ) -> str:
-        """Generate JSON report"""
+        
         
         report_data = {
             "metadata": {
@@ -684,7 +679,7 @@ class AdvancedReportGenerator:
 
     
     def _generate_timeline_chart(self, scan_history: List[Dict[str, Any]]) -> str:
-        """Generate timeline chart of scan steps"""
+        
         
         if not HAS_PLOTLY:
             return ""

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/sqlmap_ai/ai_analyzer.py

@@ -172,7 +172,7 @@ def ai_suggest_next_steps(report, scan_history=None, extracted_data=None, ai_pro
     return valid_options
 
 def create_simple_prompt(report, structured_info, scan_history=None, extracted_data=None):
-    """Create a simpler prompt for Ollama to avoid timeouts"""
+    
     prompt = """
     You are a SQLMap expert. Analyze this SQL injection scan result and suggest the next steps.
 

{sqlmap_ai-2.0.5 → sqlmap_ai-2.0.7}/sqlmap_ai/cli.py

@@ -1,9 +1,4 @@
 #!/usr/bin/env python3
-"""
-SQLMap AI CLI Entry Point
-Handles installation tasks and provides the main interface
-"""
-
 import sys
 import os
 import subprocess
@@ -16,7 +11,6 @@ project_root = Path(__file__).parent.parent
 sys.path.insert(0, str(project_root))
 
 def check_sqlmap_installation() -> bool:
-    """Check if SQLMap is installed and accessible"""
     try:
         # Try to run sqlmap --version
         result = subprocess.run(
@@ -30,14 +24,13 @@ def check_sqlmap_installation() -> bool:
         return False
 
 def install_sqlmap() -> bool:
-    """Install SQLMap if not already installed"""
-    print("🔍 Checking SQLMap installation...")
+    print("[INFO] Checking SQLMap installation...")
     
     if check_sqlmap_installation():
-        print("✅ SQLMap is already installed")
+        print("[SUCCESS] SQLMap is already installed")
         return True
     
-    print("📦 SQLMap not found. Installing...")
+    print("[INFO] SQLMap not found. Installing...")
     
     # Try pip installation first
     try:
@@ -49,10 +42,10 @@ def install_sqlmap() -> bool:
             timeout=60
         )
         if result.returncode == 0:
-            print("✅ SQLMap installed via pip")
+            print("[SUCCESS] SQLMap installed via pip")
             return True
     except (subprocess.TimeoutExpired, subprocess.SubprocessError) as e:
-        print(f"  ❌ Pip installation failed: {e}")
+        print(f"  [ERROR] Pip installation failed: {e}")
     
     # Try git clone as fallback
     try:
@@ -68,12 +61,12 @@ def install_sqlmap() -> bool:
                 timeout=120
             )
             if result.returncode == 0:
-                print("✅ SQLMap cloned from repository")
+                print("[SUCCESS] SQLMap cloned from repository")
                 return True
     except (subprocess.TimeoutExpired, subprocess.SubprocessError) as e:
-        print(f"  ❌ Git clone failed: {e}")
+        print(f"  [ERROR] Git clone failed: {e}")
     
-    print("❌ Failed to install SQLMap automatically")
+    print("[ERROR] Failed to install SQLMap automatically")
     print("Please install SQLMap manually:")
     print("  pip install sqlmap")
     print("  or")
@@ -81,14 +74,14 @@ def install_sqlmap() -> bool:
     return False
 
 def create_env_template() -> bool:
-    """Create .env template if it doesn't exist"""
+    
     # Use current working directory instead of project root
     env_file = Path.cwd() / ".env"
     if env_file.exists():
-        print("✅ .env file already exists")
+        print("[SUCCESS] .env file already exists")
         return True
     
-    print("📝 Creating .env template...")
+    print("[INFO] Creating .env template...")
     
     env_content = """# SQLMap AI Configuration
 # Copy this to .env file and fill in your API keys
@@ -125,16 +118,16 @@ DEFAULT_TIMEOUT=300
     try:
         with open(env_file, 'w') as f:
             f.write(env_content)
-        print("✅ .env template created")
+        print("[SUCCESS] .env template created")
         print("  Please edit .env file and add your API keys")
         return True
     except Exception as e:
-        print(f"❌ Failed to create .env template: {e}")
+        print(f"[ERROR] Failed to create .env template: {e}")
         return False
 
 def setup_directories() -> bool:
-    """Create necessary directories"""
-    print("📁 Setting up directories...")
+    
+    print("[INFO] Setting up directories...")
     
     directories = [
         Path.cwd() / "reports",
@@ -145,25 +138,25 @@ def setup_directories() -> bool:
     try:
         for directory in directories:
             directory.mkdir(exist_ok=True)
-        print("✅ Directories created")
+        print("[SUCCESS] Directories created")
         return True
     except Exception as e:
-        print(f"❌ Failed to create directories: {e}")
+        print(f"[ERROR] Failed to create directories: {e}")
         return False
 
 def run_installation_checks() -> bool:
-    """Run all installation checks and setup tasks"""
-    print("🚀 SQLMap AI Installation Check")
+    
+    print("SQLMap AI Installation Check")
     print("=" * 50)
     
     success = True
     
     # Check Python version
     if sys.version_info < (3, 8):
-        print("❌ Python 3.8 or higher is required")
+        print("[ERROR] Python 3.8 or higher is required")
         return False
     
-    print(f"✅ Python {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}")
+    print(f"[SUCCESS] Python {sys.version_info.major}.{sys.version_info.minor}.{sys.version_info.micro}")
     
     # Install SQLMap
     if not install_sqlmap():
@@ -178,37 +171,45 @@ def run_installation_checks() -> bool:
         success = False
     
     if success:
-        print("\n🎉 Installation check completed successfully!")
+        print("\nInstallation check completed successfully!")
         print("\nNext steps:")
         print("1. Edit .env file and add your API keys")
         print("2. Run: sqlmap-ai --config-wizard")
         print("3. Run: sqlmap-ai --help")
     else:
-        print("\n⚠️  Installation check completed with issues")
+        print("\n[WARNING] Installation check completed with issues")
         print("Please resolve the issues above before using SQLMap AI")
     
     return success
 
 def main():
-    """Main CLI entry point"""
-    # Check if this is the first run (installation check)
-    if len(sys.argv) == 1 or "--install-check" in sys.argv:
-        if run_installation_checks():
-            sys.exit(0)
-        else:
-            sys.exit(1)
     
-    # Import and run the main application
+    # Import and run the main application with new startup system
     try:
-        from run import main as run_main
+        # Import the main run.py from project root
+        import sys
+        from pathlib import Path
+        
+        # Add project root to path
+        project_root = Path(__file__).parent.parent
+        sys.path.insert(0, str(project_root))
+        
+        from sqlmap_ai.run import main as run_main
         run_main()
+        
     except ImportError as e:
-        print(f"❌ Failed to import main application: {e}")
-        print("Please ensure all dependencies are installed:")
-        print("  pip install -e .")
-        sys.exit(1)
+        print(f"[ERROR] Failed to import main application: {e}")
+        print("Running fallback installation check...")
+        
+        # Fallback to old installation check if new system fails
+        if run_installation_checks():
+            print("\nTry running 'sqlmap-ai' again after installation completes.")
+            sys.exit(0)
+        else:
+            sys.exit(1)
+            
     except Exception as e:
-        print(f"❌ Application error: {e}")
+        print(f"[ERROR] Application error: {e}")
         sys.exit(1)
 
 if __name__ == "__main__":