Fixed wrong information was passed to spooler regex

neilmunday · Nov 20, 2023 · 328f332 · 328f332
1 parent d66798b
commit 328f332
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/src/slurmmail/cli.py b/src/slurmmail/cli.py
@@ -790,10 +790,10 @@ def spool_mail_main():
         die("Incorrect number of command line arguments")
 
     try:
+        info = ",".join(sys.argv[2].split(","))
+        logging.debug("info str: %s", info)
         match = None
-        if "Array" in sys.argv[2]:
-            info = ",".join(sys.argv[2].split(","))
-            logging.debug("info str: %s", info)
+        if "Array" in info:
             match = re.search(
                 r"Slurm ((?P<array_summary>Array Summary)|Array Task)"
                 r" Job_id=[0-9]+_([0-9]+|\*)"
@@ -806,8 +806,6 @@ def spool_mail_main():
                 die("Failed to parse Slurm info.")
             array_summary = match.group("array_summary") is not None
         else:
-            info = ",".join(sys.argv[2].split(",")[0:-1])
-            logging.debug("info str: %s", info)
             match = re.search(
                 r"Slurm"
                 r" Job_id=(?P<job_id>[0-9]+).*?(?P<state>(Began|Ended|Failed|Requeued|Invalid"  # noqa