Example #1
0
def get_sidecar_container(job_name,
                          job_container_name,
                          sidecar_container_name,
                          sidecar_docker_image,
                          sidecar_docker_image_pull_policy,
                          namespace,
                          sidecar_config,
                          sidecar_args,
                          internal_health_check_url,
                          env_vars=None):
    """Return a pod sidecar container."""
    env_vars = to_list(env_vars) if env_vars else []
    env_vars += get_sidecar_env_vars(
        job_name=job_name,
        job_container_name=job_container_name,
        internal_health_check_url=internal_health_check_url)
    env_vars += get_service_env_vars(namespace=namespace)
    for k, v in sidecar_config.items():
        env_vars.append(get_env_var(name=k, value=v))
    return client.V1Container(
        name=sidecar_container_name,
        image=sidecar_docker_image,
        image_pull_policy=sidecar_docker_image_pull_policy,
        command=get_sidecar_command(),
        env=env_vars,
        args=sidecar_args)
    def get_env_vars(self):
        env_vars = get_service_env_vars(namespace=self.namespace)
        for k, v in config.get_requested_params(to_str=True).items():
            env_vars.append(get_env_var(name=k, value=v))

        # Add private registries secrets keys
        for key in config.params_startswith(settings.PRIVATE_REGISTRIES_PREFIX):
            env_vars.append(get_from_secret(key, key))

        return env_vars
Example #3
0
 def test_get_service_env_vars(self):
     env_vars = get_service_env_vars()
     assert len(env_vars) == 6
     env_var_names = [env_var.name for env_var in env_vars]
     assert 'POLYAXON_K8S_NAMESPACE' in env_var_names
     assert 'POLYAXON_SECRET_KEY' in env_var_names
     assert 'POLYAXON_INTERNAL_SECRET_TOKEN' in env_var_names
     assert 'POLYAXON_RABBITMQ_PASSWORD' in env_var_names
     assert 'POLYAXON_DB_PASSWORD' in env_var_names
     assert API_KEY_NAME in env_var_names
Example #4
0
 def get_sidecar_container(self):
     """Pod sidecar container for task logs."""
     env_vars = get_sidecar_env_vars(
         job_name=self.job_name, job_container_name=self.job_container_name)
     env_vars += get_service_env_vars(namespace=self.namespace)
     for k, v in self.sidecar_config.items():
         env_vars.append(client.V1EnvVar(name=k, value=v))
     return client.V1Container(
         name=self.sidecar_container_name,
         image=self.sidecar_docker_image,
         command=get_sidecar_command(app_label=self.app_label),
         env=env_vars,
         args=get_sidecar_args(pod_id=self.pod_name))
Example #5
0
    def get_env_vars(self):
        env_vars = get_service_env_vars(namespace=self.namespace)
        for k, v in config.get_requested_data(to_str=True).items():
            env_vars.append(get_env_var(name=k, value=v))

        # Add private registries secrets keys
        for key in config.keys_startswith(settings.PRIVATE_REGISTRIES_PREFIX):
            env_vars.append(get_from_secret(key, key))

        # Add repos access token secret key
        if settings.REPOS_ACCESS_TOKEN:
            env_vars.append(get_from_secret(settings.REPOS_ACCESS_TOKEN_KEY,
                                            settings.REPOS_ACCESS_TOKEN_KEY))

        return env_vars
Example #6
0
def get_sidecar_container(job_name,
                          job_container_name,
                          sidecar_container_name,
                          sidecar_docker_image,
                          namespace,
                          app_label,
                          sidecar_config,
                          sidecar_args,
                          env_vars=None):
    """Return a pod sidecar container."""
    env_vars = to_list(env_vars) if env_vars else []
    env_vars += get_sidecar_env_vars(job_name=job_name, job_container_name=job_container_name)
    env_vars += get_service_env_vars(namespace=namespace)
    for k, v in sidecar_config.items():
        env_vars.append(get_env_var(name=k, value=v))
    return client.V1Container(name=sidecar_container_name,
                              image=sidecar_docker_image,
                              command=get_sidecar_command(app_label=app_label),
                              env=env_vars,
                              args=sidecar_args)
Example #7
0
def get_sidecar_container(job_name,
                          job_container_name,
                          sidecar_container_name,
                          sidecar_docker_image,
                          namespace,
                          app_label,
                          sidecar_config,
                          sidecar_args,
                          env_vars=None):
    """Return a pod sidecar container."""
    env_vars = to_list(env_vars) if env_vars else []
    env_vars += get_sidecar_env_vars(job_name=job_name,
                                     job_container_name=job_container_name)
    env_vars += get_service_env_vars(namespace=namespace)
    for k, v in sidecar_config.items():
        env_vars.append(get_env_var(name=k, value=v))
    return client.V1Container(name=sidecar_container_name,
                              image=sidecar_docker_image,
                              command=get_sidecar_command(app_label=app_label),
                              env=env_vars,
                              args=sidecar_args)
Example #8
0
    def get_env_vars(self):
        env_vars = get_service_env_vars(namespace=self.namespace)
        for k, v in config.get_requested_params(to_str=True).items():
            env_vars.append(get_env_var(name=k, value=v))

        return env_vars
Example #9
0
    def get_env_vars(self):
        env_vars = get_service_env_vars(namespace=self.namespace)
        for k, v in config.get_requested_data(to_str=True).items():
            env_vars.append(get_env_var(name=k, value=v))

        return env_vars