psstree 0.1.0__py3-none-any.whl

psstree/__main__.py

@@ -0,0 +1,37 @@
+import argparse
+
+from .utils import Mapping
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="gets process' PSS via smem, sorts it descendingly within their parent-child hierarchy, and displays it in an interactive TUI with collapsible sections")
+
+    parser.add_argument(
+        "--expand",
+        action="store_true",
+        help="[TUI only] expand all collapsible sections at start"
+    )
+
+    parser.add_argument(
+        "--raw",
+        action="store_true",
+        help="disable TUI and dump formatted repr directly to stdout"
+    )
+
+    args = parser.parse_args()
+
+    mapping = Mapping()
+    mapping.create(repr=True)
+    raw_output = mapping.repr
+
+    if args.raw:
+        print(raw_output, flush=True)
+    else:
+        try:
+            from .tui import TreeApp
+            app = TreeApp(mapping=mapping, expand=args.expand)
+            app.run()
+        except Exception:
+            print(raw_output, flush=True)
+            raise
+
+    

psstree/tui.py

@@ -0,0 +1,53 @@
+from textual.widgets import Tree, Footer, Header
+from textual.app import App
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from utils import Mapping
+    from textual.widget._tree import TreeNode
+    from textual.app import ComposeResult
+
+__all__ = ["TreeApp"]
+
+
+
+def dfs(parent: "TreeNode", pid: int, mapping: "Mapping"):
+    proc = mapping.PROCS[pid]
+    node = parent.add(
+        label=proc.label,
+        data=proc,
+    )
+    for child in proc.children:
+        dfs(node, child, mapping)
+
+
+class TreeApp(App):
+    BINDINGS = [
+        ("q", "quit", "Quit"),
+    ]
+
+    def __init__(
+        self, 
+        mapping: "Mapping",
+        expand: bool = False, 
+        *args, 
+        **kwargs
+    ):
+        super().__init__(*args, **kwargs)
+        self.title = "Press q to quit"
+        self.mapping = mapping
+        self.expand = expand
+
+    def compose(self) -> "ComposeResult":
+        yield Header()
+        tree: Tree[str] = Tree(label="PSS (Proportional Set Size) `command(pid): <individual-PSS>/<total-PSS-including-all-children>`")
+        for root in self.mapping.ROOTS:
+            dfs(tree.root, root, self.mapping)
+        tree.root.expand()
+        for node in tree.root.children:
+            node.expand()
+        if self.expand:
+            tree.root.expand_all()
+        yield tree
+        yield Footer()
+    

psstree/utils.py

