richqueue 1.11__tar.gz → 1.11.1__tar.gz

{richqueue-1.11 → richqueue-1.11.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: richqueue
-Version: 1.11
+Version: 1.11.1
 Summary: RichQueue
 Project-URL: Homepage, https://github.com/mwinokan/RichQueue/issues
 Project-URL: Bug Tracker, https://github.com/mwinokan/RichQueue/issues

{richqueue-1.11 → richqueue-1.11.1}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["hatchling"]
 build-backend = "hatchling.build"
 [project]
 name = "richqueue"
-version = "1.11"
+version = "1.11.1"
 authors = [
     { name = "Max Winokan", email = "max@winokan.com" },
 ]

{richqueue-1.11 → richqueue-1.11.1}/richqueue/submit.py

@@ -9,6 +9,24 @@ from .console import console
 
 def submit():
 
+    sbatch_args = sys.argv[1:]
+
+    if not sbatch_args:
+        console.print("[red bold]Please provide sbatch arguments")
+        console.print(
+            "\n[bold yellow]Usage:[reset][bold] sb [SBATCH OPTIONS] SCRIPT [SCRIPT ARGS]...\n"
+        )
+        sys.exit(1)
+
+    elif len(sbatch_args) == 1 and sbatch_args[0] in ["-h", "--help"]:
+        console.print(
+            "\n[bold]sb[reset]: a pretty wrapper to the sbatch command that places log files in the directory specified by the LOGS variable"
+        )
+        console.print(
+            "\n[bold yellow]Usage:[reset][bold] sb [SBATCH OPTIONS] SCRIPT [SCRIPT ARGS]...\n"
+        )
+        sys.exit(1)
+
     log_dir = environ.get("LOGS")
 
     if not log_dir:
@@ -17,8 +35,6 @@ def submit():
 
     log_dir = Path(log_dir)
 
-    sbatch_args = sys.argv[1:]
-
     commands = [
         "sbatch",
         "--output=" f"{log_dir.resolve()}/%j.log",