def is_wasb(self) -> bool:
return V1ConnectionKind.is_wasb(self.kind)
def is_volume_claim(self) -> bool:
return V1ConnectionKind.is_volume_claim(self.kind)
def is_bucket(self) -> bool:
return V1ConnectionKind.is_bucket(self.kind)
def is_host_path(self) -> bool:
return V1ConnectionKind.is_host_path(self.kind)
def is_artifact(self) -> bool:
return V1ConnectionKind.is_artifact(self.kind)
def is_mount(self) -> bool:
return V1ConnectionKind.is_mount(self.kind)
def handle_init_connections(
self,
polyaxon_init: V1PolyaxonInitContainer,
artifacts_store: V1ConnectionType,
init_connections: List[V1Init],
connection_by_names: Dict[str, V1ConnectionType],
contexts: PluginsContextsSpec,
) -> List[k8s_schemas.V1Container]:
containers = []
external_host = contexts.external_host if contexts else False
# Prepare connections that Polyaxon can init automatically
for init_connection in init_connections:
if init_connection.connection:
connection_spec = connection_by_names.get(
init_connection.connection)
# Handling ssh with git
if (V1ConnectionKind.is_ssh(connection_spec.kind)
and init_connection.git):
connection_spec.schema.patch_git(init_connection.git)
containers.append(
get_git_init_container(
polyaxon_init=polyaxon_init,
connection=connection_spec,
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
track=True,
))
elif V1ConnectionKind.is_git(connection_spec.kind):
if init_connection.git: # Update the default schema
connection_spec.schema.patch(init_connection.git)
containers.append(
get_git_init_container(
polyaxon_init=polyaxon_init,
connection=connection_spec,
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
track=True,
))
elif V1ConnectionKind.is_artifact(connection_spec.kind):
containers.append(
get_store_container(
polyaxon_init=polyaxon_init,
connection=connection_spec,
artifacts=init_connection.artifacts,
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
is_default_artifacts_store=artifacts_store and
init_connection.connection == artifacts_store.name,
))
else:
containers.append(
get_custom_init_container(
connection=connection_spec,
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
))
else:
# artifacts init without connection should default to the artifactsStore
if init_connection.artifacts:
containers.append(
get_store_container(
polyaxon_init=polyaxon_init,
connection=artifacts_store,
artifacts=init_connection.artifacts,
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
is_default_artifacts_store=True,
))
# git init without connection
if init_connection.git:
git_name = init_connection.git.get_name()
containers.append(
get_git_init_container(
polyaxon_init=polyaxon_init,
connection=V1ConnectionType(
name=git_name,
kind=V1ConnectionKind.GIT,
schema=init_connection.git,
secret=None,
),
container=init_connection.container,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
track=False,
))
# Dockerfile initialization
if init_connection.dockerfile:
containers.append(
get_dockerfile_init_container(
polyaxon_init=polyaxon_init,
dockerfile_args=init_connection.dockerfile,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
run_path=self.run_path,
run_instance=self.run_instance,
))
# File initialization
if init_connection.file:
containers.append(
get_file_init_container(
polyaxon_init=polyaxon_init,
file_args=init_connection.file,
env=self.get_init_service_env_vars(
external_host=external_host),
mount_path=init_connection.path,
contexts=contexts,
run_path=self.run_path,
run_instance=self.run_instance,
))
return containers
def handle_init_connections(
self,
polyaxon_init: V1PolyaxonInitContainer,
init_connections: List[V1Init],
connection_by_names: Dict[str, V1ConnectionType],
contexts: PluginsContextsSpec,
) -> List[k8s_schemas.V1Container]:
containers = []
# Prepare connections that Polyaxon can inti automatically
for init_connection in init_connections:
if init_connection.connection:
connection_spec = connection_by_names.get(
init_connection.connection)
if init_connection.git: # Update the default schema
connection_spec.schema.patch(init_connection.git)
if connection_spec.kind == V1ConnectionKind.GIT:
containers.append(
get_git_init_container(
polyaxon_init=polyaxon_init,
connection=connection_spec,
container=init_connection.container,
env=self.get_init_service_env_vars(),
mount_path=init_connection.path,
contexts=contexts,
track=True,
))
if V1ConnectionKind.is_artifact(connection_spec.kind):
containers.append(
get_store_container(
polyaxon_init=polyaxon_init,
connection=connection_spec,
artifacts=init_connection.artifacts,
container=init_connection.container,
env=self.get_init_service_env_vars(),
mount_path=init_connection.path,
))
else:
# git init without connection
if init_connection.git:
git_name = init_connection.git.get_name()
containers.append(
get_git_init_container(
polyaxon_init=polyaxon_init,
connection=V1ConnectionType(
name=git_name,
kind=V1ConnectionKind.GIT,
schema=init_connection.git,
secret=None,
),
container=init_connection.container,
env=self.get_init_service_env_vars(),
mount_path=init_connection.path,
contexts=contexts,
track=False,
))
# Dockerfile initialization
if init_connection.dockerfile:
containers.append(
get_dockerfile_init_container(
polyaxon_init=polyaxon_init,
dockerfile_args=init_connection.dockerfile,
env=self.get_init_service_env_vars(),
mount_path=init_connection.path,
contexts=contexts,
run_path=self.run_path,
))
return containers
