adds count_active_terminals method to retrieve unique active terminal sessions

hashedalgorithm · hashedalgorithm · commit ffc92784df05 · 2025-07-06T19:10:44.000+02:00
diff --git a/src/models/process_group.py b/src/models/process_group.py
@@ -1,3 +1,4 @@
+import subprocess
 from models.process import Process
 
 
@@ -323,6 +324,30 @@ def create_fake_processes(self) -> None:
                 )
             )
 
+    def count_active_terminals():
+        try:
+            # Run 'who' to get active login sessions
+            result = subprocess.run(
+                ["who"], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True
+            )
+
+            if result.returncode != 0:
+                raise RuntimeError(f"Error running 'who': {result.stderr}")
+
+            lines = result.stdout.strip().split("\n")
+
+            # Extract terminals from the output (2nd column)
+            terminals = [line.split()[1] for line in lines if line]
+
+            # Remove duplicates, in case multiple sessions use the same terminal
+            unique_terminals = set(terminals)
+
+            return len(unique_terminals)
+
+        except Exception as e:
+            print(f"Error checking active terminals: {e}")
+            return 0
+
     def add_process(self, process: Process) -> None:
         """Add a new process to the process group."""
         if process.pid in self.processes:
