-
Notifications
You must be signed in to change notification settings - Fork 9
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
added collector to collect farm env variables
- Loading branch information
Showing
1 changed file
with
30 additions
and
0 deletions.
There are no files selected for viewing
30 changes: 30 additions & 0 deletions
30
client/ayon_ftrack/plugins/publish/collect_farm_env_variables.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,30 @@ | ||
import os | ||
|
||
import pyblish.api | ||
|
||
try: | ||
from ayon_core.pipeline.publish import FARM_JOB_ENV_DATA_KEY | ||
except ImportError: | ||
# NOTE Can be removed when ayon-core >= 1.0.10 is required in package.py | ||
FARM_JOB_ENV_DATA_KEY = "farmJobEnv" | ||
|
||
|
||
class CollectFtrackJobEnvVars(pyblish.api.ContextPlugin): | ||
"""Collect set of environment variables to submit with deadline jobs""" | ||
order = pyblish.api.CollectorOrder - 0.45 | ||
label = "Collect ftrack farm environment variables" | ||
targets = ["local"] | ||
|
||
def process(self, context): | ||
env = context.data.setdefault(FARM_JOB_ENV_DATA_KEY, {}) | ||
|
||
# Disable colored logs on farm | ||
for key in [ | ||
"FTRACK_SERVER", | ||
"FTRACK_API_USER", | ||
"FTRACK_API_KEY", | ||
]: | ||
value = os.getenv(key) | ||
if value: | ||
self.log.debug(f"Setting job env: {key}: {value}") | ||
env[key] = value |