bhtool 0.2.1__tar.gz → 0.2.3__tar.gz

{bhtool-0.2.1 → bhtool-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: bhtool
-Version: 0.2.1
+Version: 0.2.3
 Summary: Assorted CLI utilities for Bosco’s workflows (dev, media, macOS, LLM helpers).
 Author-email: Bosco Ho <apposite@gmail.com>
 Requires-Python: >=3.13
@@ -39,6 +39,7 @@ Assorted CLI utilities that are handy in Bosco’s day-to-day work—version bum
 | **clr-chmod** | Remove execute bit from non-shell files; tighten group/other write                                                                                                                              |     |
 | **npread**    | Print the shape of NumPy`.npy` files                                                                                                                                                            |     |
 | **psword**    | Find processes by name;`--kill` to terminate                                                                                                                                                    |     |
+| **portkill**  | Find processes listening on TCP ports;`--kill` to terminate                                                                                                                                     |     |
 | **rm-npm**    | Recursively delete`node_modules` dirs and `package-lock.json` files                                                                                                                             |     |
 | **text**      | Convert between Markdown, HTML, DOCX, and Pug (needs`pandoc` and related tools on `PATH`)                                                                                                       |     |
 | **todict**    | JSON or YAML file, or stdin, → Python`dict(...)`-style repr on stdout                                                                                                                           |     |

{bhtool-0.2.1 → bhtool-0.2.3}/README.md

@@ -23,6 +23,7 @@ Assorted CLI utilities that are handy in Bosco’s day-to-day work—version bum
 | **clr-chmod** | Remove execute bit from non-shell files; tighten group/other write                                                                                                                              |     |
 | **npread**    | Print the shape of NumPy`.npy` files                                                                                                                                                            |     |
 | **psword**    | Find processes by name;`--kill` to terminate                                                                                                                                                    |     |
+| **portkill**  | Find processes listening on TCP ports;`--kill` to terminate                                                                                                                                     |     |
 | **rm-npm**    | Recursively delete`node_modules` dirs and `package-lock.json` files                                                                                                                             |     |
 | **text**      | Convert between Markdown, HTML, DOCX, and Pug (needs`pandoc` and related tools on `PATH`)                                                                                                       |     |
 | **todict**    | JSON or YAML file, or stdin, → Python`dict(...)`-style repr on stdout                                                                                                                           |     |

{bhtool-0.2.1 → bhtool-0.2.3}/bhtool/cli.py

@@ -60,6 +60,21 @@ def psword_cmd(*words: str, kill: bool = False):
     psword_mod.psword(words, kill=kill)
 
 
+@app.command(name="portkill")
+def portkill_cmd(*ports: int, kill: bool = False):
+    """Find processes listening on TCP ports and optionally kill them.
+
+    :param ports: TCP ports to inspect (one or more).
+    :param kill: If true, SIGKILL each listener; otherwise list them.
+    """
+    if not ports:
+        app["portkill"].help_print()
+        return
+    import bhtool.portkill as portkill_mod
+
+    portkill_mod.portkill(ports, kill=kill)
+
+
 @app.command(name="rm_npm")
 def rm_npm_cmd():
     """Recursively remove node_modules directories and package-lock.json files."""

bhtool-0.2.3/bhtool/portkill.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+
+"""Find and kill processes listening on TCP ports"""
+
+import os
+import re
+import signal
+
+from bhtool.utils import run_output
+
+
+def _lsof_pids(port: int) -> list[int]:
+    txt = run_output(f"lsof -nP -iTCP:{port} -sTCP:LISTEN -t")
+    return [int(line) for line in txt.splitlines() if line.strip().isdigit()]
+
+
+def _ss_pids(port: int) -> list[int]:
+    txt = run_output(f"ss -ltnp 'sport = :{port}'")
+    return [int(pid) for pid in re.findall(r"pid=(\d+)", txt)]
+
+
+def _command(pid: int) -> str:
+    cmd = run_output(f"ps -p {pid} -o command=").strip()
+    if len(cmd) > 80:
+        cmd = cmd[:80] + "..."
+    return cmd
+
+
+def find_listeners(port: int) -> list[tuple[int, str]]:
+    """Return ``(pid, command)`` pairs for processes LISTENing on ``port``."""
+    pids = _lsof_pids(port) or _ss_pids(port)
+    return [(pid, _command(pid)) for pid in sorted(set(pids))]
+
+
+def portkill(ports: tuple[int, ...], kill: bool = False):
+    for port in ports:
+        listeners = find_listeners(port)
+        if not listeners:
+            print(f"no process listening on port {port}")
+            continue
+        for pid, cmd in listeners:
+            if kill:
+                print(f"kill target ({cmd})")
+                try:
+                    os.kill(pid, signal.SIGKILL)
+                except ProcessLookupError:
+                    print(f"process {pid} already gone")
+            else:
+                print(f"process {pid}: {cmd}")

bhtool-0.2.3/bhtool/psword.py

@@ -0,0 +1,53 @@
+#!/usr/bin/env python3
+
+"""Find processes with name"""
+
+import os
+import re
+import signal
+
+from bhtool.utils import run_output
+
+
+def own_pids() -> set[int]:
+    """Our process and its ancestors, which must never be killed."""
+    parents = {
+        int(pid): int(ppid)
+        for pid, ppid in re.findall(
+            r"(\d+)\s+(\d+)", run_output("ps -o pid=,ppid= -ax")
+        )
+    }
+    pids = {os.getpid()}
+    pid = os.getpid()
+    while pid in parents and parents[pid] > 1 and parents[pid] not in pids:
+        pid = parents[pid]
+        pids.add(pid)
+    return pids
+
+
+def psword(words: tuple[str, ...], kill: bool = False):
+    killed: set[int] = set()
+    skip = own_pids()
+    for word in words:
+        txt = run_output(f"ps aux | grep {word}")
+        for line in txt.splitlines():
+            tokens = line.split(None, 10)
+            if len(tokens) < 11:
+                continue
+            cmd = tokens[10]
+            if cmd.startswith("grep ") or "ps aux | grep" in cmd:
+                continue
+            pid = int(tokens[1])
+            if pid in skip or pid in killed:
+                continue
+            if len(cmd) > 80:
+                cmd = cmd[:80] + "..."
+            if kill:
+                print(f"kill target ({cmd})")
+                try:
+                    os.kill(pid, signal.SIGKILL)
+                    killed.add(pid)
+                except (ProcessLookupError, PermissionError) as err:
+                    print(f"could not kill {pid}: {err}")
+            else:
+                print(f"process {pid}: {cmd}")

{bhtool-0.2.1 → bhtool-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bhtool"
-version = "0.2.1"
+version = "0.2.3"
 description = "Assorted CLI utilities for Bosco’s workflows (dev, media, macOS, LLM helpers)."
 readme = "README.md"
 requires-python = ">=3.13"

bhtool-0.2.1/bhtool/psword.py

@@ -1,23 +0,0 @@
-#!/usr/bin/env python3
-
-"""Find processes with name"""
-
-from bhtool.utils import run, run_output
-
-
-def psword(words: tuple[str, ...], kill: bool = False):
-    for word in words:
-        txt = run_output(f"ps aux | grep {word}")
-        for line in txt.splitlines():
-            tokens = line.split()
-            i_process = tokens[1]
-            cmd = " ".join(tokens[10:])
-            if len(cmd) > 80:
-                cmd = cmd[:80] + "..."
-            if "psword" in line:
-                continue
-            if kill:
-                print(f"kill target ({cmd})")
-                run(f"kill -9 {i_process}")
-            else:
-                print(f"process {i_process}: {cmd}")

bhtool-0.2.1/movie_mapping.json

@@ -1,25 +0,0 @@
-{
-  "directories": [
-    {
-      "original": "bhtool",
-      "normalized": "Bhtool"
-    },
-    {
-      "original": "build",
-      "normalized": "Build"
-    },
-    {
-      "original": "dist",
-      "normalized": "Dist"
-    },
-    {
-      "original": "jtools",
-      "normalized": "Jtools"
-    },
-    {
-      "original": "src",
-      "normalized": "Src"
-    }
-  ],
-  "files": []
-}