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