PySimpleGUI/DemoPrograms/Demo_Desktop_Widget_CPU_Squ...

64 lines
2.5 KiB
Python
Raw Normal View History

import PySimpleGUI as sg
import psutil
import sys
"""
Another simple Desktop Widget using PySimpleGUI
This time a CPU Usage indicator. The Widget is square.
The bottom section will be shaded to
represent the total amount CPU currently in use.
Uses the theme's button color for colors.
Copyright 2020 PySimpleGUI.org
"""
ALPHA = 0.5
THEME = 'Dark purple 6'
GSIZE = (160, 160)
UPDATE_FREQUENCY_MILLISECONDS = 2 * 1000
def main(location):
graph = sg.Graph(GSIZE, (0, 0), GSIZE, key='-GRAPH-')
layout = [[graph]]
window = sg.Window('CPU Usage Widget Square', layout, location=location, no_titlebar=True, grab_anywhere=True, margins=(0, 0), element_padding=(0, 0), alpha_channel=ALPHA, finalize=True, right_click_menu=sg.MENU_RIGHT_CLICK_EDITME_EXIT, enable_close_attempted_event=True)
text_id2 = graph.draw_text(f'CPU', (GSIZE[0] // 2, GSIZE[1] // 4), font='Any 20', text_location=sg.TEXT_LOCATION_CENTER, color=sg.theme_button_color()[0])
while True: # Event Loop
# ----------- update the graphics and text in the window ------------
cpu_percent = psutil.cpu_percent(interval=1)
# Draw the filled rectangle
rect_height = int(GSIZE[1] * float(cpu_percent) / 100)
rect_id = graph.draw_rectangle((0, rect_height), (GSIZE[0], 0), fill_color=sg.theme_button_color()[1], line_width=0)
# Draw the % used text and the close "X" on bottom
text_id1 = graph.draw_text(f'{int(cpu_percent)}%', (GSIZE[0] // 2, GSIZE[1] // 2), font='Any 40', text_location=sg.TEXT_LOCATION_CENTER, color=sg.theme_button_color()[0])
# put the bar behind everything else
graph.send_figure_to_back(rect_id)
# update the window, wait for a while, then check for exit
event, values = window.read(timeout=UPDATE_FREQUENCY_MILLISECONDS)
if event in (sg.WIN_CLOSE_ATTEMPTED_EVENT, 'Exit'):
sg.user_settings_set_entry('-location-', window.current_location()) # The line of code to save the position before exiting
break
if event == 'Edit Me':
sg.execute_editor(__file__)
# erase figures so they can be redrawn
graph.delete_figure(rect_id)
graph.delete_figure(text_id1)
window.close()
if __name__ == '__main__':
sg.theme(THEME)
if len(sys.argv) > 1:
location = sys.argv[1].split(',')
location = (int(location[0]), int(location[1]))
else:
location = sg.user_settings_get_entry('-location-', (None, None))
main(location)