Add scancel

Author: lululxvi

README.md

@@ -4,12 +4,13 @@
 [![Downloads](https://pepy.tech/badge/sumsjob)](https://pepy.tech/project/sumsjob)
 [![License](https://img.shields.io/github/license/lululxvi/sumsjob)](https://github.com/lululxvi/sumsjob/blob/master/LICENSE)
 
-&Sigma;&Sigma;<sub>Job</sub> or Sums<sub>Job</sub> (**S**imple **U**tility for **M**ultiple-**S**ervers **Job** **Sub**mission) is a simple Linux command-line utility which submits a job to one of the multiple servers each with limited resources such as GPUs. &Sigma;&Sigma;<sub>Job</sub> provides similar key functions for multiple servers as [Slurm Workload Manager](https://slurm.schedmd.com) for supercomputers and computer clusters. It provides four key functions:
+&Sigma;&Sigma;<sub>Job</sub> or Sums<sub>Job</sub> (**S**imple **U**tility for **M**ultiple-**S**ervers **Job** **Sub**mission) is a simple Linux command-line utility which submits a job to one of the multiple servers each with limited resources such as GPUs. &Sigma;&Sigma;<sub>Job</sub> provides similar key functions for multiple servers as [Slurm Workload Manager](https://slurm.schedmd.com) for supercomputers and computer clusters. It provides the following key functions:
 
 - show the status of GPUs on all servers,
 - submit a job to servers in noninteractive mode, i.e., the job will be running in the background of the server,
 - submit a job to servers in interactive mode, just as the job is running in your local machine,
-- display all running jobs.
+- display all running jobs,
+- cancel running jobs.
 
 ## Motivation
 
@@ -83,6 +84,12 @@ Display all running jobs ordered by the start time. For example,
 
 ![](https://github.com/lululxvi/sumsjob/blob/master/docs/figs/sacct.png)
 
+### `$ scancel jobname`
+
+Cancel a running job.
+
+- `jobname` : Job name.
+
 ## Installation
 
 Install Sums<sub>Job</sub> with `pip`:

sumsjob/__init__.py

@@ -1,7 +1,8 @@
-__all__ = ["gpuresource", "sacct", "submit"]
+__all__ = ["gpuresource", "sacct", "scancel", "submit"]
 
 from .__about__ import __version__
 
 from . import gpuresource
 from . import sacct
+from . import scancel
 from . import submit

sumsjob/cancel.py

@@ -0,0 +1,31 @@
+__al__ = ["scancel"]
+
+import argparse
+import subprocess
+
+from .sacct import sacct
+from .utils import local_cmdline
+
+
+def scancel(jobname):
+    jobs = sacct()
+    for job in jobs:
+        if job["JobName"] == jobname:
+            cmd = f"screen -S sumsjob-{jobname} -X quit"
+            cmd = local_cmdline(job["Server"], cmd, verbose=0)
+            subprocess.run(cmd, shell=True, check=True)
+            print(f"Job {jobname} on {job['Server']} cancelled.")
+            return
+    print(f"Job {jobname} not found.")
+
+
+def main():
+    parser = argparse.ArgumentParser(description="Cancel a running job.")
+    parser.add_argument("jobname", help="Job name")
+    args = parser.parse_args()
+
+    scancel(args.jobname)
+
+
+if __name__ == "__main__":
+    main()