stop_bots.py

#!/usr/bin/env python3
"""
Stop all running bot instances safely.
"""
import os
import sys
import subprocess
import signal
from pathlib import Path

SIBLINGS_DIR = Path(__file__).parent
PID_DIR = SIBLINGS_DIR / ".pids"

SARAH_PID_FILE = PID_DIR / "sarah.pid"
ATLAS_PID_FILE = PID_DIR / "atlas.pid"
ARGUS_PID_FILE = PID_DIR / "argus.pid"

def kill_process(pid: int) -> bool:
    """Kill a process by PID."""
    try:
        if sys.platform == "win32":
            subprocess.run(["taskkill", "/F", "/PID", str(pid)],
                         capture_output=True, check=False)
        else:
            os.kill(pid, signal.SIGTERM)
        return True
    except Exception as e:
        print(f"Could not kill PID {pid}: {e}")
        return False

def stop_bot(pid_file: Path, name: str):
    """Stop a bot by reading its PID file."""
    if not pid_file.exists():
        print(f"{name}: No PID file found (not running?)")
        return

    try:
        pid = int(pid_file.read_text().strip())
        print(f"Stopping {name} (PID {pid})...")
        if kill_process(pid):
            print(f"{name} stopped.")
        pid_file.unlink(missing_ok=True)
    except (ValueError, FileNotFoundError) as e:
        print(f"{name}: Error reading PID file: {e}")

def kill_all_python_bots():
    """Kill any remaining Python bot processes."""
    if sys.platform == "win32":
        result = subprocess.run(
            ["wmic", "process", "where", "name='python.exe'", "get", "processid,commandline"],
            capture_output=True, text=True
        )
        killed = 0
        for line in result.stdout.split('\n'):
            if 'bot.main' in line:
                parts = line.strip().split()
                if parts:
                    try:
                        pid = int(parts[-1])
                        kill_process(pid)
                        killed += 1
                    except ValueError:
                        pass
        if killed:
            print(f"Killed {killed} orphan bot process(es).")

def main():
    print("Stopping Siblings bots...")
    print("-" * 30)

    stop_bot(SARAH_PID_FILE, "Sarah")
    stop_bot(ATLAS_PID_FILE, "Atlas")
    stop_bot(ARGUS_PID_FILE, "Argus")

    print("\nCleaning up any orphan processes...")
    kill_all_python_bots()

    print("\nAll bots stopped.")

if __name__ == "__main__":
    main()