Added color highlighted rows vs text (just looks cooler).

This commit is contained in:
justin committed 2025-07-31 16:57:12 -07:00
1 parent 0a43d7d6e7
commit c81bb3553a
1 file changed
+31 -18
+31 -18
View File
@@ -1,4 +1,3 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
# Description: Displays top system processes with color. # Description: Displays top system processes with color.
# Usage: python3 color-top.py # Usage: python3 color-top.py
@@ -47,45 +46,59 @@ def get_top_processes(limit):
processes.sort(reverse=True, key=lambda p: p[0]) processes.sort(reverse=True, key=lambda p: p[0])
return processes[:limit] return processes[:limit]
def colorize(cpu, name, user, use_color): def colorize(cpu, name, user, use_color, line_number=None, width=0):
name_limit = 35 name_limit = 35
if len(name) > name_limit: if len(name) > name_limit:
display_name = name[:name_limit - 3] + "..." display_name = name[:name_limit - 3] + "..."
else: else:
display_name = name display_name = name
if cpu <= 25 or not use_color: if not use_color:
return f"{cpu:6.2f}% {display_name:<35} {user[:15]:<15}" prefix = f"{line_number:0{width}d}: " if line_number is not None else ""
return f"{prefix}{cpu:6.2f}% {display_name:<35} {user[:15]:<15}"
BG_GREEN = "\033[42m"
BG_YELLOW = "\033[43m"
BG_ORANGE = "\033[48;5;214m"
BG_RED = "\033[41m"
BLACK_TEXT = "\033[30m"
RESET = "\033[0m"
if cpu <= 25:
bg_color = BG_GREEN
elif cpu <= 50:
bg_color = BG_YELLOW
elif cpu <= 75:
bg_color = BG_ORANGE
else: else:
YELLOW = "\033[33m" bg_color = BG_RED
ORANGE = "\033[38;5;214m"
RED = "\033[31m" prefix = f"{line_number:0{width}d}: " if line_number is not None else ""
RESET_TO_GREEN = "\033[32m"
color = YELLOW if cpu <= 50 else ORANGE if cpu <= 75 else RED line = f"{bg_color}{BLACK_TEXT}{prefix}{cpu:6.2f}% {display_name:<35} {user[:15]:<15}{RESET}"
cpu_str = f"{cpu:6.2f}%" return line
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, show_line_numbers=False): 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 ""
RESET = "\033[0m" if use_color else ""
if use_color:
print(GREEN, end='')
line_prefix = "Ln " if show_line_numbers else "" line_prefix = "Ln " if show_line_numbers else ""
cpu_col = " CPU" cpu_col = " CPU"
name_col = " Process" name_col = " Process"
user_col = " User" user_col = " User"
header = f"{line_prefix}{cpu_col:<6} {name_col:<35} {user_col:<15}" header = f"{line_prefix}{cpu_col:<6} {name_col:<35} {user_col:<15}"
print(header) print(f"{GREEN}{header}{RESET}")
print("-" * len(header)) print(f"{GREEN}{'-' * (len(header) +1)}{RESET}")
width = len(str(limit)) width = len(str(limit))
for i, (cpu, name, user) in enumerate(top_processes, start=1): for i, (cpu, name, user) in enumerate(top_processes, start=1):
prefix = f"{i:0{width}d}: " if show_line_numbers else "" if show_line_numbers:
print(f"{prefix}{colorize(cpu, name, user, use_color)}") print(colorize(cpu, name, user, use_color, line_number=i, width=width))
else:
print(colorize(cpu, name, user, use_color))
if show_quit_hint: if show_quit_hint:
print("\nPress 'q' to quit.") print("\nPress 'q' to quit.")