def __init__(self,
config=None,
filepath=None,
manager_path=None,
dry_run=False):
self.config = config
self.filepath = filepath
self.manager_path = manager_path
self.dry_run = dry_run
self.kubectl = KubectlOperator()
self.helm = HelmOperator()
self.docker = DockerOperator()
self.compose = ComposeOperator()
class TestDockerOperator(BaseTestCase):
def setUp(self):
super().setUp()
self.docker = DockerOperator()
@staticmethod
def mock_popen(return_code, out_msg, err_msg=None):
def popen(*args, **kwargs):
stdout = kwargs.pop("stdout")
stdout.write(out_msg)
if err_msg:
stderr = kwargs.pop("stderr")
stderr.write(err_msg)
return mock.Mock(wait=mock.Mock(return_value=return_code))
return mock.Mock(side_effect=popen)
@mock.patch("polyaxon.deploy.operators.cmd_operator.subprocess")
def test_docker(self, mock_subprocess):
mock_subprocess.Popen = self.mock_popen(0, "bar")
assert self.docker.execute(["volume"]) == "bar"
assert mock_subprocess.Popen.call_args[0][0] == ["docker", "volume"]
@mock.patch("polyaxon.deploy.operators.cmd_operator.subprocess")
def test_docker_set_volume(self, mock_subprocess):
mock_subprocess.Popen = self.mock_popen(0, "bar")
assert self.docker.set_volume("foo") == "bar"
assert mock_subprocess.Popen.call_args[0][0] == [
"docker",
"volume",
"create",
"--name=foo",
]
@mock.patch("polyaxon.deploy.operators.cmd_operator.subprocess")
def test_docker_error(self, mock_subprocess):
return_code = 1
stdout = "output"
stderr = "error"
mock_subprocess.Popen = self.mock_popen(return_code, stdout, stderr)
with self.assertRaises(PolyaxonOperatorException) as exception:
self.docker.execute(["run"])
self.assertEqual(
exception.exception.message,
"`docker` command ('docker', 'run') "
"failed with exit status {}\nstdout:\n{}\nstderr:\n{}".format(
return_code, stdout, stderr),
)
def _run(ctx, name, owner, project_name, description, tags, specification,
log):
docker = DockerOperator()
if not docker.check():
raise PolyaxonException("Docker is required to run this command.")
# Create Build
project = "{}.{}".format(owner, project_name)
build_job = Run(project=project)
specification = CompiledOperationSpecification.apply_operation_contexts(
specification)
content = specification.to_dict(dump=True)
build_job.create(name=name,
description=description,
tags=tags,
content=content)
image = _create_docker_build(build_job, specification, project)
experiment = Run(project=project)
experiment.create(name=name,
tags=tags,
description=description,
content=content)
cmd_args = ["run", "--rm"]
data_paths, bind_mounts = _get_data_bind_mounts(specification.data_refs)
for key, value in _get_env_vars(
project=project,
experiment_id=experiment.experiment_id,
params=specification.params,
data_paths=data_paths,
):
cmd_args += ["-e", "{key}={value}".format(key=key, value=value)]
cmd_args += _get_config_volume()
cmd_args += _get_data_volumes(bind_mounts)
cmd_args += [image]
# Add cmd.run
_, args = specification.container.get_container_command_args()
for arg in args:
cmd_args += arg
try:
print(cmd_args)
docker.execute(cmd_args, stream=True)
except Exception as e:
handle_cli_error(e, message="Could start local run.")
sys.exit(1)
def setUp(self):
super().setUp()
self.docker = DockerOperator()
class DeployManager(object):
def __init__(
self,
config: DeploymentConfig = None,
filepath=None,
manager_path=None,
dry_run=False,
):
self.config = config
self.filepath = filepath
self.manager_path = manager_path
self.dry_run = dry_run
self.kubectl = KubectlOperator()
self.helm = HelmOperator()
self.docker = DockerOperator()
self.compose = ComposeOperator()
@property
def deployment_type(self):
if self.config and self.config.deployment_type:
return self.config.deployment_type
return DeploymentTypes.KUBERNETES
@property
def deployment_version(self):
if self.config and self.config.deployment_version:
return self.config.deployment_version
return None
@property
def deployment_namespace(self):
if self.config and self.config.namespace:
return self.config.namespace
return "polyaxon"
@property
def k8s_chart(self):
deployment_chart = DeploymentCharts.PLATFORM
if self.config and self.config.deployment_chart:
deployment_chart = self.config.deployment_chart
if deployment_chart == DeploymentCharts.PLATFORM:
return "polyaxon/polyaxon"
else:
return "polyaxon/agent"
@property
def k8s_name(self):
deployment_chart = DeploymentCharts.PLATFORM
if self.config and self.config.deployment_chart:
deployment_chart = self.config.deployment_chart
if deployment_chart == DeploymentCharts.PLATFORM:
return "polyaxon"
else:
return "agent"
@property
def is_kubernetes(self):
return self.deployment_type in {
DeploymentTypes.KUBERNETES,
DeploymentTypes.MINIKUBE,
DeploymentTypes.MICRO_K8S,
}
@property
def is_docker_compose(self):
return self.deployment_type == DeploymentTypes.DOCKER_COMPOSE
@property
def is_docker(self):
return self.deployment_type == DeploymentTypes.DOCKER
@property
def is_heroku(self):
return self.deployment_type == DeploymentTypes.HEROKU
@property
def is_valid(self):
return self.deployment_type in DeploymentTypes.VALUES
def check_for_kubernetes(self):
# Deployment on k8s requires helm & kubectl to be installed
if not self.kubectl.check():
raise PolyaxonException("kubectl is required to run this command.")
Printer.print_success("kubectl is installed")
if not self.helm.check():
raise PolyaxonException("helm is required to run this command.")
Printer.print_success("helm is installed")
# Check that polyaxon/polyaxon is set and up-to date
self.helm.execute(
args=["repo", "add", "polyaxon", "https://charts.polyaxon.com"])
self.helm.execute(args=["repo", "update"])
return True
def check_for_docker_compose(self):
# Deployment on docker compose requires Docker & Docker Compose to be installed
if not self.docker.check():
raise PolyaxonException("Docker is required to run this command.")
Printer.print_success("Docker is installed")
if not self.compose.check():
raise PolyaxonException(
"Docker Compose is required to run this command.")
Printer.print_success("Docker Compose is installed")
# Check that .polyaxon/.compose is set and up-to date
if ComposeConfigManager.is_initialized():
Printer.print_success("Docker Compose deployment is initialised.")
return True
def check_for_docker(self):
if not self.docker.check():
raise PolyaxonException("Docker is required to run this command.")
return True
def check_for_heroku(self):
return True
def nvidia_device_plugin(self):
return "https://github.com/NVIDIA/k8s-device-plugin/blob/v1.10/nvidia-device-plugin.yml"
def check(self):
"""Add platform specific checks"""
if not self.is_valid:
raise PolyaxonException(
"Deployment type `{}` not supported".format(
self.deployment_type))
check = False
if self.is_kubernetes:
check = self.check_for_kubernetes()
elif self.is_docker_compose:
check = self.check_for_docker_compose()
elif self.is_docker:
check = self.check_for_docker()
elif self.is_heroku:
check = self.check_for_heroku()
if not check:
raise PolyaxonException("Deployment `{}` is not valid".format(
self.deployment_type))
def _get_or_create_namespace(self):
click.echo("Checking {} namespace ...".format(
self.deployment_namespace))
try:
stdout = self.kubectl.execute(
args=["get", "namespace", self.deployment_namespace],
is_json=True,
stream=settings.CLIENT_CONFIG.debug,
)
except PolyaxonOperatorException:
stdout = None
if stdout:
return
# Create a namespace
click.echo("Creating {} namespace ...".format(
self.deployment_namespace))
stdout = self.kubectl.execute(
args=["create", "namespace", self.deployment_namespace],
is_json=False,
stream=settings.CLIENT_CONFIG.debug,
)
click.echo(stdout)
def install_on_kubernetes(self):
self._get_or_create_namespace()
args = ["install", self.k8s_name]
if self.manager_path:
args += [self.manager_path]
else:
args += [self.k8s_chart]
args += ["--namespace={}".format(self.deployment_namespace)]
if self.filepath:
args += ["-f", self.filepath]
if self.deployment_version:
args += ["--version", self.deployment_version]
if self.dry_run:
args += ["--debug", "--dry-run"]
click.echo("Running install command ...")
stdout = self.helm.execute(args=args,
stream=settings.CLIENT_CONFIG.debug)
click.echo(stdout)
Printer.print_success("Deployment finished.")
def install_on_docker_compose(self):
path = ComposeConfigManager.get_config_filepath()
path = "/".join(path.split("/")[:-1])
# Fetch docker-compose
Transport().download(
url=
"https://github.com/polyaxon/polyaxon-compose/archive/master.tar.gz",
filename=path + "/file",
untar=True,
delete_tar=True,
extract_path=path,
)
# Move necessary info
shutil.copy(
path + "/polyaxon-compose-master/docker-compose.yml",
path + "/docker-compose.yml",
)
shutil.copy(path + "/polyaxon-compose-master/components.env",
path + "/components.env")
shutil.copy(path + "/polyaxon-compose-master/base.env",
path + "/base.env")
shutil.rmtree(path + "/polyaxon-compose-master/")
# Generate env from config
ComposeConfigManager.set_config(self.compose.generate_env(self.config))
Printer.print_success("Docker Compose deployment is initialised.")
if self.dry_run:
Printer.print_success("Polyaxon generated deployment env.")
return
self.docker.execute(["volume", "create", "--name=polyaxon-postgres"])
Printer.print_success("Docker volume created.")
self.compose.execute(["-f", path + "/docker-compose.yml", "up", "-d"])
Printer.print_success("Deployment is running in the background.")
Printer.print_success("You can configure your CLI by running: "
"polyaxon config set --host=localhost.")
def install_on_docker(self):
pass
def install_on_heroku(self):
pass
def install(self):
"""Install polyaxon using the current config to the correct platform."""
if not self.is_valid:
raise PolyaxonException(
"Deployment type `{}` not supported".format(
self.deployment_type))
if self.is_kubernetes:
self.install_on_kubernetes()
elif self.is_docker_compose:
self.install_on_docker_compose()
elif self.is_docker:
self.install_on_docker()
elif self.is_heroku:
self.install_on_heroku()
def upgrade_on_kubernetes(self):
args = ["upgrade", self.k8s_name]
if self.manager_path:
args += [self.manager_path]
else:
args += [self.k8s_chart]
if self.filepath:
args += ["-f", self.filepath]
if self.deployment_version:
args += ["--version", self.deployment_version]
args += ["--namespace={}".format(self.deployment_namespace)]
if self.dry_run:
args += ["--debug", "--dry-run"]
click.echo("Running upgrade command ...")
stdout = self.helm.execute(args=args,
stream=settings.CLIENT_CONFIG.debug)
click.echo(stdout)
Printer.print_success("Deployment upgraded.")
def upgrade_on_docker_compose(self):
self.install_on_docker_compose()
def upgrade_on_docker(self):
pass
def upgrade_on_heroku(self):
pass
def upgrade(self):
"""Upgrade deployment."""
if not self.is_valid:
raise PolyaxonException(
"Deployment type `{}` not supported".format(
self.deployment_type))
if self.is_kubernetes:
self.upgrade_on_kubernetes()
elif self.is_docker_compose:
self.upgrade_on_docker_compose()
elif self.is_docker:
self.upgrade_on_docker()
elif self.is_heroku:
self.upgrade_on_heroku()
def teardown_on_kubernetes(self, hooks):
args = ["delete", self.k8s_name]
if not hooks:
args += ["--no-hooks"]
args += ["--namespace={}".format(self.deployment_namespace)]
click.echo("Running teardown command ...")
self.helm.execute(args=args)
Printer.print_success("Deployment successfully deleted.")
def teardown_on_docker_compose(self):
path = ComposeConfigManager.get_config_filepath()
path = "/".join(path.split("/")[:-1])
self.compose.execute(["-f", path + "/docker-compose.yml", "down"])
def teardown_on_docker(self, hooks):
pass
def teardown_on_heroku(self, hooks):
pass
def teardown(self, hooks=True):
"""Teardown Polyaxon."""
if not self.is_valid:
raise PolyaxonException(
"Deployment type `{}` not supported".format(
self.deployment_type))
if self.is_kubernetes:
self.teardown_on_kubernetes(hooks=hooks)
elif self.is_docker_compose:
self.teardown_on_docker_compose()
elif self.is_docker:
self.teardown_on_docker(hooks=hooks)
elif self.is_heroku:
self.teardown_on_heroku(hooks=hooks)
def setUp(self):
self.docker = DockerOperator()