pmflow 1.0.0__py3-none-any.whl

pm/main.py

@@ -0,0 +1,150 @@
+#!/usr/bin/env python3
+import os
+import signal
+import sys
+
+import typer
+import psutil
+import subprocess
+from rich.console import Console
+from rich.table import Table
+from pm.utils import StateManager
+
+app = typer.Typer()
+
+STATE_FILE = os.path.join(os.path.dirname(__file__), 'processes_state.json')
+state = StateManager(STATE_FILE)
+
+@app.command()
+def greet(name: str):
+    print(f"Hello, {name}!")
+
+
+@app.command()
+def create(command: str, name: str = None) -> int:
+    """Create a new subprocess and optionally assign a name."""
+    proc = subprocess.Popen(command, shell=True)
+    pid = proc.pid
+    data = {
+        "command": command,
+        "name": name,
+    }
+    state.add_process(pid, data)
+    typer.echo(f"{pid}")
+
+
+@app.command()
+def pause(pid: int):
+    """Pause a subprocess and all its children by PID."""
+    pid_str = str(pid)
+    if pid_str in state.processes:
+        try:
+            process = psutil.Process(int(pid))
+            all_processes = [process] + process.children(recursive=True)
+            for proc in all_processes:
+                proc.send_signal(signal.SIGSTOP)
+            typer.echo(f"Process {pid} and its child processes have been paused.")
+        except psutil.NoSuchProcess:
+            typer.echo("Process not found.")
+    else:
+        typer.echo("Process not managed by this tool.")
+
+
+@app.command()
+def ls():
+    """List all managed subprocesses."""
+    table = Table(title="Processes")
+    table.add_column("PID", justify="right", style="cyan")
+    table.add_column("Name", justify="right", style="magenta")
+    table.add_column("Status", justify="right", style="green")
+    table.add_column("Command", justify="right", style="yellow")
+
+    for pid, properties in state.processes.items():
+        try:
+            process = psutil.Process(int(pid))
+            if process.status() == psutil.STATUS_STOPPED:
+                status = 'paused'
+            else:
+                status = 'running'
+        except psutil.NoSuchProcess:
+            status = "doesn't exist"
+
+        table.add_row(pid, properties["name"], status, properties["command"])
+
+    console = Console()
+    console.print(table)
+
+@app.command()
+def recreate():
+    """Recreate all managed subprocesses."""
+    new_processes = {}
+    for pid, data in state.processes.items():
+        proc = subprocess.Popen(data["command"], shell=True)
+        new_processes[str(proc.pid)] = data
+        typer.echo(f"Process {proc.pid} recreated with command: {data['command']}")
+
+    state.bulk_update(new_processes)
+
+
+@app.command()
+def respawn_all():
+    """Respawn processes that are in the JSON file but not running."""
+
+    for pid, data in state.processes.items():
+        if psutil.pid_exists(int(pid)):
+            process = psutil.Process(int(pid))
+            if not process.is_running():
+                typer.echo(f"Process {pid} not running. Respawning...")
+                process.resume()
+                typer.echo(f"Process {pid} respawed.")
+
+    typer.echo("Respawn complete.")
+
+
+@app.command()
+def kill(pid: int):
+    """Kill a subprocess by PID."""
+    pid_str = str(pid)
+    if pid_str in state.processes:
+        try:
+            process = psutil.Process(int(pid))
+            for child in process.children(recursive=True):
+                child.terminate()
+            process.terminate()
+            state.remove_process(pid_str)
+            typer.echo(f"Process {pid} killed.")
+        except psutil.NoSuchProcess:
+            state.remove_process(pid_str)
+            typer.echo("Process not found. Removed from the state file.")
+    else:
+        typer.echo("Process not managed by this tool.")
+
+
+@app.command()
+def kill_all():
+    """Kill all managed processes and clear the state."""
+
+    for pid in state.processes.keys():
+        try:
+            process = psutil.Process(int(pid))
+            for child in process.children(recursive=True):
+                child.terminate()
+            process.terminate()
+            typer.echo(f"Process {pid} terminated.")
+        except psutil.NoSuchProcess:
+            typer.echo(f"Process {pid} not found.")
+        except Exception as e:
+            typer.echo(f"Error terminating process {pid}: {str(e)}")
+
+    state.remove_all_processes()
+    typer.echo("All processes have been terminated and removed from the state.")
+
+
+
+def signal_handler(sig, frame):
+    typer.echo("Ctrl+C pressed. Terminating all managed processes...")
+    sys.exit(0)
+
+if __name__ == "__main__":
+    signal.signal(signal.SIGINT, signal_handler)
+    app()

