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)
