110 lines
		
	
	
		
			No EOL
		
	
	
		
			5.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			110 lines
		
	
	
		
			No EOL
		
	
	
		
			5.4 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
| #!/usr/bin/env python
 | |
| import sys
 | |
| if sys.version_info[0] >= 3:
 | |
|     import PySimpleGUIQt 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 = 60
 | |
| TRANSPARENCY = .8       # how transparent the window looks. 0 = invisible, 1 = normal window
 | |
| NUM_COLS = 4
 | |
| POLL_FREQUENCY = 800    # 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=(0,0), border_width=0)
 | |
| 
 | |
|     # ----------------  Create Layout  ----------------
 | |
|     layout = [[ sg.Button('', image_data=red_x , button_color=('red', 'black'), key='Exit', tooltip='Closes window'),sg.Stretch(),
 | |
|                 sg.Text('     CPU Core Usage'), sg.Stretch()] ]
 | |
| 
 | |
|     # 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=(20, 15),
 | |
|                        return_keyboard_events=True,
 | |
|                        alpha_channel=TRANSPARENCY,
 | |
|                        use_default_focus=False,
 | |
|                        ).Layout(layout)
 | |
| 
 | |
|     # 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(interval=.2, 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 = b"R0lGODlhEAAQAPeQAIsAAI0AAI4AAI8AAJIAAJUAAJQCApkAAJoAAJ4AAJkJCaAAAKYAAKcAAKcCAKcDA6cGAKgAAKsAAKsCAKwAAK0AAK8AAK4CAK8DAqUJAKULAKwLALAAALEAALIAALMAALMDALQAALUAALYAALcEALoAALsAALsCALwAAL8AALkJAL4NAL8NAKoTAKwbAbEQALMVAL0QAL0RAKsREaodHbkQELMsALg2ALk3ALs+ALE2FbgpKbA1Nbc1Nb44N8AAAMIWAMsvAMUgDMcxAKVABb9NBbVJErFYEq1iMrtoMr5kP8BKAMFLAMxKANBBANFCANJFANFEB9JKAMFcANFZANZcANpfAMJUEMZVEc5hAM5pAMluBdRsANR8AM9YOrdERMpIQs1UVMR5WNt8X8VgYMdlZcxtYtx4YNF/btp9eraNf9qXXNCCZsyLeNSLd8SSecySf82kd9qqc9uBgdyBgd+EhN6JgtSIiNuJieGHhOGLg+GKhOKamty1ste4sNO+ueenp+inp+HHrebGrefKuOPTzejWzera1O7b1vLb2/bl4vTu7fbw7ffx7vnz8f///wAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAACH5BAEAAJAALAAAAAAQABAAAAjUACEJHEiwYEEABniQKfNFgQCDkATQwAMokEU+PQgUFDAjjR09e/LUmUNnh8aBCcCgUeRmzBkzie6EeQBAoAAMXuA8ciRGCaJHfXzUMCAQgYooWN48anTokR8dQk4sELggBhQrU9Q8evSHiJQgLCIIfMDCSZUjhbYuQkLFCRAMAiOQGGLE0CNBcZYmaRIDLqQFGF60eTRoSxc5jwjhACFWIAgMLtgUocJFy5orL0IQRHAiQgsbRZYswbEhBIiCCH6EiJAhAwQMKU5DjHCi9gnZEHMTDAgAOw=="
 | |
| 
 | |
|     main()
 | |
|     sys.exit(69) |