process_manager.py

import psutil


def list_processes():
    """
    Listează toate procesele active: PID, nume, user, memorie (%).
    """
    print("\n=== Lista proceselor active ===")
    print(f"{'PID':>6}  {'USER':<15}  {'NAME':<30}  {'MEM%':>6}")
    print("-" * 65)

    for proc in psutil.process_iter(['pid', 'name', 'username', 'memory_percent']):
        try:
            info = proc.info
            pid = info.get("pid")
            name = info.get("name") or "?"
            user = info.get("username") or "?"
            mem = info.get("memory_percent") or 0.0

            print(f"{pid:>6}  {user:<15.15}  {name:<30.30}  {mem:6.2f}")
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            continue


def search_processes(query: str):
    """
    Caută procese după nume (case-insensitive).
    """
    print(f"\n=== Căutare procese care conțin: '{query}' ===")
    found = False

    for proc in psutil.process_iter(['pid', 'name', 'username', 'memory_percent']):
        try:
            info = proc.info
            name = info.get("name") or ""
            if query.lower() in name.lower():
                found = True
                pid = info.get("pid")
                user = info.get("username") or "?"
                mem = info.get("memory_percent") or 0.0
                print(f"PID={pid}  USER={user}  NAME={name}  MEM={mem:.2f}%")
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            continue

    if not found:
        print("⚠️  Niciun proces găsit cu acest nume.")


def show_top_memory(n: int = 5):
    """
    Arată top N procese care consumă cea mai multă memorie.
    """
    processes = []

    for proc in psutil.process_iter(['pid', 'name', 'username', 'memory_percent']):
        try:
            info = proc.info
            processes.append(info)
        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
            continue

    # sortăm descrescător după memorie
    processes.sort(key=lambda p: p.get("memory_percent") or 0.0, reverse=True)

    print(f"\n=== Top {n} procese după memorie ===")
    print(f"{'PID':>6}  {'USER':<15}  {'NAME':<30}  {'MEM%':>6}")
    print("-" * 65)

    for info in processes[:n]:
        pid = info.get("pid")
        name = info.get("name") or "?"
        user = info.get("username") or "?"
        mem = info.get("memory_percent") or 0.0
        print(f"{pid:>6}  {user:<15.15}  {name:<30.30}  {mem:6.2f}")


def kill_process(pid: int):
    """
    Încearcă să termine (terminate) un proces după PID.
    """
    try:
        proc = psutil.Process(pid)
        name = proc.name()
        proc.terminate()
        proc.wait(timeout=3)
        print(f"✅ Procesul cu PID={pid} ({name}) a fost terminat.")
    except psutil.NoSuchProcess:
        print("❌ Procesul nu există.")
    except psutil.AccessDenied:
        print("❌ Acces refuzat. Încearcă să rulezi cu privilegii mai mari (sudo).")
    except psutil.TimeoutExpired:
        print("⚠️ Procesul nu a răspuns la terminate(). Poți încerca kill (proc.kill()).")
    except Exception as e:
        print(f"❌ Eroare neașteptată: {e}")


def main_menu():
    while True:
        print("\n=== Process Manager CLI ===")
        print("1. Listează toate procesele")
        print("2. Caută proces după nume")
        print("3. Arată top 5 procese după memorie")
        print("4. Termină un proces după PID")
        print("5. Ieșire")

        choice = input("Alege o opțiune (1-5): ").strip()

        if choice == "1":
            list_processes()
        elif choice == "2":
            query = input("Introdu numele (sau parte din nume) procesului: ").strip()
            if query:
                search_processes(query)
        elif choice == "3":
            show_top_memory(5)
        elif choice == "4":
            pid_input = input("Introdu PID-ul procesului de terminat: ").strip()
            if pid_input.isdigit():
                kill_process(int(pid_input))
            else:
                print("⚠️  PID invalid.")
        elif choice == "5":
            print("👋 Ieșire din Process Manager CLI.")
            break
        else:
            print("⚠️ Opțiune invalidă. Încearcă din nou.")


if __name__ == "__main__":
    main_menu()