From 96a9ed8e73f040cb8a1855f4798ebe561262ebed Mon Sep 17 00:00:00 2001
From: Alex Wiens <alex.wiens@uni-paderborn.de>
Date: Fri, 30 May 2025 14:59:21 +0200
Subject: [PATCH] prule: Fix SLURM info parsing, Add new field TresPerTask

---
 prule/__init__.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/prule/__init__.py b/prule/__init__.py
index ea2b6c7..09ad65b 100644
--- a/prule/__init__.py
+++ b/prule/__init__.py
@@ -385,9 +385,15 @@ def parse_slurminfo(info: str) -> dict:
                 slurm["Reservation"] = l[15:].strip("\n")
             elif l.startswith("   Comment="):
                 slurm["Comment"] = l[11:].strip("\n")
+            elif l.startswith("   TresPerTask="):
+                slurm["TresPerTask"] = l[15:].strip("\n")
             else:
                 for t in l.strip().split(" "):
-                    k,v = t.split("=")
+                    v_ix = t.find("=")
+                    if t.find("=", v_ix+1) != -1:
+                        log.print_color(log.color.yellow, log.warn, "Unknown field during SLURM info parsing: {}".format(l))
+                    k = t[:v_ix]
+                    v = t[v_ix+1:]
                     slurm[k] = v
         elif l.startswith("JobId="): # first line
             # first line
-- 
GitLab