def get_init_env_vars(self):
env_vars = get_internal_env_vars(service_internal_header=InternalServices.DOCKERIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
# Add containers env vars
env_vars += [
get_env_var(name='POLYAXON_CONTAINER_BUILD_STEPS', value=self.build_steps),
get_env_var(name='POLYAXON_CONTAINER_ENV_VARS', value=self.env_vars),
get_env_var(name='POLYAXON_MOUNT_PATHS_NVIDIA', value=conf.get(MOUNT_PATHS_NVIDIA)),
]
# Add security context if set
sid = conf.get(SECURITY_CONTEXT_USER)
gid = conf.get(SECURITY_CONTEXT_GROUP)
should_set_security_context = conf.get(BUILD_JOBS_SET_SECURITY_CONTEXT) and sid and gid
if should_set_security_context:
env_vars.append(get_env_var('POLYAXON_SECURITY_CONTEXT_USER', value=sid))
env_vars.append(get_env_var('POLYAXON_SECURITY_CONTEXT_GROUP', value=gid))
# Add set env lang
env_lang = self.lang_env or conf.get(BUILD_JOBS_LANG_ENV)
if env_lang:
env_vars.append(get_env_var(name='POLYAXON_LANG_ENV', value=env_lang))
return env_vars
def get_init_env_vars(self):
env_vars = get_internal_env_vars(
service_internal_header=InternalServices.INITIALIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
return env_vars
def get_env_vars(self):
env_vars = get_internal_env_vars(service_internal_header=InternalServices.DOCKERIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
if conf.get('REGISTRY_PASSWORD') and conf.get('REGISTRY_USER'):
env_vars += [
get_env_var(name='POLYAXON_REGISTRY_USER', value=conf.get('REGISTRY_USER')),
get_env_var(name='POLYAXON_REGISTRY_URI', value=conf.get('REGISTRY_LOCAL_URI')),
get_from_secret('POLYAXON_REGISTRY_PASSWORD',
'registry-password',
settings.POLYAXON_K8S_REGISTRY_SECRET_NAME),
]
# Add private registries secrets keys
for key in config.keys_startswith(settings.PRIVATE_REGISTRIES_PREFIX):
env_vars.append(get_from_secret(key, key))
# Add set env lang
get_env_var(name='POLYAXON_JOB_DOCKERIZER_SET_LANG_ENV',
value=conf.get('JOB_DOCKERIZER_SET_LANG_ENV'))
# Add security context if set
if conf.get('JOB_DOCKERIZER_SET_SECURITY_CONTEXT'):
get_env_var('POLYAXON_SECURITY_CONTEXT_USER', value=conf.get('SECURITY_CONTEXT_USER'))
get_env_var('POLYAXON_SECURITY_CONTEXT_GROUP', value=conf.get('SECURITY_CONTEXT_GROUP'))
return env_vars
def get_init_env_vars(self):
env_vars = get_internal_env_vars(service_internal_header=InternalServices.DOCKERIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
# Add containers env vars
env_vars += [
get_env_var(name='POLYAXON_CONTAINER_BUILD_STEPS', value=self.build_steps),
get_env_var(name='POLYAXON_CONTAINER_ENV_VARS', value=self.env_vars),
get_env_var(name='POLYAXON_MOUNT_PATHS_NVIDIA', value=conf.get('MOUNT_PATHS_NVIDIA')),
]
return env_vars
def get_sidecar_env_vars(namespace, resource_name, job_container_name,
internal_health_check_url):
env_vars = get_internal_env_vars(
namespace=namespace,
service_internal_header=InternalServices.SIDECAR,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
return env_vars + [
get_env_var(name='POLYAXON_POD_ID', value=resource_name),
get_env_var(name='POLYAXON_CONTAINER_ID', value=job_container_name),
get_env_var(name='POLYAXON_INTERNAL_HEALTH_CHECK_URL',
value=internal_health_check_url),
]
def get_sidecar_env_vars(namespace,
job_container_name,
internal_health_check_url,
internal_reconcile_url):
env_vars = get_internal_env_vars(namespace=namespace,
service_internal_header=InternalServices.SIDECAR,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
return env_vars + [
get_env_var(name='POLYAXON_CONTAINER_ID', value=job_container_name),
get_env_var(name='POLYAXON_INTERNAL_HEALTH_CHECK_URL', value=internal_health_check_url),
get_env_var(name='POLYAXON_INTERNAL_RECONCILE_URL', value=internal_reconcile_url),
get_from_field_ref(name='POLYAXON_POD_ID', field_path='metadata.name'),
]
def get_env_vars(self):
env_vars = get_internal_env_vars(
service_internal_header=InternalServices.DOCKERIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN,
include_internal_token=True)
# Add set env lang
get_env_var(name='POLYAXON_LANG_ENV',
value=conf.get(BUILD_JOBS_LANG_ENV))
# Add security context if set
if conf.get(BUILD_JOBS_SET_SECURITY_CONTEXT):
get_env_var('POLYAXON_SECURITY_CONTEXT_USER',
value=conf.get(SECURITY_CONTEXT_USER))
get_env_var('POLYAXON_SECURITY_CONTEXT_GROUP',
value=conf.get(SECURITY_CONTEXT_GROUP))
return env_vars
def get_env_vars(self):
env_vars = get_internal_env_vars(
service_internal_header=InternalServices.DOCKERIZER,
namespace=self.namespace,
authentication_type=AuthenticationTypes.INTERNAL_TOKEN)
if conf.get('REGISTRY_PASSWORD') and conf.get('REGISTRY_USER'):
env_vars += [
get_env_var(name='POLYAXON_REGISTRY_USER',
value=conf.get('REGISTRY_USER')),
get_env_var(name='POLYAXON_REGISTRY_URI',
value=conf.get('REGISTRY_LOCAL_URI')),
get_from_secret('POLYAXON_REGISTRY_PASSWORD',
'registry-password',
settings.POLYAXON_K8S_REGISTRY_SECRET_NAME),
]
# Add private registries secrets keys
for key in config.keys_startswith(settings.PRIVATE_REGISTRIES_PREFIX):
env_vars.append(get_from_secret(key, key))
return env_vars
