forked from KlipperScreen/KlipperScreen
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsystem.py
170 lines (151 loc) · 6.58 KB
/
system.py
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
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
import logging
import gi
gi.require_version("Gtk", "3.0")
from gi.repository import Gtk
from ks_includes.screen_panel import ScreenPanel
class Panel(ScreenPanel):
def __init__(self, screen, title):
title = title or _("System")
super().__init__(screen, title)
self.current_row = 0
self.mem_multiplier = None
self.scales = {}
self.labels = {}
self.grid = Gtk.Grid(column_spacing=10, row_spacing=5)
self.sysinfo = screen.printer.system_info
if not self.sysinfo:
logging.debug("Asking for info")
self.sysinfo = screen.apiclient.send_request("machine/system_info")
if 'system_info' in self.sysinfo:
screen.printer.system_info = self.sysinfo['system_info']
self.sysinfo = self.sysinfo['system_info']
logging.debug(self.sysinfo)
if self.sysinfo:
self.content.add(self.create_layout())
else:
self.content.add(Gtk.Label(label=_("No info available"), vexpand=True))
def back(self):
if not self.sysinfo:
self._screen.panels_reinit.append("system")
return False
def create_layout(self):
self.cpu_count = int(self.sysinfo["cpu_info"]["cpu_count"])
self.labels["cpu_usage"] = Gtk.Label(label="", xalign=0)
self.grid.attach(self.labels["cpu_usage"], 0, self.current_row, 1, 1)
self.scales["cpu_usage"] = Gtk.ProgressBar(
hexpand=True, show_text=False, fraction=0
)
self.grid.attach(self.scales["cpu_usage"], 1, self.current_row, 1, 1)
self.current_row += 1
for i in range(self.cpu_count):
self.labels[f"cpu_usage_{i}"] = Gtk.Label(label="", xalign=0)
self.grid.attach(self.labels[f"cpu_usage_{i}"], 0, self.current_row, 1, 1)
self.scales[f"cpu_usage_{i}"] = Gtk.ProgressBar(
hexpand=True, show_text=False, fraction=0
)
self.grid.attach(self.scales[f"cpu_usage_{i}"], 1, self.current_row, 1, 1)
self.current_row += 1
self.labels["memory_usage"] = Gtk.Label(label="", xalign=0)
self.grid.attach(self.labels["memory_usage"], 0, self.current_row, 1, 1)
self.scales["memory_usage"] = Gtk.ProgressBar(
hexpand=True, show_text=False, fraction=0
)
self.grid.attach(self.scales["memory_usage"], 1, self.current_row, 1, 1)
self.current_row += 1
self.grid.attach(Gtk.Separator(), 0, self.current_row, 2, 1)
self.current_row += 1
self.populate_info()
scroll = self._gtk.ScrolledWindow()
scroll.add(self.grid)
return scroll
def set_mem_multiplier(self, data):
memory_units = data.get("memory_units", "kB").lower()
units_mapping = {
"kb": 1024,
"mb": 1024**2,
"gb": 1024**3,
"tb": 1024**4,
"pb": 1024**5,
}
self.mem_multiplier = units_mapping.get(memory_units, 1)
def add_label_to_grid(self, text, column, bold=False):
if bold:
text = f"<b>{text}</b>"
label = Gtk.Label(label=text, use_markup=True, xalign=0, wrap=True)
self.grid.attach(label, column, self.current_row, 1, 1)
self.current_row += 1
def populate_info(self):
for category, data in self.sysinfo.items():
if category == "python":
self.add_label_to_grid(self.prettify(category), 0, bold=True)
self.current_row -= 1
self.add_label_to_grid(
f'Version: {data["version_string"].split(" ")[0]}', 1
)
continue
if (
category
in (
"virtualization",
"provider",
"available_services",
"service_state",
"instance_ids",
)
or not self.sysinfo[category]
):
continue
self.add_label_to_grid(self.prettify(category), 0, bold=True)
if isinstance(data, dict):
for key, value in data.items():
if key in ("version_parts", "memory_units") or not value:
continue
if key == "total_memory":
if not self.mem_multiplier:
self.set_mem_multiplier(data)
value = self.format_size(int(value) * self.mem_multiplier)
if isinstance(value, dict):
self.add_label_to_grid(self.prettify(key), 0)
self.current_row -= 1
for sub_key, sub_value in value.items():
if not sub_value:
continue
elif (
isinstance(sub_value, list)
and sub_key == "ip_addresses"
):
for _ip in sub_value:
self.add_label_to_grid(
f"{self.prettify(sub_key)}: {_ip['address']}", 1
)
continue
self.add_label_to_grid(
f"{self.prettify(sub_key)}: {sub_value}", 1
)
else:
self.add_label_to_grid(f"{self.prettify(key)}: {value}", 1)
def process_update(self, action, data):
if not self.sysinfo:
return
if action == "notify_proc_stat_update":
self.labels["cpu_usage"].set_label(
f'CPU: {data["system_cpu_usage"]["cpu"]:.0f}%'
)
self.scales["cpu_usage"].set_fraction(
float(data["system_cpu_usage"]["cpu"]) / 100
)
for i in range(self.cpu_count):
self.labels[f"cpu_usage_{i}"].set_label(
f'CPU {i}: {data["system_cpu_usage"][f"cpu{i}"]:.0f}%'
)
self.scales[f"cpu_usage_{i}"].set_fraction(
float(data["system_cpu_usage"][f"cpu{i}"]) / 100
)
self.labels["memory_usage"].set_label(
_("Memory")
+ f': {(data["system_memory"]["used"] / data["system_memory"]["total"]) * 100:.0f}%'
)
self.scales["memory_usage"].set_fraction(
float(data["system_memory"]["used"])
/ float(data["system_memory"]["total"])
)