110 lines
5.3 KiB
Python
110 lines
5.3 KiB
Python
#!/usr/bin/env python
|
|
import sys
|
|
if sys.version_info[0] >= 3:
|
|
import PySimpleGUI as sg
|
|
else:
|
|
import PySimpleGUI27 as sg
|
|
import psutil
|
|
|
|
"""
|
|
Desktop floating widget - CPU Cores
|
|
Uses psutil to display:
|
|
CPU usage on each individual core
|
|
Information is updated once a second and is shown as an area graph that scrolls
|
|
"""
|
|
|
|
GRAPH_WIDTH = 120 # each individual graph size in pixels
|
|
GRAPH_HEIGHT = 40
|
|
TRANSPARENCY = .8 # how transparent the window looks. 0 = invisible, 1 = normal window
|
|
NUM_COLS = 4
|
|
POLL_FREQUENCY = 500 # how often to update graphs in milliseconds
|
|
|
|
colors = ('#23a0a0', '#56d856', '#be45be', '#5681d8', '#d34545', '#BE7C29')
|
|
|
|
# DashGraph does the drawing of each graph
|
|
class DashGraph(object):
|
|
def __init__(self, graph_elem, text_elem, starting_count, color):
|
|
self.graph_current_item = 0
|
|
self.graph_elem = graph_elem
|
|
self.text_elem = text_elem
|
|
self.prev_value = starting_count
|
|
self.max_sent = 1
|
|
self.color = color
|
|
|
|
def graph_percentage_abs(self, value):
|
|
self.graph_elem.DrawLine((self.graph_current_item, 0), (self.graph_current_item, value), color=self.color)
|
|
if self.graph_current_item >= GRAPH_WIDTH:
|
|
self.graph_elem.Move(-1,0)
|
|
else:
|
|
self.graph_current_item += 1
|
|
|
|
def text_display(self, text):
|
|
self.text_elem.Update(text)
|
|
|
|
def main():
|
|
# A couple of "Uber Elements" that combine several elements and enable bulk edits
|
|
def Txt(text, **kwargs):
|
|
return(sg.Text(text, font=('Helvetica 8'), **kwargs))
|
|
|
|
def GraphColumn(name, key):
|
|
col = sg.Column([[Txt(name, key=key+'_TXT_'), ],
|
|
[sg.Graph((GRAPH_WIDTH, GRAPH_HEIGHT), (0, 0), (GRAPH_WIDTH, 100), background_color='black',
|
|
key=key+'_GRAPH_')]], pad=(2, 2))
|
|
return col
|
|
|
|
|
|
num_cores = len(psutil.cpu_percent(percpu=True)) # get the number of cores in the CPU
|
|
|
|
sg.ChangeLookAndFeel('Black')
|
|
sg.SetOptions(element_padding=(0,0), margins=(1,1), border_width=0)
|
|
|
|
# ---------------- Create Layout ----------------
|
|
layout = [[ sg.Button('', image_data=red_x, button_color=('black', 'black'), key='Exit', tooltip='Closes window'),
|
|
sg.Text(' CPU Core Usage')] ]
|
|
|
|
# add on the graphs
|
|
for rows in range(num_cores//NUM_COLS+1):
|
|
row = []
|
|
for cols in range(min(num_cores-rows*NUM_COLS, NUM_COLS)):
|
|
row.append(GraphColumn('CPU '+str(rows*NUM_COLS+cols), '_CPU_'+str(rows*NUM_COLS+cols)))
|
|
layout.append(row)
|
|
|
|
# ---------------- Create Window ----------------
|
|
window = sg.Window('PSG System Dashboard',
|
|
keep_on_top=True,
|
|
auto_size_buttons=False,
|
|
grab_anywhere=True,
|
|
no_titlebar=True,
|
|
default_button_element_size=(12, 1),
|
|
return_keyboard_events=True,
|
|
alpha_channel=TRANSPARENCY,
|
|
use_default_focus=False,
|
|
).Layout(layout).Finalize()
|
|
|
|
# setup graphs & initial values
|
|
graphs = []
|
|
for i in range(num_cores):
|
|
graphs.append(DashGraph(window.FindElement('_CPU_'+str(i)+'_GRAPH_'),
|
|
window.FindElement('_CPU_'+str(i) + '_TXT_'),
|
|
0, colors[i%6]))
|
|
|
|
# ---------------- main loop ----------------
|
|
while (True):
|
|
# --------- Read and update window once every Polling Frequency --------
|
|
event, values = window.Read(timeout=POLL_FREQUENCY)
|
|
if event in (None, 'Exit'): # Be nice and give an exit
|
|
break
|
|
# read CPU for each core
|
|
stats = psutil.cpu_percent(percpu=True)
|
|
# Update each graph
|
|
for i in range(num_cores):
|
|
graphs[i].graph_percentage_abs(stats[i])
|
|
graphs[i].text_display('{} CPU {:2.0f}'.format(i, stats[i]))
|
|
window.Close()
|
|
|
|
if __name__ == "__main__":
|
|
# the clever Red X graphic
|
|
red_x = "R0lGODlhEAAQAPeQAIsAAI0AAI4AAI8AAJIAAJUAAJQCApkAAJoAAJ4AAJkJCaAAAKYAAKcAAKcCAKcDA6cGAKgAAKsAAKsCAKwAAK0AAK8AAK4CAK8DAqUJAKULAKwLALAAALEAALIAALMAALMDALQAALUAALYAALcEALoAALsAALsCALwAAL8AALkJAL4NAL8NAKoTAKwbAbEQALMVAL0QAL0RAKsREaodHbkQELMsALg2ALk3ALs+ALE2FbgpKbA1Nbc1Nb44N8AAAMIWAMsvAMUgDMcxAKVABb9NBbVJErFYEq1iMrtoMr5kP8BKAMFLAMxKANBBANFCANJFANFEB9JKAMFcANFZANZcANpfAMJUEMZVEc5hAM5pAMluBdRsANR8AM9YOrdERMpIQs1UVMR5WNt8X8VgYMdlZcxtYtx4YNF/btp9eraNf9qXXNCCZsyLeNSLd8SSecySf82kd9qqc9uBgdyBgd+EhN6JgtSIiNuJieGHhOGLg+GKhOKamty1ste4sNO+ueenp+inp+HHrebGrefKuOPTzejWzera1O7b1vLb2/bl4vTu7fbw7ffx7vnz8f///wAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACH5BAEAAJAALAAAAAAQABAAAAjUACEJHEiwYEEABniQKfNFgQCDkATQwAMokEU+PQgUFDAjjR09e/LUmUNnh8aBCcCgUeRmzBkzie6EeQBAoAAMXuA8ciRGCaJHfXzUMCAQgYooWN48anTokR8dQk4sELggBhQrU9Q8evSHiJQgLCIIfMDCSZUjhbYuQkLFCRAMAiOQGGLE0CNBcZYmaRIDLqQFGF60eTRoSxc5jwjhACFWIAgMLtgUocJFy5orL0IQRHAiQgsbRZYswbEhBIiCCH6EiJAhAwQMKU5DjHCi9gnZEHMTDAgAOw=="
|
|
|
|
main()
|
|
sys.exit(69) |