justin/Python
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added support for showing line numbers.

Browse Source
This commit is contained in:
JustinOros
2025-07-31 16:16:03 -07:00
parent e12e3ed614
commit ddc3a514a3
1 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
color-top.py
+19 -13
View File
@@ -9,7 +9,6 @@ import time
import psutil import psutil
import sys import sys
import select import select
import socket
import argparse import argparse
import tty import tty
import termios import termios
@@ -48,9 +47,9 @@ def get_top_processes(limit):
return processes[:limit] return processes[:limit]
def colorize(cpu, name, user, use_color): def colorize(cpu, name, user, use_color):
limit = 35 name_limit = 35
if len(name) > limit: if len(name) > name_limit:
display_name = name[:limit - 3] + "..." display_name = name[:name_limit - 3] + "..."
else: else:
display_name = name display_name = name
@@ -65,20 +64,25 @@ def colorize(cpu, name, user, use_color):
cpu_str = f"{cpu:6.2f}%" cpu_str = f"{cpu:6.2f}%"
return f"{color}{cpu_str} {display_name:<35} {user[:15]:<15}{RESET_TO_GREEN}" return f"{color}{cpu_str} {display_name:<35} {user[:15]:<15}{RESET_TO_GREEN}"
def print_processes(limit, show_quit_hint=True, use_color=True): def print_processes(limit, show_quit_hint=True, use_color=True, show_line_numbers=False):
clear_screen() clear_screen()
top_processes = get_top_processes(limit) top_processes = get_top_processes(limit)
GREEN = "\033[32m" if use_color else "" GREEN = "\033[32m" if use_color else ""
if use_color: if use_color:
print(GREEN, end='') print(GREEN, end='')
print(" CPU Process User") line_number_width = len(str(limit))
print("----------------------------------------------------------") line_prefix = f"{'Ln':<{line_number_width}} " if show_line_numbers else ""
cpu_col = " CPU"
for (cpu, name, user) in top_processes: name_col = " Process"
print(colorize(cpu, name, user, use_color)) user_col = " User"
header = f"{line_prefix}{cpu_col:<6} {name_col:<35} {user_col:<15}"
print(header)
print("-" * len(header))
for i, (cpu, name, user) in enumerate(top_processes, start=1):
prefix = f"{i:0{line_number_width}}: " if show_line_numbers else ""
print(f"{prefix}{colorize(cpu, name, user, use_color)}")
if show_quit_hint: if show_quit_hint:
print("\nPress 'q' to quit.") print("\nPress 'q' to quit.")
@@ -99,11 +103,13 @@ def main():
parser.add_argument("-n", "--number", type=int, default=10, help="Number of processes to monitor (default is 10).") parser.add_argument("-n", "--number", type=int, default=10, help="Number of processes to monitor (default is 10).")
parser.add_argument("-i", "--interval", type=int, default=1, help="Refresh interval in seconds (default is 1).") parser.add_argument("-i", "--interval", type=int, default=1, help="Refresh interval in seconds (default is 1).")
parser.add_argument("--no-color", action="store_true", help="Disable colored output.") parser.add_argument("--no-color", action="store_true", help="Disable colored output.")
parser.add_argument("-l", "--line", action="store_true", help="Show line numbers.")
args = parser.parse_args() args = parser.parse_args()
limit = args.number limit = args.number
interval = max(1, args.interval) interval = max(1, args.interval)
use_color = not args.no_color and supports_color() use_color = not args.no_color and supports_color()
show_lines = args.line
refresh_count = 0 refresh_count = 0
start_time = time.time() start_time = time.time()
@@ -113,7 +119,7 @@ def main():
refresh_count += 1 refresh_count += 1
elapsed = time.time() - start_time elapsed = time.time() - start_time
show_quit = refresh_count < 3 and elapsed < 3 show_quit = refresh_count < 3 and elapsed < 3
print_processes(limit, show_quit_hint=show_quit, use_color=use_color) print_processes(limit, show_quit_hint=show_quit, use_color=use_color, show_line_numbers=show_lines)
key = wait_for_keypress(timeout=interval) key = wait_for_keypress(timeout=interval)
if key and key.lower() == 'q': if key and key.lower() == 'q':
print("Exiting...") print("Exiting...")