def env(): env = os.environ.copy() env.update( dict( KUBECONFIG=env.get('KUBECONFIG', ''), PATH='{}:{}'.format( polyaxon_user_path(), env.get('PATH', ''), ), )) return env
def get_config_file_path(cls, create=True): if not cls.IS_GLOBAL: # local to this directory base_path = os.path.join(".") if cls.IS_POLYAXON_DIR: # Add it to the current "./.polyaxon" base_path = os.path.join(base_path, ".polyaxon") if create: cls._create_dir(base_path) else: base_path = polyaxon_user_path() if create: cls._create_dir(base_path) return os.path.join(base_path, cls.CONFIG_FILE_NAME)
import rhea from hestia.auth import AuthenticationTypes from hestia.user_path import polyaxon_user_path from marshmallow import EXCLUDE, RAISE from rhea import RheaError # noqa TMP_POLYAXON_PATH = '/tmp/.polyaxon/' TMP_AUTH_TOKEN_PATH = os.path.join(TMP_POLYAXON_PATH, '.authtoken') CONTEXT_AUTH_TOKEN_PATH = '/plx-context/.authtoken' TMP_CLIENT_CONFIG_PATH = os.path.join(TMP_POLYAXON_PATH, '.polyaxonclient') TMP_CONFIG_PATH = os.path.join(TMP_POLYAXON_PATH, '.polyaxonconfig') TMP_AUTH_PATH = os.path.join(TMP_POLYAXON_PATH, '.polyaxonauth') USER_CLIENT_CONFIG_PATH = os.path.join(polyaxon_user_path(), '.polyaxonclient') USER_CONFIG_PATH = os.path.join(polyaxon_user_path(), '.polyaxonconfig') USER_AUTH_PATH = os.path.join(polyaxon_user_path(), '.polyaxonauth') global_config = rhea.Rhea.read_configs([ rhea.ConfigSpec(TMP_CONFIG_PATH, config_type='.json', check_if_exists=False), rhea.ConfigSpec(USER_CONFIG_PATH, config_type='.json', check_if_exists=False), { 'dummy': 'dummy' } ]) auth_config = rhea.Rhea.read_configs([ rhea.ConfigSpec(TMP_AUTH_PATH, config_type='.json', check_if_exists=False),
def _get_config_volume(): return ['-v', '{}:{}'.format(polyaxon_user_path(), TMP_POLYAXON_PATH)]
import rhea from hestia.user_path import polyaxon_user_path from marshmallow import EXCLUDE, RAISE from polyaxon.containers.contexts import CONTEXT_MOUNT_AUTH from polyaxon.env_vars.keys import ( POLYAXON_KEYS_HASH_LENGTH, POLYAXON_KEYS_HEALTH_CHECK_URL, POLYAXON_KEYS_RECONCILE_URL, ) from polyaxon.schemas.api.authentication import AccessTokenConfig from polyaxon.schemas.cli.client_configuration import ClientConfig TMP_POLYAXON_PATH = "/tmp/.polyaxon/" USER_POLYAXON_PATH = polyaxon_user_path() TMP_AUTH_PATH = os.path.join(TMP_POLYAXON_PATH, ".polyaxonauth") TMP_CLIENT_CONFIG_PATH = os.path.join(TMP_POLYAXON_PATH, ".polyaxonclient") USER_AUTH_PATH = os.path.join(USER_POLYAXON_PATH, ".polyaxonauth") USER_CLIENT_CONFIG_PATH = os.path.join(USER_POLYAXON_PATH, ".polyaxonclient") auth_config = rhea.Rhea.read_configs( [ os.environ, rhea.ConfigSpec(TMP_AUTH_PATH, config_type=".json", check_if_exists=False), rhea.ConfigSpec(USER_AUTH_PATH, config_type=".json", check_if_exists=False), rhea.ConfigSpec(CONTEXT_MOUNT_AUTH, config_type=".json", check_if_exists=False), {"dummy": "dummy"}, ]
# -*- coding: utf-8 -*- from __future__ import absolute_import, division, print_function import os import rhea from hestia.auth import AuthenticationTypes from hestia.user_path import polyaxon_user_path TMP_AUTH_TOKEN_PATH = '/tmp/.polyaxon/.authtoken' CLIENT_CONFIG_PATH = os.path.join(polyaxon_user_path(), '.polyaxonclient') config = rhea.Rhea.read_configs([ rhea.ConfigSpec(CLIENT_CONFIG_PATH, config_type='.json', check_if_exists=False), os.environ, rhea.ConfigSpec(TMP_AUTH_TOKEN_PATH, config_type='.json', check_if_exists=False) ]) IN_CLUSTER = config.get_boolean('POLYAXON_IN_CLUSTER', is_optional=True, default=False) NO_OP = config.get_boolean('POLYAXON_NO_OP', is_optional=True, default=False) API_HOST = config.get_string('POLYAXON_API_HOST', is_optional=True) HTTP_PORT = config.get_int('POLYAXON_HTTP_PORT', is_optional=True) WS_PORT = config.get_int('POLYAXON_WS_PORT', is_optional=True)