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)