PySimpleGUI/DemoPrograms/Demo_Desktop_Widget_CPU_Top_Processes.py

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

"""
    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 FROM THREAD-', (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(font=('Helvetica', 20), text_color=sg.YELLOWS[0], key='-CPU PERCENT-')],
              [sg.Text(size=(35, 12), font=('Courier New', 12), key='-PROCESSES-')],     # size will determine how many processes shown
              [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_VER_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()
        # print(event, values)
        # --------- 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
        if event == 'Edit Me':
            sp = sg.execute_editor(__file__)
        elif event == 'Version':
            sg.popup_scrolled(__file__, sg.get_versions(), keep_on_top=True, location=window.current_location())
        elif event == '-CPU UPDATE FROM THREAD-':                   # 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:23}' for proc, cpu in top_sorted])
                # --------- Display timer and proceses in window --------
                window['-CPU PERCENT-'].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()