def deploy(file, check, upgrade): # pylint:disable=redefined-builtin
"""Deploy polyaxon."""
config = read_deployment_config(file)
manager = DeployManager(config=config, filepath=file)
exception = None
if check:
manager.check()
Printer.print_success('Polyaxon deployment file is valid.',
add_sign=True)
elif upgrade:
try:
manager.upgrade()
except Exception as e:
Printer.print_error('Polyaxon could not upgrade the deployment.',
add_sign=True)
exception = e
else:
try:
manager.install()
except Exception as e:
Printer.print_error('Polyaxon could not be installed.',
add_sign=True)
exception = e
if exception:
Printer.print_error('Error message `{}`.'.format(exception))
def test_deployment_type(self):
manager = DeployManager(config=DeploymentConfig.from_dict(
{'deploymentType': DeploymentTypes.DOCKER_COMPOSE}))
assert manager.deployment_type == DeploymentTypes.DOCKER_COMPOSE
assert manager.is_docker_compose is True
manager = DeployManager(config=DeploymentConfig.from_dict(
{'deploymentType': DeploymentTypes.KUBERNETES}))
assert manager.deployment_type == DeploymentTypes.KUBERNETES
assert manager.is_kubernetes is True
manager = DeployManager(config=DeploymentConfig.from_dict(
{'deploymentType': DeploymentTypes.HEROKU}))
assert manager.deployment_type == DeploymentTypes.HEROKU
assert manager.is_heroku is True
manager = DeployManager(config=DeploymentConfig.from_dict(
{'deploymentType': DeploymentTypes.DOCKER}))
assert manager.deployment_type == DeploymentTypes.DOCKER
assert manager.is_docker is True
def upgrade(file, manager_path, check, dry_run): # pylint:disable=redefined-builtin
"""Upgrade a Polyaxon deployment."""
config = read_deployment_config(file)
manager = DeployManager(config=config,
filepath=file,
manager_path=manager_path,
dry_run=dry_run)
exception = None
if check:
try:
manager.check()
except Exception as e:
Printer.print_error("Polyaxon deployment file is not valid ")
Printer.print_error('Error message `{}`.'.format(e))
sys.exit(1)
Printer.print_success('Polyaxon deployment file is valid.')
else:
try:
manager.upgrade()
except Exception as e:
Printer.print_error('Polyaxon could not upgrade the deployment.')
exception = e
if exception:
Printer.print_error('Error message `{}`.'.format(exception))
def teardown(file): # pylint:disable=redefined-builtin
"""Teardown a polyaxon deployment given a config file."""
config = read_deployment_config(file)
manager = DeployManager(config=config, filepath=file)
exception = None
try:
if click.confirm('Would you like to execute pre-delete hooks?', default=True):
manager.teardown(hooks=True)
else:
manager.teardown(hooks=False)
except Exception as e:
Printer.print_error('Polyaxon could not teardown the deployment.')
exception = e
if exception:
Printer.print_error('Error message `{}`.'.format(exception))
def deploy(file, manager_path, check, dry_run): # pylint:disable=redefined-builtin
"""Deploy polyaxon."""
config = read_deployment_config(file)
manager = DeployManager(config=config,
filepath=file,
manager_path=manager_path,
dry_run=dry_run)
exception = None
if check:
manager.check()
Printer.print_success('Polyaxon deployment file is valid.')
else:
try:
manager.install()
except Exception as e:
Printer.print_error('Polyaxon could not be installed.')
exception = e
if exception:
Printer.print_error('Error message `{}`.'.format(exception))
def test_default_props(self):
manager = DeployManager()
assert manager.deployment_type == DeploymentTypes.KUBERNETES
assert manager.is_kubernetes is True
assert isinstance(manager.helm, HelmOperator)
assert isinstance(manager.kubectl, KubectlOperator)