Headless-Version.py

# BETA
#!/usr/bin/env python3
import time
import argparse
import os

from app import (
    find_hwmon,
    find_cooling_device,
    find_temp_sensor_path,
    find_cpu_freq_path,
    read_temp_c,
    read_cpu_freq_mhz,
    read_mem_used_total_mb,
    read_rpm,
    read_pwm_value,
    read_cooling_state,
    set_mode_auto,
    set_mode_percent,
    is_root,
)

MODES = {
    "auto": "auto",
    "25": 25,
    "50": 50,
    "100": 100,
}

def apply_mode(hw, cd, mode):
    if mode == "auto":
        return set_mode_auto(hw, cd)
    else:
        return set_mode_percent(hw, cd, mode)


def status_loop(hw, cd, sensor, freq_path, interval):
    while True:
        temp = read_temp_c(sensor)
        rpm = read_rpm(hw)
        pwm = read_pwm_value(hw)
        cd_state = read_cooling_state(cd)
        freq = read_cpu_freq_mhz(freq_path)
        used, total = read_mem_used_total_mb()

        print(
            f"Temp: {temp:.1f}°C | "
            f"RPM: {rpm or '--'} | "
            f"PWM: {pwm or '--'} | "
            f"Cooling: {cd_state or '--'} | "
            f"CPU: {freq:.0f} MHz | "
            f"RAM: {used/1024:.1f}/{total/1024:.1f} GB"
        )

        time.sleep(interval)


def main():
    parser = argparse.ArgumentParser(
        description="Raspberry Pi 5 Fan Controller (Console / Headless)"
    )
    parser.add_argument(
        "--mode",
        choices=MODES.keys(),
        default="auto",
        help="Fan mode: auto, 25, 50, 100",
    )
    parser.add_argument(
        "--interval",
        type=float,
        default=2.0,
        help="Status refresh interval in seconds",
    )
    parser.add_argument(
        "--once",
        action="store_true",
        help="Apply mode once and exit",
    )

    args = parser.parse_args()

    if not is_root():
        print("❌ Bitte mit sudo starten (GPIO / PWM benötigt Root).")
        exit(1)

    hw = find_hwmon()
    cd = find_cooling_device()
    sensor = find_temp_sensor_path()
    freq_path = find_cpu_freq_path()

    if not hw and not cd:
        print("❌ Keine hwmon- oder cooling_device-Pfade gefunden.")
        exit(1)

    mode = MODES[args.mode]
    ok = apply_mode(hw, cd, mode)

    print(f"✔ Modus gesetzt: {args.mode.upper()} ({'OK' if ok else 'teilweise'})")

    if args.once:
        return

    print("▶ Statusanzeige gestartet (CTRL+C zum Beenden)")
    status_loop(hw, cd, sensor, freq_path, args.interval)


if __name__ == "__main__":
    main()