Add background color for Seperator Bar.

topless.py

@@ -12,91 +12,44 @@ import select
 import argparse
 import tty
 import termios
-import shutil
+from shutil import get_terminal_size
-
-first_run = True
-current_sort_key = 'cpu'
-reverse_sort = True
-
-def supports_color():
-    if not sys.stdout.isatty():
-        return False
-    term = os.environ.get('TERM', '')
-    if 'color' in term.lower():
-        return True
-    if os.name == 'nt':
-        return True
-    return False
 
 def clear_screen():
-    global first_run
+    print("\033[H\033[J", end='')
-    if first_run:
-        os.system('cls' if os.name == 'nt' else 'clear')
-        first_run = False
-    else:
-        print("\033[H", end='')
-
-def get_top_processes(limit, sort_key='cpu', reverse=True):
-    for proc in psutil.process_iter(['pid']):
-        try:
-            proc.cpu_percent(interval=None)
-        except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
-            continue
-
-    time.sleep(0.1)
 
+def get_top_processes(limit=10, sort_key='cpu', reverse=True):
     processes = []
-    for proc in psutil.process_iter(['cpu_percent', 'memory_percent', 'name', 'username']):
+    for proc in psutil.process_iter(['pid', 'name', 'cpu_percent', 'memory_percent', 'username']):
         try:
-            cpu = proc.info['cpu_percent']
+            pinfo = proc.info
-            mem = proc.info['memory_percent']
+            processes.append({
-            name = proc.info['name']
+                'cpu': pinfo['cpu_percent'] or 0.0,
-            user = proc.info['username']
+                'mem': pinfo['memory_percent'] or 0.0,
-            processes.append({'cpu': cpu, 'mem': mem, 'name': name, 'user': user})
+                'name': pinfo['name'][:30] if pinfo['name'] else '',
+                'user': pinfo['username'][:14] if pinfo['username'] else ''
+            })
         except (psutil.NoSuchProcess, psutil.AccessDenied, psutil.ZombieProcess):
             continue
+    return sorted(processes, key=lambda x: x[sort_key], reverse=reverse)[:limit]
 
-    processes = [
+def colorize(cpu, mem, name, user, use_color=True, line_number=None, width=0):
-        p for p in processes
-        if p['cpu'] is not None and
-           p['mem'] is not None and
-           p['name'] is not None and
-           p['user'] is not None
-    ]
-
-    processes.sort(key=lambda p: p[sort_key], reverse=reverse)
-    return processes[:limit]
-
-def colorize(cpu, mem, name, user, use_color, line_number=None, width=0):
-    name_limit = 35
-    if len(name) > name_limit:
-        display_name = name[:name_limit - 3] + "..."
-    else:
-        display_name = name
-
     if not use_color:
-        prefix = f"{line_number:0{width}d}: " if line_number is not None else ""
+        return f"{str(line_number).rjust(width)}  {cpu:5.1f}    {mem:5.1f}    {name:35}  {user:15}" if line_number else f"{cpu:5.1f}    {mem:5.1f}    {name:35}  {user:15}"
-        return f"{prefix}{cpu:6.2f}%  {mem:6.2f}%  {display_name:<35}  {user[:15]:<15}"
 
-    BG_GREEN = "\033[42m"
+    if cpu >= 50:
-    BG_YELLOW = "\033[43m"
+        color = "\033[91m"  # red
-    BG_ORANGE = "\033[48;5;214m"
+    elif cpu >= 20:
-    BG_RED = "\033[41m"
+        color = "\033[93m"  # yellow
-    BLACK_TEXT = "\033[30m"
+    elif cpu >= 10:
-    RESET = "\033[0m"
+        color = "\033[92m"  # green
-
-    if cpu <= 25:
-        bg_color = BG_GREEN
-    elif cpu <= 50:
-        bg_color = BG_YELLOW
-    elif cpu <= 75:
-        bg_color = BG_ORANGE
     else:
-        bg_color = BG_RED
+        color = "\033[90m"  # gray
 
-    prefix = f"{line_number:0{width}d}: " if line_number is not None else ""
+    RESET = "\033[0m"
-    line = f"{bg_color}{BLACK_TEXT}{prefix}{cpu:6.2f}%  {mem:6.2f}%  {display_name:<35}  {user[:15]:<15}{RESET}"
+    line = f"{cpu:5.1f}    {mem:5.1f}    {name:35}  {user:15}"
-    return line
+    if line_number is not None:
+        return f"{color}{str(line_number).rjust(width)}  {line}{RESET}"
+    return f"{color}{line}{RESET}"
 
 def print_processes(limit, sort_key='cpu', reverse=True, show_quit_hint=True, use_color=True, show_line_numbers=False):
     clear_screen()
