souleyez 2.43.29__py3-none-any.whl → 3.0.0__py3-none-any.whl

souleyez/ui/errors.py

@@ -26,11 +26,13 @@ class ErrorHelper:
         error_msg(ErrorType.CUSTOM, message="Something went wrong",
                   suggestions=["Try this", "Or this"])
     """
+
     pass
 
 
 class ErrorType:
     """Enum-like class for error types."""
+
     ENGAGEMENT_NOT_FOUND = "engagement_not_found"
     TOOL_NOT_FOUND = "tool_not_found"
     DATABASE_ERROR = "database_error"
@@ -53,128 +55,116 @@ ERROR_DEFINITIONS = {
         "suggestions": [
             "Run 'souleyez interactive' to create or select an engagement",
             "Use '[+] Create New' in the engagement menu",
-            "Check the engagement ID is correct"
+            "Check the engagement ID is correct",
         ],
-        "help_cmd": "souleyez interactive"
+        "help_cmd": "souleyez interactive",
     },
-
     ErrorType.NO_CURRENT_ENGAGEMENT: {
         "message": "No engagement selected",
         "suggestions": [
             "Select an engagement first in the interactive menu",
             "Create a new engagement with '[+] Create New'",
-            "Run 'souleyez interactive' to get started"
+            "Run 'souleyez interactive' to get started",
         ],
-        "help_cmd": "souleyez interactive"
+        "help_cmd": "souleyez interactive",
     },
-
     ErrorType.TOOL_NOT_FOUND: {
         "message": "Tool not installed or not found in PATH",
         "suggestions": [
             "Run 'souleyez setup' to install pentesting tools",
             "Check if the tool is installed: 'which {tool_name}'",
-            "Run 'souleyez doctor' to diagnose installation issues"
+            "Run 'souleyez doctor' to diagnose installation issues",
         ],
-        "help_cmd": "souleyez setup"
+        "help_cmd": "souleyez setup",
     },
-
     ErrorType.DATABASE_ERROR: {
         "message": "Database error occurred",
         "suggestions": [
             "Run 'souleyez doctor' to check database health",
             "Check ~/.souleyez/souleyez.db file permissions",
-            "Try backing up and recreating the database"
+            "Try backing up and recreating the database",
         ],
-        "help_cmd": "souleyez doctor"
+        "help_cmd": "souleyez doctor",
     },
-
     ErrorType.PERMISSION_DENIED: {
         "message": "Permission denied",
         "suggestions": [
             "Check file/directory permissions",
             "Some tools require sudo - try running with elevated privileges",
-            "Run 'souleyez doctor' to check for permission issues"
+            "Run 'souleyez doctor' to check for permission issues",
         ],
-        "help_cmd": "souleyez doctor"
+        "help_cmd": "souleyez doctor",
     },
-
     ErrorType.CONFIG_ERROR: {
         "message": "Configuration error",
         "suggestions": [
             "Check ~/.souleyez/config.json for syntax errors",
             "Run 'souleyez doctor' to validate configuration",
-            "Delete config file to reset to defaults: rm ~/.souleyez/config.json"
+            "Delete config file to reset to defaults: rm ~/.souleyez/config.json",
         ],
-        "help_cmd": "cat ~/.souleyez/config.json"
+        "help_cmd": "cat ~/.souleyez/config.json",
     },
-
     ErrorType.NETWORK_ERROR: {
         "message": "Network connection error",
         "suggestions": [
             "Check your internet connection",
             "Verify the target is reachable: ping {target}",
             "Check if a firewall is blocking the connection",
-            "Try using a proxy if configured"
+            "Try using a proxy if configured",
         ],
-        "help_cmd": None
+        "help_cmd": None,
     },
-
     ErrorType.AUTH_REQUIRED: {
         "message": "Authentication required",
         "suggestions": [
             "Log in with 'souleyez login'",
             "Create an account if you don't have one",
-            "Check that your session hasn't expired"
+            "Check that your session hasn't expired",
         ],
-        "help_cmd": "souleyez login"
+        "help_cmd": "souleyez login",
     },
-
     ErrorType.SESSION_EXPIRED: {
         "message": "Session expired",
         "suggestions": [
             "Log in again with 'souleyez login'",
             "Increase session timeout in config if needed",
-            "Check security.session_timeout_minutes in config"
+            "Check security.session_timeout_minutes in config",
         ],
-        "help_cmd": "souleyez login"
+        "help_cmd": "souleyez login",
     },
-
     ErrorType.METASPLOIT_ERROR: {
         "message": "Metasploit error",
         "suggestions": [
             "Check if Metasploit is installed: 'msfconsole -v'",
             "Initialize the MSF database: 'msfdb init'",
             "Run 'souleyez doctor --verbose' for MSF diagnostics",
-            "Try running msfconsole directly to see errors"
+            "Try running msfconsole directly to see errors",
         ],
-        "help_cmd": "souleyez doctor --verbose"
+        "help_cmd": "souleyez doctor --verbose",
     },
-
     ErrorType.ENCRYPTION_ERROR: {
         "message": "Encryption/decryption error",
         "suggestions": [
             "Make sure you're using the correct master password",
             "The credential vault may be locked - unlock it first",
-            "If you forgot your password, credentials cannot be recovered"
+            "If you forgot your password, credentials cannot be recovered",
         ],
-        "help_cmd": None
+        "help_cmd": None,
     },
-
     ErrorType.IMPORT_ERROR: {
         "message": "Failed to import required module",
         "suggestions": [
             "Run 'pip install souleyez' or 'pip install -e .' to install dependencies",
             "Check that all requirements are installed",
-            "Run 'souleyez doctor' to check for missing packages"
+            "Run 'souleyez doctor' to check for missing packages",
         ],
-        "help_cmd": "souleyez doctor"
+        "help_cmd": "souleyez doctor",
     },
-
     ErrorType.CUSTOM: {
         "message": "An error occurred",
         "suggestions": [],
-        "help_cmd": "souleyez doctor"
-    }
+        "help_cmd": "souleyez doctor",
+    },
 }
 
 
@@ -183,7 +173,7 @@ def error_msg(
     message: Optional[str] = None,
     suggestions: Optional[List[str]] = None,
     show_help: bool = True,
-    **context
+    **context,
 ) -> None:
     """
     Display a user-friendly error message with fix suggestions.
