Adjust PID column width for longer PIDs

Closes #15. Thanks to @rookielxy .
peci1 · Aug 31, 2021 · bbec8f8 · bbec8f8
1 parent 7b53fba
commit bbec8f8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/nvidia-htop.py b/nvidia-htop.py
@@ -165,7 +165,11 @@ def colorize(_lines):
         time[idx] = parts[4] if "-" not in parts[4] else parts[4].split("-")[0] + " days"
         command[idx] = parts[5]
 
-format = ("|  %3s %5s %8s   %8s %5s %5s %9s  %-" + str(command_length) + "." + str(command_length) + "s  |")
+max_pid_length = max([len(str(x)) for x in pid])
+if max_pid_length >= 6:
+    command_length -= (max_pid_length - 6)
+
+format = ("|  %3s %" + str(max_pid_length) + "s %8s   %8s %5s %5s %9s  %-" + str(command_length) + "." + str(command_length) + "s |")
 
 print(format % (
     "GPU", "PID", "USER", "GPU MEM", "%CPU", "%MEM", "TIME", "COMMAND"