ptop: Enable searching by PID

px/px_process.py

@@ -214,6 +214,9 @@ def match(self, string, require_exact_user=True):
         if string in self.cmdline.lower():
             return True
 
+        if str(self.pid).startswith(string):
+            return True
+
         return False
 
     def get_command_line_array(self):

px/px_top.py

@@ -292,8 +292,15 @@ def get_screen_lines(
 
         # Put exact search matches first. Useful for "px cat" or other short
         # search strings with tons of hits.
+        search_pid = -1
+        try:
+            search_pid = int(search)
+        except ValueError:
+            pass
         toplist = sorted(
-            toplist, key=lambda p: search in (p.command, p.username), reverse=True
+            toplist,
+            key=lambda p: search in (p.command, p.username) or p.pid == search_pid,
+            reverse=True,
         )
 
     # Hand out different amount of lines to the different sections

tests/px_process_test.py

@@ -278,6 +278,14 @@ def test_match():
     assert p.match("air")
     assert p.match("play")
 
+    # Match PID by prefix but not substring. Exact matches are used for
+    # searching in ptop. Prefix matching is used to not throw the right answer
+    # away while the user is typing their search in ptop. Substring matching has
+    # no value.
+    assert p.match("47536")
+    assert p.match("4753")
+    assert not p.match("7536")
+
 
 def test_seconds_to_str():
     assert px_process.seconds_to_str(0.54321) == "0.54s"