hwmon/tui.py at main · shalonjovan/hwmon · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
from textual.app import App, ComposeResult
from textual.widgets import Static
from textual.containers import Horizontal, Vertical
from pathlib import Path

from state import SystemState

ASSETS = Path(__file__).parent / "assets"


def load_ascii(name: str) -> str:
    return (ASSETS / name).read_text()


def bar(value, max_value, width, char="█"):
    if max_value <= 0 or width <= 0:
        return ""
    ratio = min(max(value / max_value, 0), 1)
    filled = int(ratio * width)
    return char * filled + " " * (width - filled)


class HwmonTUI(App):
    CSS_PATH = "tui.tcss"

    def on_mount(self):
        self.state = SystemState()
        self.set_interval(1, self.refresh_state)

    def on_shutdown(self):
        self.state.cleanup()

    def compose(self) -> ComposeResult:
        pc_ascii = load_ascii("linux.txt")

        self.pc_ascii = Static(pc_ascii, id="pc_ascii")
        self.system_info = Static("", id="system_info")
        self.fans = Static("", id="fans")
        self.cpu_mem = Static("", id="cpu_mem")

        left = Vertical(self.pc_ascii, self.system_info, id="left")
        right = Vertical(self.fans, self.cpu_mem, id="right")

        yield Horizontal(left, right, id="main")

    def refresh_state(self):
        state = self.state.snapshot()

        container_width = self.size.width
        right_width = max(int(container_width * 0.6) - 30, 15)

        # ================= LEFT =================

        left_text = f"[b]{state['system']['product_name']}[/b]\n"

        for idx, battery in enumerate(state["batteries"], 1):
            left_text += (
                f"\nBattery {idx}\n"
                f"  {battery.get('voltage', 0)} V\n"
                f"  {battery.get('power', 0)} W\n"
            )

        if state["nvme"]:
            left_text += "\nNVMe\n"
            for label, temp in state["nvme"].items():
                left_text += f"  {label}: {temp:.1f}°C\n"

        self.system_info.update(left_text.strip())

        # ================= RIGHT =================

        cpu = state["cpu"]
        mem = state["memory"]
        net = state["network"]

        right_text = (
            f"CPU\n"
            f"  Temp: {cpu.get('temp', 0) or 0:.1f}°C\n"
            f"  Usage: {cpu['usage_percent']}%\n"
            f"  [{bar(cpu['usage_percent'], 100, right_width)}]\n\n"
            f"Memory\n"
            f"  Used: {mem['used_mb']} MB\n"
            f"  Free: {mem['available_mb']} MB\n"
            f"  [{bar(mem['percent_used'], 100, right_width)}]\n\n"
            f"Network\n"
            f"  Sent: {net['bytes_sent'] / (1024**2):.2f} MB\n"
            f"  Received: {net['bytes_recv'] / (1024**2):.2f} MB\n"
            f"  ↑ {net['upload_speed'] / 1024:.2f} KB/s\n"
            f"  ↓ {net['download_speed'] / 1024:.2f} KB/s\n"
        )

        if state["gpu"]:
            right_text += "\nGPU\n"
            for name, info in state["gpu"].items():
                right_text += (
                    f"  {name}\n"
                    f"    Usage: {info['usage_percent']}%\n"
                    f"    Temp: {info['temp']}°C\n"
                )

        self.cpu_mem.update(right_text.strip())

        # ================= FANS =================

        fan_ascii = load_ascii("fan.txt").splitlines()
        ascii_width = max(len(line) for line in fan_ascii) if fan_ascii else 0

        fan_lines = []
        for name, rpm in state["fans"].items():
            fan_lines.append(f"{name}: {rpm} RPM")
            fan_lines.append(f"[{bar(rpm, 7600, max(right_width - 15, 5))}]")
            fan_lines.append("")

        max_lines = max(len(fan_ascii), len(fan_lines))
        combined_lines = []

        for i in range(max_lines):
            left_part = fan_ascii[i] if i < len(fan_ascii) else ""
            right_part = fan_lines[i] if i < len(fan_lines) else ""
            combined_lines.append(left_part.ljust(ascii_width + 4) + right_part)

        self.fans.update("\n".join(combined_lines))


def main():
    HwmonTUI().run()


if __name__ == "__main__":
    main()