PySimpleGUI/Demo_psutil_Kill_Processes.py

#!/usr/bin/env python
import sys
import sys
if sys.version_info[0] >= 3:
    import PySimpleGUI as sg
else:
    import PySimpleGUI27 as sg
import os
import signal
import psutil
import operator


"""
    Utility to show running processes, CPU usage and provides way to kill processes.
    Based on psutil package that is easily installed using pip
"""

def kill_proc_tree(pid, sig=signal.SIGTERM, include_parent=True,
                   timeout=None, on_terminate=None):
    """Kill a process tree (including grandchildren) with signal
    "sig" and return a (gone, still_alive) tuple.
    "on_terminate", if specified, is a callabck function which is
    called as soon as a child terminates.
    """
    if pid == os.getpid():
        raise RuntimeError("I refuse to kill myself")
    parent = psutil.Process(pid)
    children = parent.children(recursive=True)
    if include_parent:
        children.append(parent)
    for p in children:
        p.send_signal(sig)
    gone, alive = psutil.wait_procs(children, timeout=timeout,
                                    callback=on_terminate)
    return (gone, alive)


def main():

    # ----------------  Create Form  ----------------
    sg.ChangeLookAndFeel('Topanga')
    layout = [[sg.Text('Process Killer - Choose one or more processes',
                       size=(45,1), font=('Helvetica', 15))],
              [sg.Listbox(values=[' '], size=(50, 30), select_mode=sg.SELECT_MODE_EXTENDED,  font=('Courier', 12), key='_processes_')],
              [sg.Text('Click refresh once or twice.. once for list, second to get CPU usage')],
              [sg.T('Filter by typing name', font='ANY 14'), sg.In(size=(15,1), font='any 14', key='_filter_')],
              [sg.RButton('Refresh'),
               sg.RButton('Kill', button_color=('white','red'), bind_return_key=True),
               sg.Exit(button_color=('white', 'sea green'))]]

    window = sg.Window('Process Killer',
                       keep_on_top=True,
                       auto_size_buttons=False,
                       default_button_element_size=(9,1),
                       return_keyboard_events=True,
                       grab_anywhere=False).Layout(layout)

    proc_list = []
    # ----------------  main loop  ----------------
    while (True):
        # --------- Read and update window --------
        button, values = window.Read()

        # --------- Do Button Operations --------
        if values is None or button == 'Exit':
            break

        if button == 'Refresh':
            psutil.cpu_percent(interval=1)
            procs = psutil.process_iter()
            top = {proc.name(): (proc.cpu_percent(), proc.pid) for proc in procs}

            # cpu_percent = psutil.cpu_percent(interval=interval)       # if don't wan to use a task

            # --------- Create list of top % CPU porocesses --------

            top_sorted = sorted(top.items(), key=operator.itemgetter(0), reverse=False)
            if top_sorted:
                top_sorted.pop(0)
            proc_list = []
            for proc, pair in top_sorted:
                cpu, pid = pair
                proc_list.append('{:5d} {:5.2f}  {}\n'.format(pid, cpu/10, proc))

            window.FindElement('_processes_').Update(proc_list)
        elif button == 'Kill':
            processes_to_kill = values['_processes_']
            for proc in processes_to_kill:
                pid = int(proc[0:5])
                if sg.PopupYesNo('About to kill {} {}'.format(pid, proc[13:]), keep_on_top=True) == 'Yes':
                    kill_proc_tree(pid=pid)
        else:
            new_output = []
            for line in proc_list:
                if values['_filter_'] in line.lower():
                    new_output.append(line)
            window.FindElement('_processes_').Update(new_output)


if __name__ == "__main__":
    main()