From 9ce9a866bb1b4f58062d758e0abc316f9e84a5c7 Mon Sep 17 00:00:00 2001 From: Simon Mazenoux Date: Wed, 6 Sep 2023 11:11:16 +0200 Subject: [PATCH] feat: make TaskQueueDB VO aware for diracx transition --- src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py b/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py index dd5ea79be74..3f437c8d241 100755 --- a/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py +++ b/src/DIRAC/WorkloadManagementSystem/DB/TaskQueueDB.py @@ -97,6 +97,7 @@ def __initializeDB(self): "Owner": "VARCHAR(255) NOT NULL", "OwnerDN": "VARCHAR(255)", "OwnerGroup": "VARCHAR(32) NOT NULL", + "VO": "VARCHAR(32) NOT NULL", "CPUTime": "BIGINT(20) UNSIGNED NOT NULL", "Priority": "FLOAT NOT NULL", "Enabled": "TINYINT(1) NOT NULL DEFAULT 0", @@ -251,6 +252,8 @@ def __createTaskQueue(self, tqDefDict, priority=1, connObj=False): for field in singleValueDefFields: sqlSingleFields.append(field) sqlValues.append(tqDefDict[field]) + sqlSingleFields.append("VO") + sqlValues.append(Registry.getVOForGroup(tqDefDict["OwnerGroup"])) # Insert the TQ Disabled sqlSingleFields.append("Enabled") sqlValues.append("0")