colin/nwg-panel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
83ee9af03345e690cacab013df3aeaa1935ce08e
nwg-panel/nwg_panel/processes.py
piotr 83ee9af033 column widths
2023-02-16 16:02:42 +01:00

298 lines
9.0 KiB
Python
Raw Blame History

import os
import psutil
from i3ipc import Connection
import gi
gi.require_version('Gtk', '3.0')
from gi.repository import Gtk, Gdk, GLib
from nwg_panel.tools import get_config_dir, load_json, save_json, check_key, eprint
W_KILL = 5
W_PID = 7
W_PPID = 7
W_OWNER = 20
W_CPU = 6
W_MEM = 6
W_NAME = 20
W_WINDOW = 20
common_settings = {}
scrolled_window = None
grid = Gtk.Grid()
scroll = 0.0
max_num_items = 0
theme = Gtk.IconTheme.get_default()
def handle_keyboard(win, event):
if event.type == Gdk.EventType.KEY_RELEASE and event.keyval == Gdk.KEY_Escape:
win.destroy()
def terminate(btn, pid):
print("Terminating {}".format(pid))
try:
os.kill(pid, 15)
except Exception as e:
eprint(e)
def on_scroll(s_window, event):
adj = s_window.get_vadjustment()
global scroll
scroll = adj.get_value()
def on_list_changed(s_window, rect):
s_window.show_all()
adj = s_window.get_vadjustment()
adj.set_value(scroll)
print("scroll:", scroll, "upper:", adj.get_upper(), "current:", adj.get_value())
def list_processes(widget):
tree = Connection().get_tree()
processes = {}
user = os.getenv('USER')
for proc in psutil.process_iter(['pid', 'ppid', 'name', 'username', 'cpu_percent', 'memory_percent']):
if proc.info['username'] == os.getenv('USER') or not common_settings["processes-own-only"]:
processes[proc.info['pid']] = proc.info
if scrolled_window and scrolled_window.get_children():
viewport = scrolled_window.get_children()[0]
else:
viewport = None
global grid
if grid:
grid.destroy()
grid = Gtk.Grid.new()
grid.set_row_spacing(3)
grid.set_row_homogeneous(True)
grid.set_column_spacing(9)
if viewport:
viewport.add(grid)
else:
scrolled_window.add(grid)
idx = 1
for pid in processes:
cons = tree.find_by_pid(pid)
if not cons or not common_settings["processes-background-only"]:
lbl = Gtk.Label.new(str(pid))
lbl.set_width_chars(W_PID)
lbl.set_property("halign", Gtk.Align.END)
grid.attach(lbl, 1, idx, 1, 1)
lbl = Gtk.Label.new(str(processes[pid]["ppid"]))
lbl.set_width_chars(W_PPID)
lbl.set_property("halign", Gtk.Align.END)
grid.attach(lbl, 2, idx, 1, 1)
lbl = Gtk.Label.new(processes[pid]["username"])
lbl.set_width_chars(W_OWNER)
lbl.set_property("halign", Gtk.Align.END)
grid.attach(lbl, 3, idx, 1, 1)
lbl = Gtk.Label.new("{}%".format(str(processes[pid]["cpu_percent"])))
lbl.set_width_chars(W_CPU)
lbl.set_property("halign", Gtk.Align.END)
grid.attach(lbl, 4, idx, 1, 1)
lbl = Gtk.Label.new("{}%".format(str(round(processes[pid]["memory_percent"], 1))))
lbl.set_width_chars(W_MEM)
lbl.set_property("halign", Gtk.Align.END)
grid.attach(lbl, 5, idx, 1, 1)
win_name = ""
if cons:
if cons[0].app_id:
win_name = cons[0].app_id
elif cons[0].window_class:
win_name = cons[0].window_class
elif cons[0].name:
win_name = cons[0].name
elif cons[0].window_title:
win_name = cons[0].window_title
if win_name:
lbl = Gtk.Label.new(win_name)
lbl.set_width_chars(W_WINDOW)
lbl.set_property("halign", Gtk.Align.START)
grid.attach(lbl, 8, idx, 1, 1)
name = processes[pid]["name"]
if theme.lookup_icon(name, 16, Gtk.IconLookupFlags.FORCE_SYMBOLIC):
img = Gtk.Image.new_from_icon_name(name, Gtk.IconSize.MENU)
img.set_property("halign", Gtk.Align.END)
grid.attach(img, 6, idx, 1, 1)
# fallback icon name
elif win_name and theme.lookup_icon(win_name, 16, Gtk.IconLookupFlags.FORCE_SYMBOLIC):
img = Gtk.Image.new_from_icon_name(win_name, Gtk.IconSize.MENU)
img.set_property("halign", Gtk.Align.END)
grid.attach(img, 6, idx, 1, 1)
lbl = Gtk.Label.new(name)
lbl.set_width_chars(W_NAME)
lbl.set_property("halign", Gtk.Align.START)
grid.attach(lbl, 7, idx, 1, 1)
if processes[pid]["username"] == user:
btn = Gtk.Button.new_from_icon_name("gtk-close", Gtk.IconSize.MENU)
btn.connect("clicked", terminate, pid)
grid.attach(btn, 0, idx, 1, 1)
idx += 1
global max_num_items
if max_num_items < idx:
max_num_items = idx
if idx < max_num_items:
for i in range(idx, max_num_items):
lbl = Gtk.Label()
lbl.set_markup(" ")
grid.attach(lbl, 0, i, 1, 1)
grid.show_all()
scrolled_window.get_vadjustment().set_value(scroll)
return True
def on_background_cb(check_button):
common_settings["processes-background-only"] = check_button.get_active()
global max_num_items
max_num_items = 0
list_processes(None)
def on_own_cb(check_button):
common_settings["processes-own-only"] = check_button.get_active()
global max_num_items
max_num_items = 0
list_processes(None)
def main():
GLib.set_prgname('nwg-processes')
global common_settings
common_settings = load_json(os.path.join(get_config_dir(), "common-settings.json"))
defaults = {
"processes-background-only": False,
"processes-own-only": True
}
for key in defaults:
check_key(common_settings, key, defaults[key])
eprint("Common settings", common_settings)
win = Gtk.Window.new(Gtk.WindowType.TOPLEVEL)
win.connect('destroy', Gtk.main_quit)
win.connect("key-release-event", handle_keyboard)
box = Gtk.Box.new(Gtk.Orientation.VERTICAL, 6)
box.set_property("margin", 12)
win.add(box)
wrapper = Gtk.Box.new(Gtk.Orientation.HORIZONTAL, 0)
box.pack_start(wrapper, False, False, 0)
desc_box = Gtk.Box.new(Gtk.Orientation.HORIZONTAL, 0)
wrapper.pack_start(desc_box, False, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>Kill</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_KILL)
desc_box.pack_start(lbl, False, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>PID</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_PID)
desc_box.pack_start(lbl, False, True, 0)
lbl_ppid = Gtk.Label()
lbl_ppid.set_markup("<b>PPID</b>")
lbl_ppid.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_PPID)
desc_box.pack_start(lbl_ppid, False, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>Owner</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_OWNER)
desc_box.pack_start(lbl, False, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>CPU%</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_CPU)
desc_box.pack_start(lbl, True, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>Mem%</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_MEM)
desc_box.pack_start(lbl, True, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>Name</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_NAME)
desc_box.pack_start(lbl, True, True, 0)
lbl = Gtk.Label()
lbl.set_markup("<b>Window</b>")
lbl.set_property("halign", Gtk.Align.START)
lbl.set_width_chars(W_WINDOW)
desc_box.pack_start(lbl, True, True, 0)
global scrolled_window
scrolled_window = Gtk.ScrolledWindow.new(None, None)
scrolled_window.set_policy(Gtk.PolicyType.NEVER, Gtk.PolicyType.ALWAYS)
scrolled_window.set_propagate_natural_width(True)
scrolled_window.set_propagate_natural_height(True)
scrolled_window.connect("scroll-event", on_scroll)
box.pack_start(scrolled_window, True, True, 0)
hbox = Gtk.Box.new(Gtk.Orientation.HORIZONTAL, 12)
box.pack_start(hbox, False, False, 0)
cb = Gtk.CheckButton.new_with_label("Background only")
cb.set_tooltip_text("Processes that don't belong to the sway tree")
cb.set_active(common_settings["processes-background-only"])
cb.connect("toggled", on_background_cb)
hbox.pack_start(cb, False, False, 6)
cb = Gtk.CheckButton.new_with_label("{}'s only".format(os.getenv('USER')))
cb.set_tooltip_text("Processes that belong to the current $USER")
cb.set_active(common_settings["processes-own-only"])
cb.connect("toggled", on_own_cb)
hbox.pack_start(cb, False, False, 6)
btn = Gtk.Button.new_with_label("Close")
hbox.pack_end(btn, False, False, 0)
btn.connect("clicked", Gtk.main_quit)
btn = Gtk.Button.new_with_label("Refresh")
hbox.pack_end(btn, False, False, 0)
btn.connect("clicked", list_processes)
win.show_all()
list_processes(None)
GLib.timeout_add(2000, list_processes, None)
Gtk.main()
if __name__ == '__main__':
main()
Reference in New Issue View Git Blame Copy Permalink