polyaxon 2.1.8__py3-none-any.whl → 2.2.0.post1__py3-none-any.whl

polyaxon/_cli/artifacts.py

@@ -73,7 +73,7 @@ def ls(ctx, project, query, sort, limit, offset):
     \b
     $ polyaxon artifacts ls --project=acme/data-versioning
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     list_project_versions(
@@ -146,11 +146,12 @@ def register(
     $ polyaxon artifacts register -p owner/name -ver v1 --tags="tag1,tag2"
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     register_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.ARTIFACT,
@@ -228,11 +229,12 @@ def copy(
     $ polyaxon artifacts copy -p owner/name -ver v1 --tags="tag1,tag2" --name new-v1
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     copy_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.ARTIFACT,
@@ -266,11 +268,12 @@ def get(ctx, project, version):
     $ polyaxon artifacts get -p owner/data-versioning -ver rc1
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     get_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.ARTIFACT,
         version=version,
@@ -297,7 +300,7 @@ def stages(ctx, project, version):
     $ polyaxon artifacts stages -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     get_project_version_stages(
@@ -333,7 +336,7 @@ def delete(ctx, project, version, yes):
     $ polyaxon artifacts get -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     delete_project_version(
@@ -376,7 +379,7 @@ def update(ctx, project, version, name, description, tags):
     $ polyaxon artifacts update --tags="foo, bar"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     update_project_version(
@@ -424,7 +427,7 @@ def stage(ctx, project, version, to, reason, message):
     $ polyaxon artifacts stage -p acme/foobar -ver rc12 --to=staging --reason GithubAction --message="Use carefully!"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     stage_project_version(
@@ -462,7 +465,7 @@ def transfer(ctx, project, version, to_project):
     $ polyaxon artifacts transfer -p acme/foobar -ver rc12 --to-project=dest-project
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     transfer_project_version(
@@ -531,7 +534,7 @@ def pull(
     \b
     $ polyaxon artifacts pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     pull_one_or_many_project_versions(
@@ -612,7 +615,7 @@ def push(
     \b
     $ polyaxon artifacts pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     push_one_or_many_project_versions(
@@ -650,11 +653,12 @@ def push(
 def dashboard(ctx, project, version, yes, url):
     """Open this artifact version's dashboard details in browser."""
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     open_project_version_dashboard(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.ARTIFACT,
         version=version,

polyaxon/_cli/components.py

@@ -114,7 +114,7 @@ def ls(ctx, project, query, sort, limit, offset):
     \b
     $ polyaxon components ls -p=acme/kaniko
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     polyaxon_client = get_current_or_public_client()
@@ -170,7 +170,7 @@ def register(ctx, polyaxonfile, project, version, description, tags, force):
     $ polyaxon components register -f polyaxonfile.yaml -p owner/name -ver v1 --tags="tag1,tag2"
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -188,6 +188,7 @@ def register(ctx, polyaxonfile, project, version, description, tags, force):
 
     register_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.COMPONENT,
@@ -256,11 +257,12 @@ def copy(
     $ polyaxon components copy -p owner/name -ver v1 --tags="tag1,tag2" --name new-v1
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     copy_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.COMPONENT,
@@ -292,13 +294,14 @@ def get(ctx, project, version):
     $ polyaxon components get -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     polyaxon_client = get_current_or_public_client()
 
     get_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.COMPONENT,
         version=version,
@@ -327,7 +330,7 @@ def stages(ctx, project, version):
     $ polyaxon components stages -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     polyaxon_client = get_current_or_public_client()
@@ -369,7 +372,7 @@ def delete(ctx, project, version, yes):
     $ polyaxon components get -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     delete_project_version(
@@ -412,7 +415,7 @@ def update(ctx, project, version, name, description, tags):
     $ polyaxon components update --tags="foo, bar"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     update_project_version(
@@ -461,7 +464,7 @@ def stage(ctx, project, version, to, reason, message):
     $ polyaxon components stage -p acme/foobar -ver rc12 --to=staging --reason GithubAction --message="Use carefully!"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     stage_project_version(
@@ -499,7 +502,7 @@ def transfer(ctx, project, version, to_project):
     $ polyaxon components transfer -p acme/foobar -ver rc12 --to-project=dest-project
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     transfer_project_version(
@@ -560,7 +563,7 @@ def pull(
     \b
     $ polyaxon components pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     pull_one_or_many_project_versions(
@@ -641,7 +644,7 @@ def push(
     \b
     $ polyaxon components pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     push_one_or_many_project_versions(
@@ -679,11 +682,12 @@ def push(
 def dashboard(ctx, project, version, yes, url):
     """Open this component version's dashboard details in browser."""
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     open_project_version_dashboard(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.COMPONENT,
         version=version,

polyaxon/_cli/config.py

@@ -44,6 +44,27 @@ def set_home_path(home_path: str):
     settings.set_home_config(_config)
 
 
+def set_owner(owner: str):
+    try:
+        _config = UserConfigManager.get_config_or_default()
+    except Exception as e:
+        logger.debug(
+            "Home configuration could not be loaded.\n"
+            "Error: %s\n"
+            "Purging home configuration and resetting values.",
+            e,
+        )
+        UserConfigManager.purge()
+        _config = UserConfigManager.get_config_or_default()
+
+    data = owner.split("/")
+    if len(data) > 2:
+        Printer.error("Invalid owner format, please provide a valid owner.")
+        sys.exit(1)
+    _config.organization = owner
+    UserConfigManager.set_config(_config)
+
+
 def validate_options(ctx, param, value):
     possible_values = ["verbose", "host"]
     if value and value not in possible_values:
@@ -199,6 +220,11 @@ def get(keys):
     type=click.Path(exists=False),
     help="To set POLYAXON_HOME to specify the context where the CLI/Client reads/writes global configuration.",
 )
+@click.option(
+    "--owner",
+    type=click.Path(exists=False),
+    help="To set current organization or team space.",
+)
 @click.option(
     "--disable-errors-reporting",
     type=bool,
@@ -230,6 +256,11 @@ def set_(**kwargs):  # pylint:disable=redefined-builtin
         home_path = kwargs.pop("home", None)
         set_home_path(home_path)
 
+    if kwargs.get("owner") is not None:
+        owner = kwargs.pop("owner", None)
+        if owner:
+            set_owner(owner)
+
     from polyaxon._managers.auth import AuthConfigManager
 
     try:

polyaxon/_cli/dashboard.py

@@ -1,5 +1,7 @@
 import sys
 
+from typing import Optional
+
 import click
 
 from clipped.formatting import Printer
@@ -10,10 +12,21 @@ from polyaxon.api import POLYAXON_CLOUD_HOST
 from polyaxon.logger import clean_outputs
 
 
+def get_project_subpath_url(owner: str, team: Optional[str], project: str):
+    if team:
+        return "{}/ts/{}/{}".format(owner, team, project)
+    return "{}/{}".format(owner, project)
+
+
 def get_dashboard_url(
-    base: str = "ui", subpath: str = "", use_cloud: bool = False
+    base: str = "ui", subpath: str = "", use_cloud: bool = False, host: str = None
 ) -> str:
-    host = POLYAXON_CLOUD_HOST if use_cloud else clean_host(settings.CLIENT_CONFIG.host)
+    if not host:
+        host = (
+            POLYAXON_CLOUD_HOST
+            if use_cloud
+            else clean_host(settings.CLIENT_CONFIG.host)
+        )
     dashboard_url = "{}/{}/".format(host, base)
     if subpath:
         return "{}{}/".format(dashboard_url, subpath.rstrip("/"))

polyaxon/_cli/init.py

@@ -112,7 +112,7 @@ def init(project, git_connection, git_url, polyaxonfile, polyaxonignore, yes):
             command_help="polyaxon init",
         )
     if project:
-        owner, project_name = get_project_or_local(project, is_cli=True)
+        owner, _, project_name = get_project_or_local(project, is_cli=True)
         try:
             polyaxon_client = ProjectClient(
                 owner=owner, project=project_name, manual_exceptions_handling=True

polyaxon/_cli/models.py

@@ -69,7 +69,7 @@ def ls(ctx, project, query, sort, limit, offset):
     \b
     $ polyaxon models ls -p=acme/project-name
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
 
@@ -143,11 +143,12 @@ def register(
     $ polyaxon models register -p owner/name -ver v1 --tags="tag1,tag2"
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     register_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.MODEL,
@@ -225,11 +226,12 @@ def copy(
     $ polyaxon models copy -p owner/name -ver v1 --tags="tag1,tag2" --name new-v1
     """
     version = version or ctx.obj.get("version")
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     copy_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         version=version,
         kind=V1ProjectVersionKind.MODEL,
@@ -262,11 +264,12 @@ def get(ctx, project, version):
     $ polyaxon models get -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     get_project_version(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.MODEL,
         version=version,
@@ -293,7 +296,7 @@ def stages(ctx, project, version):
     $ polyaxon models stages -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     get_project_version_stages(
@@ -332,7 +335,7 @@ def delete(ctx, project, version, yes):
     $ polyaxon models get -p owner/my-project -ver rc12
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     delete_project_version(
@@ -375,7 +378,7 @@ def update(ctx, project, version, name, description, tags):
     $ polyaxon models update --tags="foo, bar"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     update_project_version(
@@ -423,7 +426,7 @@ def stage(ctx, project, version, to, reason, message):
     $ polyaxon models stage -p acme/foobar -ver rc12 --to=staging --reason GithubAction --message="Use carefully!"
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     stage_project_version(
@@ -461,7 +464,7 @@ def transfer(ctx, project, version, to_project):
     $ polyaxon models transfer -p acme/foobar -ver rc12 --to-project=dest-project
     """
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     transfer_project_version(
@@ -530,7 +533,7 @@ def pull(
     \b
     $ polyaxon models pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     pull_one_or_many_project_versions(
@@ -611,7 +614,7 @@ def push(
     \b
     $ polyaxon models pull -p acme/foobar -a --path /tmp/versions
     """
-    owner, project_name = get_project_or_local(
+    owner, _, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     push_one_or_many_project_versions(
@@ -649,11 +652,12 @@ def push(
 def dashboard(ctx, project, version, yes, url):
     """Open this model version's dashboard details in browser."""
     version = version or ctx.obj.get("version") or "latest"
-    owner, project_name = get_project_or_local(
+    owner, team, project_name = get_project_or_local(
         project or ctx.obj.get("project"), is_cli=True
     )
     open_project_version_dashboard(
         owner=owner,
+        team=team,
         project_name=project_name,
         kind=V1ProjectVersionKind.MODEL,
         version=version,