pm/utils.py

@@ -0,0 +1,59 @@
+import json
+import os
+import sys
+
+import typer
+
+
+class StateManager:
+    def __init__(self, STATE_FILE):
+        self.processes = {}
+        self.STATE_FILE = STATE_FILE
+        self.load_state()
+
+    def load_state(self):
+        print(f"Loading state from {self.STATE_FILE}")
+
+        if not os.path.exists(self.STATE_FILE):
+            with open(self.STATE_FILE, "w") as file:
+                json.dump({}, file)
+                print("Created new state file")
+
+        with open(self.STATE_FILE, "r") as file:
+            self.processes = json.load(file)
+
+    def save(self):
+        with open(self.STATE_FILE, "w") as file:
+            json.dump(self.processes, file)
+
+    def get_processes(self):
+        return self.processes
+
+    def add_process(self, pid, data):
+        self.processes[str(pid)] = data
+        self.save()
+
+    def remove_process(self, pid):
+        self.processes.pop(str(pid), None)
+        self.save()
+
+    def remove_all_processes(self):
+        self.processes = {}
+        self.save()
+
+    def update_process(self, pid, key, value):
+        self.processes[str(pid)][key] = value
+        self.save()
+
+    def bulk_update(self, bulk_data):
+        self.processes = bulk_data
+        self.save()
+
+
+
+def load_state(path):
+    """Load processes state from a file."""
+    global processes
+    if os.path.exists(path):
+        with open(path, "r") as file:
+            processes = json.load(file)

pmflow-1.0.0.dist-info/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Md. Mahmudul Hasan Riyad
+
+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.

pmflow-1.0.0.dist-info/METADATA

@@ -0,0 +1,39 @@
+Metadata-Version: 2.1
+Name: pmflow
+Version: 1.0.0
+Summary: Manages processes
+Author-email: "Md. Mahmudul Hasan" <mhriyad98@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2024 Md. Mahmudul Hasan Riyad
+        
+        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.
+Project-URL: Homepage, https://github.com/mhriyad99/process_manager
+Keywords: process manager,process,pm
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: typer==0.12.5
+Requires-Dist: psutil==6.0.0
+Provides-Extra: dev
+
+# process_manager

pmflow-1.0.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+pm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pm/main.py,sha256=eazhzDlMO7vF2bIfElQaw2cxTVbuQY6quzir3Wzb0n4,4508
+pm/utils.py,sha256=_1sqLtujb8F40i3mT-Ca09kOcqaonuvVGjtbpWJfmK8,1420
+pmflow-1.0.0.dist-info/LICENSE,sha256=HpmlPSgC0Aaxf2PwgtpD6xPlsb9j66ZipQmC8jdeKqE,1080
+pmflow-1.0.0.dist-info/METADATA,sha256=lf-GeRzapHlRciSdmep8cRE-7Ya5QGd9xhn96pTDEhc,1814
+pmflow-1.0.0.dist-info/WHEEL,sha256=OVMc5UfuAQiSplgO0_WdW7vXVGAt9Hdd6qtN4HotdyA,91
+pmflow-1.0.0.dist-info/entry_points.txt,sha256=nP0oRbAZsbgglMxaXniGwXOew_4qYamJbstYx0FoalE,35
+pmflow-1.0.0.dist-info/top_level.txt,sha256=EyGjaLlndAMIL4TKnkAuosD9vW0v2Ci2tenil9yD3MU,3
+pmflow-1.0.0.dist-info/RECORD,,

pmflow-1.0.0.dist-info/WHEEL

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

pmflow-1.0.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+pm = pm.main:app

pmflow-1.0.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+pm