Function run_script_in_venv

StegoScan.py:126–155

Restarts the script inside the virtual environment if not already inside it.

()

def run_script_in_venv():
    """Restarts the script inside the virtual environment if not already inside it."""
    python_exec = get_venv_python()

    # Avoid infinite loops by checking if we're already inside the venv
    if sys.prefix == os.path.abspath(VENV_DIR):  # Already inside the venv
        prGreen("Script is already running inside the virtual environment.")
        return  # Do nothing, exit the function

    print(f"Running script inside virtual environment: {VENV_DIR}")

    # Ensure that the python executable exists
    if not os.path.isfile(python_exec):
        prRed(f"Error: Python executable not found at {python_exec}")
        sys.exit(1)

    # Build the command to restart the script inside the virtual environment
    command = [python_exec] + sys.argv

    # Explicitly set the working directory to where the script is
    script_dir = os.path.dirname(os.path.abspath(__file__))

    try:
        print(f"Restarting the script with command: {command}")
        subprocess.check_call(command, cwd=script_dir, stderr=subprocess.STDOUT)
    except subprocess.CalledProcessError as e:
        prRed(f"Error during restart: {e}")
        sys.exit(1)

    sys.exit()  # Ensure the script exits after launching the new one


# Setup environment and dependencies