@@ -109,6 +62,9 @@ def print_processes(limit, sort_key='cpu', reverse=True, show_quit_hint=True, us
     ARROW_UP = "↑"
     ARROW_DOWN = "↓"
 
+    BG_GRAY = "\033[100m" if use_color else ""
+    BLACK_TEXT = "\033[30m" if use_color else ""
+
     arrow = ARROW_DOWN if reverse else ARROW_UP
 
     def col_label(col_key, label, width):
@@ -117,18 +73,15 @@ def print_processes(limit, sort_key='cpu', reverse=True, show_quit_hint=True, us
         else:
             return f"  {label}".ljust(width)
 
-    cpu_col = col_label('cpu', "CPU", 6)
+    cpu_col = col_label('cpu', "CPU", 7)
-    mem_col = col_label('mem', "MEM", 6)
+    mem_col = col_label('mem', "MEM", 5)
     name_col = col_label('name', "Process", 35)
     user_col = col_label('user', "User", 15)
 
     line_prefix = "Ln  " if show_line_numbers else ""
     header = f"{line_prefix}{cpu_col}  {mem_col}  {name_col}  {user_col}"
-    separator = '-' * (len(header) + 2)  # <-- Increased length by 1 here
-    term_width = shutil.get_terminal_size((80, 20)).columns
 
-    print(f"{GREEN}{header.ljust(term_width)}{RESET}")
+    print(f"{BG_GRAY}{BLACK_TEXT}{header.ljust(len(header))}{RESET}")
-    print(f"{GREEN}{separator.ljust(term_width)}{RESET}")
 
     width = len(str(limit))
     for i, proc in enumerate(top_processes, start=1):
@@ -140,86 +93,71 @@ def print_processes(limit, sort_key='cpu', reverse=True, show_quit_hint=True, us
             line = colorize(cpu, mem, name, user, use_color, line_number=i, width=width)
         else:
             line = colorize(cpu, mem, name, user, use_color)
-        print(line.ljust(term_width))
+        print(line.ljust(len(header)))
 
-    print(f"{GREEN}{separator.ljust(term_width)}{RESET}", end='')
+    bottom_bar = "Sort: [C]PU [M]EM [P]rocess [U]ser or [Q]uit."
     if show_quit_hint:
-        print((
+        print(f"{BG_GRAY}{BLACK_TEXT}{bottom_bar.ljust(len(header))}{RESET}")
-            f"{GREEN} Sort: "
+    else:
-            f"[{UNDERLINE}C{RESET_UNDERLINE}]PU "
+        print(f"{BG_GRAY}{' '.ljust(len(header))}{RESET}")
-            f"[{UNDERLINE}M{RESET_UNDERLINE}]EM "
-            f"[{UNDERLINE}P{RESET_UNDERLINE}]rocess "
-            f"[{UNDERLINE}U{RESET_UNDERLINE}]ser or "
-            f"[{UNDERLINE}Q{RESET_UNDERLINE}]uit.{RESET}"
-        ).ljust(term_width))
 
-def wait_for_keypress(timeout=1.0):
+def timed_input(timeout=0.1):
     fd = sys.stdin.fileno()
     old_settings = termios.tcgetattr(fd)
     try:
         tty.setcbreak(fd)
-        if select.select([sys.stdin], [], [], timeout)[0]:
+        rlist, _, _ = select.select([fd], [], [], timeout)
+        if rlist:
             return sys.stdin.read(1)
     finally:
         termios.tcsetattr(fd, termios.TCSADRAIN, old_settings)
     return None
 
-class SortedHelpFormatter(argparse.ArgumentDefaultsHelpFormatter):
-    def add_arguments(self, actions):
-        actions = sorted(actions, key=lambda a: a.option_strings)
-        super().add_arguments(actions)
-
 def main():
-    global current_sort_key, reverse_sort
+    parser = argparse.ArgumentParser()
-
+    parser.add_argument('-n', type=int, help="Number of processes to display")
-    parser = argparse.ArgumentParser(
+    parser.add_argument('--no-color', action='store_true', help="Disable color output")
-        description="Monitor system processes by CPU and memory usage.",
+    parser.add_argument('--lines', action='store_true', help="Show line numbers")
-        formatter_class=SortedHelpFormatter
-    )
-    parser.add_argument("-i", "--interval", type=int, default=1, help="Refresh interval in seconds.")
-    parser.add_argument("-l", "--line", action="store_true", help="Show line numbers.")
-    parser.add_argument("-n", "--number", type=int, default=10, help="Number of processes to monitor.")
-    parser.add_argument("--no-color", action="store_true", help="Disable colored output.")
     args = parser.parse_args()
 
-    limit = args.number
+    use_color = not args.no_color
-    interval = max(1, args.interval)
+    show_line_numbers = args.lines
 
-    use_color = not args.no_color and supports_color()
+    term_height = get_terminal_size().lines
-    show_lines = args.line
+    limit = args.n if args.n else max(5, term_height - 10)
 
+    sort_key = 'cpu'
+    reverse = True
     refresh_count = 0
     start_time = time.time()
 
-    try:
     while True:
-            refresh_count += 1
         elapsed = time.time() - start_time
-            show_quit = refresh_count < 3 and elapsed < 3
+        show_quit = True
-            print_processes(
+
-                limit,
+        print_processes(limit, sort_key, reverse, show_quit, use_color, show_line_numbers)
-                sort_key=current_sort_key,
+
-                reverse=reverse_sort,
+        key = timed_input(1)
-                show_quit_hint=show_quit,
+        refresh_count += 1
-                use_color=use_color,
-                show_line_numbers=show_lines
-            )
 
-            key = wait_for_keypress(timeout=interval)
         if key:
             key = key.lower()
             if key == 'q':
                 break
-                elif key in ['c', 'm', 'p', 'u']:
+            elif key == 'c':
-                    key_map = {'c': 'cpu', 'm': 'mem', 'p': 'name', 'u': 'user'}
+                reverse = not reverse if sort_key == 'cpu' else True
-                    chosen_sort = key_map[key]
+                sort_key = 'cpu'
-                    if current_sort_key == chosen_sort:
+            elif key == 'm':
-                        reverse_sort = not reverse_sort
+                reverse = not reverse if sort_key == 'mem' else True
-                    else:
+                sort_key = 'mem'
-                        current_sort_key = chosen_sort
+            elif key == 'p':
-                        reverse_sort = True
+                reverse = not reverse if sort_key == 'name' else True
-    except KeyboardInterrupt:
+                sort_key = 'name'
-        print("\nProgram interrupted. Exiting...")
+            elif key == 'u':
+                reverse = not reverse if sort_key == 'user' else True
+                sort_key = 'user'
+            elif key == 'r':
+                reverse = not reverse
 
 if __name__ == "__main__":
     main()