def delete(ctx, _project):
"""Delete project.
Uses /docs/core/cli/#caching
"""
owner, project_name = get_project_or_local(_project
or ctx.obj.get("project"),
is_cli=True)
if not click.confirm("Are sure you want to delete project `{}/{}`".format(
owner, project_name)):
click.echo("Existing without deleting project.")
sys.exit(1)
try:
polyaxon_client = ProjectClient(owner=owner, project=project_name)
polyaxon_client.delete()
local_project = ProjectConfigManager.get_config()
if local_project and (owner, project_name) == (
local_project.user,
local_project.name,
):
# Purge caching
ProjectConfigManager.purge()
except (ApiException, HTTPError) as e:
handle_cli_error(e,
message="Could not delete project `{}/{}`.".format(
owner, project_name))
sys.exit(1)
Printer.print_success("Project `{}/{}` was delete successfully".format(
owner, project_name))
def get_project_or_local(project=None, is_cli: bool = False):
from polyaxon import settings
if not project and not ProjectConfigManager.is_initialized():
if is_cli:
Printer.print_error("Please provide a valid project.")
sys.exit(1)
else:
raise PolyaxonClientException("Please provide a valid project.")
if project:
owner, project_name = get_project_info(project)
else:
project = ProjectConfigManager.get_config()
owner, project_name = project.owner, project.name
if not owner and (not settings.CLI_CONFIG or settings.CLI_CONFIG.is_ce):
owner = DEFAULT
if not all([owner, project_name]):
if is_cli:
Printer.print_error("Please provide a valid project.")
sys.exit(1)
else:
raise PolyaxonClientException("Please provide a valid project.")
return owner, project_name
def purge(cache_only):
"""Purge the global config values."""
if not cache_only:
ClientConfigManager.purge()
CliConfigManager.purge()
AuthConfigManager.purge()
UserConfigManager.purge()
ProjectConfigManager.purge()
RunConfigManager.purge()
Printer.print_success("Configs was removed.")
def cache(config_manager, config, owner=None, project=None):
if config_manager == ProjectConfigManager:
_cache_project(config=config, project=project, owner=owner)
# Set caching only if we have an initialized project
if not ProjectConfigManager.is_initialized():
return
if not _is_same_project(owner, project):
return
visibility = (ProjectConfigManager.VISIBILITY_LOCAL
if ProjectConfigManager.is_locally_initialized() else
ProjectConfigManager.VISIBILITY_GLOBAL)
config_manager.set_config(config, visibility=visibility)
def get_project_or_local(project=None, is_cli: bool = False):
from polyaxon import settings
if not project and not ProjectConfigManager.is_initialized():
error_message = "Please provide a valid project or initialize a project in the current path."
if is_cli:
Printer.print_error(error_message)
sys.exit(1)
else:
raise PolyaxonClientException(error_message)
if project:
owner, project_name = get_entity_info(project)
else:
project = get_local_project()
owner, project_name = project.owner, project.name
if not owner:
owner = get_local_owner(is_cli=is_cli)
if not owner and (not settings.CLI_CONFIG or settings.CLI_CONFIG.is_ce):
owner = DEFAULT
if not all([owner, project_name]):
error_message = get_project_error_message(owner, project_name)
if is_cli:
Printer.print_error(error_message)
sys.exit(1)
else:
raise PolyaxonClientException(error_message)
return owner, project_name
def get_local_project(is_cli: bool = False):
try:
return ProjectConfigManager.get_config()
except Exception: # noqa
if is_cli:
Printer.print_error(CACHE_ERROR, sys_exit=True)
else:
raise PolyaxonSchemaError(CACHE_ERROR)
def _cache_project(config, owner=None, project=None):
if (ProjectConfigManager.is_initialized()
and ProjectConfigManager.is_locally_initialized()):
if _is_same_project(owner, project):
ProjectConfigManager.set_config(config)
return
ProjectConfigManager.set_config(
config, visibility=ProjectConfigManager.VISIBILITY_GLOBAL)
def upload():
"""N.B. This is not available in all distributions.
Upload code of the current directory while respecting the .polyaxonignore file.
"""
import sys
from polyaxon.cli.errors import handle_cli_error
from polyaxon.client import PolyaxonClient
from polyaxon.exceptions import (
PolyaxonClientException,
PolyaxonHTTPError,
PolyaxonShouldExitError,
)
from polyaxon.managers.ignore import IgnoreConfigManager
from polyaxon.managers.project import ProjectConfigManager
from polyaxon.utils.formatting import Printer
from polyaxon.utils.path_utils import create_project_tarfile, get_files_by_paths
project = ProjectConfigManager.get_config_or_raise()
files = IgnoreConfigManager.get_unignored_filepaths()
try:
with create_project_tarfile(files, project.name) as filepath:
with get_files_by_paths("repo", [filepath]) as (files, files_size):
try:
PolyaxonClient().project.upload_repo(
project.user, project.name, files, files_size)
except (
PolyaxonHTTPError,
PolyaxonShouldExitError,
PolyaxonClientException,
) as e:
handle_cli_error(
e,
message="Could not upload code for project `{}`.".
format(project.name),
)
Printer.print_error(
"Check that the project exists, "
"and that you have access rights, "
"this could happen as well when uploading large files. "
"Please also make sure that you have enough space to upload the data."
)
sys.exit(1)
Printer.print_success("Files uploaded.")
except Exception as e:
handle_cli_error(e, message="Could not upload the file.")
sys.exit(1)
def _is_same_project(owner=None, project=None):
local_project = ProjectConfigManager.get_config()
if project and project == local_project.name:
return not all([owner, local_project.owner]) or owner == local_project.owner
def init(project, git_connection, git_url, polyaxonfile, polyaxonignore):
"""Initialize a new local project and cache directory.
Note: Make sure to add the local cache `.polyaxon`
to your `.gitignore` and `.dockerignore` files.
"""
if not any(
[project, git_connection, git_url, polyaxonfile, polyaxonignore]):
Printer.print_warning(
"`polyaxon init` did not receive any valid option.",
command_help="polyaxon init",
)
if project:
owner, project_name = get_project_or_local(project, is_cli=True)
try:
polyaxon_client = ProjectClient(owner=owner, project=project_name)
polyaxon_client.refresh_data()
except (ApiException, HTTPError) as e:
Printer.print_error(
"Make sure you have a project with this name `{}`".format(
project))
handle_cli_error(
e,
message="You can a create new project with this command: "
"polyaxon project create "
"--name={} [--description=...] [--tags=...]".format(
project_name),
)
sys.exit(1)
init_project = False
if ProjectConfigManager.is_initialized():
local_project = get_local_project()
click.echo(
"Warning! This project is already initialized with the following project:"
)
with indentation.indent(4):
indentation.puts("Owner: {}".format(local_project.owner))
indentation.puts("Project: {}".format(local_project.name))
if click.confirm("Would you like to override this current config?",
default=False):
init_project = True
else:
init_project = True
if init_project:
ProjectConfigManager.purge(
visibility=ProjectConfigManager.VISIBILITY_LOCAL)
config = polyaxon_client.client.sanitize_for_serialization(
polyaxon_client.project_data)
ProjectConfigManager.set_config(
config,
init=True,
visibility=ProjectConfigManager.VISIBILITY_LOCAL)
Printer.print_success("Project was initialized")
Printer.print_header(
"Make sure to add the local cache `.polyaxon` "
"to your `.gitignore` and `.dockerignore` files.")
else:
Printer.print_header("Project config was not changed.")
if git_connection or git_url:
init_git = False
if GitConfigManager.is_initialized():
click.echo("Warning! A {} file was found.".format(
GitConfigManager.CONFIG_FILE_NAME))
if click.confirm("Would you like to override it?", default=False):
init_git = True
else:
init_git = True
if init_git:
GitConfigManager.purge(
visibility=GitConfigManager.VISIBILITY_LOCAL)
config = GitConfigManager.CONFIG(
connection=git_connection,
git=V1GitType(url=git_url) if git_url else None,
)
GitConfigManager.set_config(config=config, init=True)
Printer.print_success("New {} file was created.".format(
GitConfigManager.CONFIG_FILE_NAME))
else:
Printer.print_header("{} file was not changed.".format(
GitConfigManager.CONFIG_FILE_NAME))
if polyaxonfile:
create_polyaxonfile()
if polyaxonignore:
init_ignore = False
if IgnoreConfigManager.is_initialized():
click.echo("Warning! A {} file was found.".format(
IgnoreConfigManager.CONFIG_FILE_NAME))
if click.confirm("Would you like to override it?", default=False):
init_ignore = True
else:
init_ignore = True
if init_ignore:
IgnoreConfigManager.init_config()
Printer.print_success("New {} file was created.".format(
IgnoreConfigManager.CONFIG_FILE_NAME))
else:
Printer.print_header("{} file was not changed.".format(
IgnoreConfigManager.CONFIG_FILE_NAME))
def test_default_props(self):
assert ProjectConfigManager.is_all_visibility() is True
assert ProjectConfigManager.IS_POLYAXON_DIR is True
assert ProjectConfigManager.CONFIG_FILE_NAME == ".project"
assert ProjectConfigManager.CONFIG == V1Project
def get_local_project():
try:
return ProjectConfigManager.get_config()
except:
Printer.print_error(CACHE_ERROR, sys_exit=True)
