sweep: DIRACGrid#7266 Preinstalled env pilot options in SiteDirector

fstagni · Nov 6, 2023 · 790f7c5 · 790f7c5
1 parent e87844d
commit 790f7c5
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/src/DIRAC/WorkloadManagementSystem/Agent/SiteDirector.py b/src/DIRAC/WorkloadManagementSystem/Agent/SiteDirector.py
@@ -1022,6 +1022,18 @@ def _getPilotOptions(self, queue, **kwargs):
         else:
             self.log.info("DIRAC project will be installed by pilots")
 
+        # Preinstalled environment defined ?
+        preinstalledEnv = opsHelper.getValue("Pilot/PreinstalledEnv", "")
+        preinstalledEnvPrefix = opsHelper.getValue("Pilot/PreinstalledEnvPrefix", "")
+        if preinstalledEnv:
+            pilotOptions.append(f"--preinstalledEnv={preinstalledEnv}")
+        elif preinstalledEnvPrefix:
+            pilotOptions.append(f"--preinstalledEnvPrefix={preinstalledEnvPrefix}")
+
+        # Pilot Logging defined?
+        if opsHelper.getValue("/Services/JobMonitoring/usePilotsLoggingFlag", False):
+            pilotOptions.append("-z ")
+
         pilotOptions.append("--pythonVersion=3")
 
         # DIRAC Extensions to be used in pilots