PySimpleGUI/DemoPrograms/Demo_Desktop_Widget_CPU_Top_Processes.py

#!/usr/bin/env python
import PySimpleGUI as sg
import psutil
import time
import operator
import sys

"""
    PSUTIL "Top" CPU Processes - Desktop Widget
    
    Creates a floating CPU utilization window running something similar to a "top" command.

    Use the spinner to adjust the number of seconds between readings of the CPU utilizaiton
    Rather than calling the threading module this program uses the PySimpleGUI perform_long_operation method.
        The result is similar.  The function is run as a thread... the call is simply wrapped.
            
    Copyright 2022 PySimpleGUI
"""

# global used to communicate with thread.
g_interval = 1      # how often to poll for CPU usage

def CPU_thread(window:sg.Window):

    while True:
        cpu_percent = psutil.cpu_percent(interval=g_interval)
        procs = psutil.process_iter()
        window.write_event_value('-CPU UPDATE-', (cpu_percent, procs))


def main():
    global g_interval

    location =  sg.user_settings_get_entry('-location-', (None, None))


    # ----------------  Create Form  ----------------
    sg.theme('Black')
    layout = [[sg.Text('', size=(8, 1), font=('Helvetica', 20),
              text_color=sg.YELLOWS[0], justification='center', key='text'), sg.Push(), sg.Text('❎', enable_events=True, key='Exit')],
              [sg.Text('', size=(30, 8), font=('Courier New', 12),
              text_color='white', justification='left', key='processes')],
              [sg.Text('Update every '), sg.Spin([x+1 for x in range(10)], 3, key='spin'), sg.T('seconds       ')]]

    window = sg.Window('Top CPU Processes', layout, no_titlebar=True, keep_on_top=True,location=location, use_default_focus=False, alpha_channel=.8, grab_anywhere=True, right_click_menu=sg.MENU_RIGHT_CLICK_EDITME_EXIT, enable_close_attempted_event=True)

    # start cpu measurement thread
    # using the PySimpleGUI call to start and manage the thread
    window.perform_long_operation(lambda: CPU_thread(window), '-THREAD FINISHED-')
    g_interval = 1
    # ----------------  main loop  ----------------
    while True:
        # --------- Read and update window --------
        event, values = window.read()
        # --------- Do Button Operations --------
        if event in (sg.WIN_CLOSE_ATTEMPTED_EVENT, 'Exit'):
            sg.user_settings_set_entry('-location-', window.current_location())     # save window location before exiting
            break
        elif event == 'Edit Me':
            sg.execute_editor(__file__)
        elif event == '-CPU UPDATE-':                   # indicates data from the thread has arrived
            cpu_percent, procs = values[event]          # the thread sends a tuple
            if procs:
                # --------- Create dictionary of top % CPU processes.  Format is name:cpu_percent --------
                top = {proc.name(): proc.cpu_percent() for proc in procs}

                top_sorted = sorted(top.items(), key=operator.itemgetter(1), reverse=True)  # reverse sort to get highest CPU usage on top
                if top_sorted:
                    top_sorted.pop(0)           # remove the idle process
                display_string =  '\n'.join([f'{cpu/10:2.2f} {proc}' for proc, cpu in top_sorted])
                # --------- Display timer and proceses in window --------
                window['text'].update(f'CPU {cpu_percent}')
                window['processes'].update(display_string)
        # get the timeout from the spinner
        g_interval = int(values['spin'])

    window.close()

if __name__ == '__main__':
    main()