@@ -0,0 +1,134 @@
+import os
+import subprocess
+from dataclasses import dataclass, field
+from typing import List, Dict, Set, Optional
+
+@dataclass
+class Proc:
+    pid: int
+    ppid: int
+    comm: str
+    val: int
+
+    depth: int = 0
+    children: List[int] = field(default_factory=list)
+    traversal_order: List[str] = field(default_factory=list)
+    total: int = 0
+    label: str = "<??>"
+
+
+def get_indents(depths):
+    size = len(depths)
+
+    # depths[ < 0 ] -> infty
+    # depths[ >= size ] -> -1
+
+    result = []
+
+    for idx in range(size):
+        if idx>0 and depths[idx-1] < depths[idx]:
+            start = "  "*(depths[idx]-1) + "╰─"
+            if (idx+1)<size and depths[idx] <= depths[idx+1]:
+                start += "┬─"
+            else:
+                start += "──"
+        elif idx>0 and depths[idx-1] == depths[idx]:
+            start = "  "*depths[idx] 
+            if (idx+1)<size and depths[idx] <= depths[idx+1]:
+                start += "├─"
+            else:
+                start += "╰─"
+        elif idx==0 or depths[idx-1] > depths[idx]:
+            start = "  "*depths[idx] 
+            if (idx+1)<size and depths[idx] <= depths[idx+1]:
+                start += "┌─"
+            else:
+                start += "──"
+        
+        result.append(start)
+    
+    return result
+        
+
+class Mapping:
+    def __init__(self):
+        self.PROCS: Dict[int, Proc] = {}
+        self.ROOTS: List[int] = []
+        self.traversal_order: Optional[List[int]] = None
+        self.repr: str = "<??>"
+
+    def dfs(self, pid: int, depth: int, visited: Set[int], visiting: Set[int]):
+        if pid in visiting:
+            raise ValueError(f"circular reference \n{pid=}, {depth=}, {visited=}, {visiting=}, {self.ROOTS=}, {self.PROCS=}\n circular reference")
+        
+        if pid in visited:
+            raise ValueError(f"multiple parents \n{pid=}, {depth=}, {visited=}, {visiting=}, {self.ROOTS=}, {self.PROCS=}\n multiple parents")
+
+        visiting.add(pid)
+
+        proc = self.PROCS[pid]
+        
+        for child in proc.children:
+            self.dfs(child, depth+1, visited, visiting)
+            
+        proc.depth = depth
+        proc.children.sort(key=lambda id: self.PROCS[id].total, reverse=True)
+        proc.traversal_order = sum((self.PROCS[child].traversal_order for child in proc.children), start=[pid])
+        proc.total = sum((self.PROCS[child].total for child in proc.children), start=proc.val)
+        proc.label = f"{proc.comm}({proc.pid}): {proc.val/1024:.2f}MB/{proc.total/1024:.2f}MB"
+
+        visiting.remove(pid)
+        visited.add(pid)
+
+
+    def create(self, repr=False):
+        self.PROCS.clear()
+        self.ROOTS.clear()
+
+        lines = [
+            line.strip().split() 
+            for line in subprocess.check_output(
+                ["smem", "-c", "pid pss", "-H"], 
+                text=True
+            ).splitlines()
+            if line.strip()
+        ]
+        
+        PSSs = {int(line[0]): int(line[1]) for line in lines}
+
+        lines = [
+            line.strip().split() 
+            for line in subprocess.check_output(
+                ["ps", "-eo", "pid,ppid,comm", "--no-header"], 
+                text=True
+            ).splitlines() 
+            if line.strip()
+        ]
+
+        for pid, ppid, comm in lines:
+            pid = int(pid)
+            ppid = int(ppid)
+            if pid not in PSSs:
+                continue
+            self.PROCS[pid] = Proc(pid=pid, ppid=ppid, comm=comm, val=PSSs[pid])
+
+        for proc in self.PROCS.values():
+            if proc.ppid in self.PROCS:
+                self.PROCS[proc.ppid].children.append(proc.pid)
+            else:
+                self.ROOTS.append(proc.pid)
+        
+        for root in self.ROOTS:
+            self.dfs(root, 0, set(), set())
+
+        self.ROOTS.sort(key=lambda pid: self.PROCS[pid].total, reverse=True)
+        self.traversal_order = sum((self.PROCS[root].traversal_order for root in self.ROOTS), start=[])
+
+        if repr:
+            indents = get_indents([self.PROCS[pid].depth for pid in self.traversal_order])
+            
+            self.repr = "\n".join([
+                indent + self.PROCS[pid].label 
+                for indent, pid in zip(indents, self.traversal_order)
+            ])
+

psstree-0.1.0.dist-info/METADATA

@@ -0,0 +1,14 @@
+Metadata-Version: 2.4
+Name: psstree
+Version: 0.1.0
+Summary: list process' PSS sorted descendingly within their parent-child hierarchy
+Author: Anosrep Enilno
+Project-URL: Repository, https://github.com/anosrepenilno/psstree
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: textual>=8.1.1
+Dynamic: license-file
+
+# PSS-tree
+gets process' PSS via smem, sorts it descendingly within their parent-child hierarchy, and displays it in an interactive TUI with collapsible sections

psstree-0.1.0.dist-info/RECORD

@@ -0,0 +1,8 @@
+psstree/__main__.py,sha256=usGKnJai5BVcoKvVD2Rr6yajLl-idWKwzSv_cRTol2c,980
+psstree/tui.py,sha256=qK1wWTjXOJBQjh0VnwojNSsORjVVtCTZ8eGMkHlux6I,1339
+psstree/utils.py,sha256=jFl_Msa42RwO0cr3CCMUKVeUber81JONcSOa465zxns,4260
+psstree-0.1.0.dist-info/licenses/LICENSE,sha256=fyRfZ3h3EKGD6oFFvtUw8t1ksjijbQJZOoR_usLM29U,1070
+psstree-0.1.0.dist-info/METADATA,sha256=a1T7PiBXjwXAfsKaJ56FcnVzaBYOViIy-PbYyMkfFJw,524
+psstree-0.1.0.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+psstree-0.1.0.dist-info/top_level.txt,sha256=bUhxyIBQf_rY3BCj0prniIHV0oHedmEr2CDC5t_QtpA,8
+psstree-0.1.0.dist-info/RECORD,,

psstree-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (83.0.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

psstree-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 anosrepenilno
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

psstree-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+psstree