winci72

Author: jet-logic

runce/cli.py

@@ -102,6 +102,8 @@ class Kill(Main):
     ids: list[str] = arg("ID", "run ids", nargs="+")
     dry_run: bool = flag("dry-run", "dry run (don't actually kill)", default=False)
     remove: bool = flag("remove", "remove entry after killing", default=False)
+    group: bool = flag("group", "kill process group", default=False)
+    signal: str = flag("signal", "send signal", default=None)
 
     def init_argparse(self, argp: ArgumentParser) -> None:
         argp.description = "Kill the process of a run id"
@@ -111,6 +113,7 @@ def start(self) -> None:
         _errdef = ["❌", "Error"]
         _noproc = ["👻", "No process"]
         _killed = ["💀", "Killed"]
+        signal = int(self.signal) if self.signal else None
         sp = Manager()
         if self.ids:
             for x in sp.find_names(self.ids, ambiguous, no_record):
@@ -119,7 +122,7 @@ def start(self) -> None:
                     s = _killed
                 else:
                     if check_pid(x["pid"]):
-                        if kill_pid(x["pid"], process_group=True):
+                        if kill_pid(x["pid"], process_group=self.group):
                             s = _killed
                     else:
                         s = _noproc

tests/test_windows.py

@@ -230,10 +230,15 @@ def test_emoji(self):
             # Try printing a test emoji
             print("Test emoji: 😊")
         except UnicodeEncodeError:
+            print("No emoji :-(")
             self.assertTrue(1)
         else:
+            print("Has emoji :-)")
             self.assertTrue(1)
 
+    def test_emoji_2(self):
+        sys.stdout.buffer.write("Test emoji: 😊".encode())
+
 
 if __name__ == "__main__":
     unittest.main()