@@ -207,13 +197,13 @@ def error_msg(
 
     # Display error
     click.echo()
-    click.echo(click.style(f"  ✗ {final_message}", fg='red', bold=True))
+    click.echo(click.style(f"  ✗ {final_message}", fg="red", bold=True))
 
     # Get and display suggestions
     final_suggestions = suggestions or error_def["suggestions"]
     if final_suggestions:
         click.echo()
-        click.echo(click.style("  How to fix:", fg='yellow'))
+        click.echo(click.style("  How to fix:", fg="yellow"))
         for i, suggestion in enumerate(final_suggestions[:4], 1):
             # Substitute context variables in suggestion
             for key, value in context.items():
@@ -227,7 +217,10 @@ def error_msg(
         for key, value in context.items():
             help_cmd = help_cmd.replace(f"{{{key}}}", str(value))
         click.echo()
-        click.echo(click.style(f"  Quick fix: ", fg='cyan') + click.style(help_cmd, fg='white', bold=True))
+        click.echo(
+            click.style(f"  Quick fix: ", fg="cyan")
+            + click.style(help_cmd, fg="white", bold=True)
+        )
 
     click.echo()
 
@@ -241,7 +234,7 @@ def warn_msg(message: str, suggestions: Optional[List[str]] = None) -> None:
         suggestions: Optional list of suggestions
     """
     click.echo()
-    click.echo(click.style(f"  ⚠ {message}", fg='yellow', bold=True))
+    click.echo(click.style(f"  ⚠ {message}", fg="yellow", bold=True))
 
     if suggestions:
         click.echo()
@@ -260,11 +253,11 @@ def success_msg(message: str, next_steps: Optional[List[str]] = None) -> None:
         next_steps: Optional list of next steps
     """
     click.echo()
-    click.echo(click.style(f"  ✓ {message}", fg='green', bold=True))
+    click.echo(click.style(f"  ✓ {message}", fg="green", bold=True))
 
     if next_steps:
         click.echo()
-        click.echo(click.style("  Next steps:", fg='cyan'))
+        click.echo(click.style("  Next steps:", fg="cyan"))
         for i, step in enumerate(next_steps[:3], 1):
             click.echo(f"    {i}. {step}")
 
@@ -280,7 +273,7 @@ def info_msg(message: str, details: Optional[List[str]] = None) -> None:
         details: Optional list of details
     """
     click.echo()
-    click.echo(click.style(f"  ℹ {message}", fg='cyan'))
+    click.echo(click.style(f"  ℹ {message}", fg="cyan"))
 
     if details:
         for detail in details[:5]:
@@ -292,14 +285,21 @@ def info_msg(message: str, details: Optional[List[str]] = None) -> None:
 # Convenience functions for common errors
 def engagement_not_found(engagement_id: Optional[int] = None) -> None:
     """Show engagement not found error with fix suggestions."""
-    msg = f"Engagement ID {engagement_id} not found" if engagement_id else "Engagement not found"
+    msg = (
+        f"Engagement ID {engagement_id} not found"
+        if engagement_id
+        else "Engagement not found"
+    )
     error_msg(ErrorType.ENGAGEMENT_NOT_FOUND, message=msg)
 
 
 def tool_not_found(tool_name: str) -> None:
     """Show tool not found error with fix suggestions."""
-    error_msg(ErrorType.TOOL_NOT_FOUND, tool_name=tool_name,
-              message=f"Tool '{tool_name}' not installed or not found in PATH")
+    error_msg(
+        ErrorType.TOOL_NOT_FOUND,
+        tool_name=tool_name,
+        message=f"Tool '{tool_name}' not installed or not found in PATH",
+    )
 
 
 def no_engagement_selected() -> None: