cometx 3.6.4__tar.gz → 3.6.6__tar.gz

{cometx-3.6.4 → cometx-3.6.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cometx
-Version: 3.6.4
+Version: 3.6.6
 Summary: Python tools for Comet
 Home-page: https://github.com/comet-ml/cometx/
 Author: cometx development team
@@ -102,6 +102,7 @@ for ML management.
 * [cometx download](#cometx-download)
 * [cometx list](#cometx-list)
 * [cometx log](#cometx-log)
+* [cometx migrate-users](#cometx-migrate-users)
 * [cometx rename-duplicates](#cometx-rename-duplicates)
 * [cometx reproduce](#cometx-reproduce)
 * [cometx smoke-test](#cometx-smoke-test)
@@ -324,6 +325,7 @@ Not all combinations are possible:
 * `--symlink` - Instead of copying, create a link to an experiment in a project
 * `--sync` - Check to see if experiment name has been created first; if so, skip
 * `--path PATH` - Path to prepend to workspace_src when accessing files (supports ~ for home directory)
+* `--create-workspaces` - Attempt to create the destination workspace if it does not exist (requires appropriate API permissions)
 
 ### Using --path
 
@@ -385,6 +387,47 @@ Where TYPE is one of the following names:
 
 For more information, `cometx log --help`
 
+## cometx migrate-users
+
+This command migrates users into workspaces from a source Comet environment to a destination environment. It reads workspace membership from a chargeback report fetched from the source environment's admin API (or a pre-downloaded local file) and invites each user to the corresponding workspace in the destination environment by email. If a user with that email does not yet exist in the destination, they will be provisioned access once they sign up.
+
+```
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY [FLAGS ...]
+cometx migrate-users --api-key DEST_KEY --chargeback-report /path/to/report.json [FLAGS ...]
+```
+
+**Arguments:**
+* `--api-key API_KEY` - API key for the destination environment. Falls back to `COMET_API_KEY` if not provided.
+* `--url URL` - Base URL of the destination Comet environment (e.g. `https://comet.example.com`). Required for self-hosted instances when the API key does not encode the server URL.
+* `--source-api-key SOURCE_API_KEY` - API key for the source environment. Used to fetch the chargeback report. Required unless `--chargeback-report` is given.
+* `--source-url SOURCE_URL` - Base URL of the source Comet environment. Required for self-hosted source instances when the source API key does not encode the server URL.
+* `--chargeback-report PATH` - Path to a local chargeback report JSON file. When provided, `--source-api-key` is not required.
+
+### Flags
+
+* `--create-workspaces` - Create workspaces on the destination if they don't already exist (default: off)
+* `--dry-run` - Print what would happen without making any changes
+* `--failures-output PATH` - Path to write failed operations JSON (default: `bulk_add_failures_by_email.json`)
+
+### Examples
+
+```bash
+# Dry run — preview what would happen
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY --dry-run
+
+# Execute the migration
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY
+
+# Use a local chargeback report file
+cometx migrate-users --api-key DEST_KEY --chargeback-report /tmp/chargeback_reports.json
+
+# Self-hosted environments — provide explicit URLs
+cometx migrate-users --api-key DEST_KEY --url https://comet.dest.example.com \
+    --source-api-key SOURCE_KEY --source-url https://comet.src.example.com
+```
+
+For more information, `cometx migrate-users --help`
+
 ## cometx rename-duplicates
 
 This command is used to rename duplicate experiments within projects. When multiple experiments share the same name in a project, this command renames the duplicates to NAME-1, NAME-2, etc. while avoiding conflicts with existing names.
@@ -535,12 +578,8 @@ cometx admin chargeback-report [YEAR-MONTH]
 **Examples:**
 ```
 cometx admin chargeback-report
-<<<<<<< Updated upstream
-cometx admin usage-report
-=======
 cometx admin chargeback-report 2024-09
 ```
->>>>>>> Stashed changes
 
 #### usage-report
 

{cometx-3.6.4 → cometx-3.6.6}/README.md

@@ -46,6 +46,7 @@ for ML management.
 * [cometx download](#cometx-download)
 * [cometx list](#cometx-list)
 * [cometx log](#cometx-log)
+* [cometx migrate-users](#cometx-migrate-users)
 * [cometx rename-duplicates](#cometx-rename-duplicates)
 * [cometx reproduce](#cometx-reproduce)
 * [cometx smoke-test](#cometx-smoke-test)
@@ -268,6 +269,7 @@ Not all combinations are possible:
 * `--symlink` - Instead of copying, create a link to an experiment in a project
 * `--sync` - Check to see if experiment name has been created first; if so, skip
 * `--path PATH` - Path to prepend to workspace_src when accessing files (supports ~ for home directory)
+* `--create-workspaces` - Attempt to create the destination workspace if it does not exist (requires appropriate API permissions)
 
 ### Using --path
 
@@ -329,6 +331,47 @@ Where TYPE is one of the following names:
 
 For more information, `cometx log --help`
 
+## cometx migrate-users
+
+This command migrates users into workspaces from a source Comet environment to a destination environment. It reads workspace membership from a chargeback report fetched from the source environment's admin API (or a pre-downloaded local file) and invites each user to the corresponding workspace in the destination environment by email. If a user with that email does not yet exist in the destination, they will be provisioned access once they sign up.
+
+```
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY [FLAGS ...]
+cometx migrate-users --api-key DEST_KEY --chargeback-report /path/to/report.json [FLAGS ...]
+```
+
+**Arguments:**
+* `--api-key API_KEY` - API key for the destination environment. Falls back to `COMET_API_KEY` if not provided.
+* `--url URL` - Base URL of the destination Comet environment (e.g. `https://comet.example.com`). Required for self-hosted instances when the API key does not encode the server URL.
+* `--source-api-key SOURCE_API_KEY` - API key for the source environment. Used to fetch the chargeback report. Required unless `--chargeback-report` is given.
+* `--source-url SOURCE_URL` - Base URL of the source Comet environment. Required for self-hosted source instances when the source API key does not encode the server URL.
+* `--chargeback-report PATH` - Path to a local chargeback report JSON file. When provided, `--source-api-key` is not required.
+
+### Flags
+
+* `--create-workspaces` - Create workspaces on the destination if they don't already exist (default: off)
+* `--dry-run` - Print what would happen without making any changes
+* `--failures-output PATH` - Path to write failed operations JSON (default: `bulk_add_failures_by_email.json`)
+
+### Examples
+
+```bash
+# Dry run — preview what would happen
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY --dry-run
+
+# Execute the migration
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY
+
+# Use a local chargeback report file
+cometx migrate-users --api-key DEST_KEY --chargeback-report /tmp/chargeback_reports.json
+
+# Self-hosted environments — provide explicit URLs
+cometx migrate-users --api-key DEST_KEY --url https://comet.dest.example.com \
+    --source-api-key SOURCE_KEY --source-url https://comet.src.example.com
+```
+
+For more information, `cometx migrate-users --help`
+
 ## cometx rename-duplicates
 
 This command is used to rename duplicate experiments within projects. When multiple experiments share the same name in a project, this command renames the duplicates to NAME-1, NAME-2, etc. while avoiding conflicts with existing names.
@@ -479,12 +522,8 @@ cometx admin chargeback-report [YEAR-MONTH]
 **Examples:**
 ```
 cometx admin chargeback-report
-<<<<<<< Updated upstream
-cometx admin usage-report
-=======
 cometx admin chargeback-report 2024-09
 ```
->>>>>>> Stashed changes
 
 #### usage-report
 

{cometx-3.6.4 → cometx-3.6.6}/cometx/_version.py

@@ -12,5 +12,5 @@
 #  the express permission of Comet ML Inc.
 # *******************************************************
 
-version_info = (3, 6, 4)
+version_info = (3, 6, 6)
 __version__ = ".".join(map(str, version_info))

{cometx-3.6.4 → cometx-3.6.6}/cometx/api.py

@@ -171,7 +171,13 @@ class API(API):
         else:
             raise Exception("I don't know what to do with %r" % parsed_url.path)
 
-    def upload_panel_code(self, workspace: str, panel_name: str, code: str) -> None:
+    def upload_panel_code(
+        self,
+        workspace: str,
+        panel_name: str,
+        code: str,
+        template_id: str = None,
+    ) -> Dict[str, Any]:
         """
         Upload Python code as a panel in a workspace.
 
@@ -179,6 +185,8 @@ class API(API):
             workspace (str): the workspace to place the panel into
             panel_name (str): the name of the panel
             code (str): the code to turn into a panel
+            template_id (str, optional): if provided, overwrites the existing
+                panel with this ID instead of creating a new one
 
         Example:
         ```python linenums="1"
@@ -194,15 +202,19 @@ class API(API):
         ```
         """
         filename = create_panel_zip(panel_name, code)
-        self.upload_panel_zip(workspace, filename)
+        return self.upload_panel_zip(workspace, filename, template_id=template_id)
 
-    def upload_panel_zip(self, workspace: str, filename: str) -> Dict[str, str]:
+    def upload_panel_zip(
+        self, workspace: str, filename: str, template_id: str = None
+    ) -> Dict[str, str]:
         """
         Upload a panel zip file to a workspace.
 
         Args:
             workspace (str): the workspace to place the panel into
             filename (str): the name of the panel zip to upload
+            template_id (str, optional): if provided, overwrites the existing
+                panel with this ID instead of creating a new one
 
         Returns: dictionary of results
 
@@ -215,6 +227,8 @@ class API(API):
         ```
         """
         params = {"teamName": workspace}
+        if template_id is not None:
+            params["templateId"] = template_id
         payload = {}
         with open(filename, "rb") as fp:
             files = {"file": (filename, fp)}
@@ -226,6 +240,120 @@ class API(API):
             )
         return results.json()
 
+    def _get_project_id(self, workspace: str, project_name: str) -> str:
+        projects = self._client.get_from_endpoint(
+            "projects", params={"workspaceName": workspace}
+        )
+        project = next(
+            (p for p in projects["projects"] if p["projectName"] == project_name),
+            None,
+        )
+        if project is None:
+            raise Exception(
+                f"Project {project_name!r} not found in workspace {workspace!r}"
+            )
+        return project["projectId"]
+
+    def create_dashboard(
+        self,
+        workspace: str,
+        project_name: str,
+        template_name: str,
+        template_id: str = None,
+        panels: List[str] = None,
+        **kwargs,
+    ) -> Dict[str, Any]:
+        """
+        Create a new dashboard in a project.
+
+        Args:
+            workspace (str): the workspace name the project belongs to
+            project_name (str): the project name to create the dashboard in
+            template_name (str): name for the new dashboard
+            template_id (str, optional): source template ID to clone code
+                panel associations from an existing dashboard
+            panels (list, optional): list of panel template IDs to include
+                in the new dashboard
+            **kwargs: additional DashboardTemplate fields
+
+        Returns: dictionary representing the created DashboardTemplate
+
+        Example:
+        ```python linenums="1"
+        from cometx import API
+
+        api = API()
+        dashboard = api.create_dashboard(
+            workspace="my-workspace",
+            project_name="my-project",
+            template_name="My Dashboard",
+            panels=["panel-template-id-1", "panel-template-id-2"],
+        )
+        ```
+        """
+        project_id = self._get_project_id(workspace, project_name)
+
+        body = {"project_id": project_id, "template_name": template_name, **kwargs}
+        if template_id is not None:
+            body["template_id"] = template_id
+        if panels is not None:
+            body["codePanelTemplateIds"] = panels
+        results = self._client.post_from_endpoint(
+            "write/dashboard-template/create",
+            payload=body,
+        )
+        return results.json()
+
+    def update_dashboard(
+        self,
+        workspace: str,
+        project_name: str,
+        template_id: str,
+        template_name: str = None,
+        panels: List[str] = None,
+        **kwargs,
+    ) -> Dict[str, Any]:
+        """
+        Update an existing dashboard in a project.
+
+        Args:
+            workspace (str): the workspace name the project belongs to
+            project_name (str): the project name the dashboard belongs to
+            template_id (str): the ID of the dashboard to update
+            template_name (str, optional): new name for the dashboard
+            panels (list, optional): list of panel template IDs to set on
+                the dashboard
+            **kwargs: additional DashboardTemplate fields
+
+        Returns: dictionary representing the updated DashboardTemplate
+
+        Example:
+        ```python linenums="1"
+        from cometx import API
+
+        api = API()
+        dashboard = api.update_dashboard(
+            workspace="my-workspace",
+            project_name="my-project",
+            template_id="existing-dashboard-id",
+            template_name="Renamed Dashboard",
+            panels=["panel-template-id-1", "panel-template-id-2"],
+        )
+        ```
+        """
+        project_id = self._get_project_id(workspace, project_name)
+
+        body = {"project_id": project_id, "template_id": template_id, **kwargs}
+        if template_name is not None:
+            body["template_name"] = template_name
+        if panels is not None:
+            body["codePanelTemplateIds"] = panels
+        results = self._client.post_from_endpoint(
+            "dashboard-templates/project/upsert",
+            payload=body,
+        )
+        return results.json()
+
     def log_pr_curves(
         self, experiment, y_true, y_predicted, labels=None, overwrite=False, step=None
     ):

{cometx-3.6.4 → cometx-3.6.6}/cometx/cli/__init__.py

@@ -26,6 +26,7 @@ Examples:
     cometx smoke-test
     cometx update
     cometx admin
+    cometx migrate-users
 
 For more information:
     cometx COMMAND --help
@@ -47,6 +48,7 @@ from . import (
     download,
     list_command,
     log,
+    migrate_users,
     rename_duplicates,
     reproduce,
     smoke_test,
@@ -108,6 +110,7 @@ def main(raw_args=sys.argv[1:]):
     add_subparser(subparsers, config, "config")
     add_subparser(subparsers, rename_duplicates, "rename-duplicates")
     add_subparser(subparsers, smoke_test, "smoke-test")
+    add_subparser(subparsers, migrate_users, "migrate-users")
     add_subparser(
         subparsers, upload_optimizer_experiments, "upload-optimizer-experiments"
     )

{cometx-3.6.4 → cometx-3.6.6}/cometx/cli/copy.py

@@ -78,6 +78,8 @@ import threading
 import time
 import urllib.parse
 import zipfile
+
+import requests
 from datetime import datetime, timedelta
 
 from comet_ml import APIExperiment, Artifact, Experiment, OfflineExperiment
@@ -102,6 +104,7 @@ from ..api import API
 from ..framework.comet.download_manager import sanitize_filename
 from ..utils import remove_extra_slashes
 from .copy_utils import upload_single_offline_experiment
+from .migrate_users import _create_workspace
 
 ADDITIONAL_ARGS = False
 
@@ -197,6 +200,12 @@ def get_parser_arguments(parser):
         type=str,
         default=None,
     )
+    parser.add_argument(
+        "--create-workspaces",
+        help="Attempt to create the destination workspace if it does not exist",
+        default=False,
+        action="store_true",
+    )
 
 
 def copy(parsed_args, remaining=None):
@@ -214,6 +223,7 @@ def copy(parsed_args, remaining=None):
             parsed_args.ignore,
             parsed_args.debug,
             parsed_args.sync,
+            parsed_args.create_workspaces,
         )
 
         # Wait for all uploads to complete
@@ -790,7 +800,7 @@ class CopyManager:
         # Restore default signal handler
         signal.signal(signal.SIGINT, signal.default_int_handler)
 
-    def copy(self, source, destination, symlink, ignore, debug, sync):
+    def copy(self, source, destination, symlink, ignore, debug, sync, create_workspaces=False):
         """ """
         self.ignore = ignore
         self.debug = debug
@@ -823,9 +833,35 @@ class CopyManager:
         # First check to make sure workspace_dst exists:
         workspaces = self.api.get_workspaces()
         if workspace_dst not in workspaces:
-            raise Exception(
-                f"{workspace_dst} does not exist; use the Comet UI to create it"
-            )
+            if create_workspaces:
+                print(
+                    f"Workspace {workspace_dst!r} does not exist, attempting to create it..."
+                )
+                dest_url = self.api._get_url_server()
+                headers = {
+                    "Authorization": self.api.api_key,
+                    "Content-Type": "application/json",
+                }
+                try:
+                    _create_workspace(dest_url, headers, workspace_dst)
+                    print(f"Workspace {workspace_dst!r} created successfully.")
+                except requests.exceptions.HTTPError as exc:
+                    raise Exception(
+                        f"Workspace {workspace_dst!r} does not exist and could not be "
+                        f"created automatically (HTTP {exc.response.status_code}). "
+                        f"Please create it via the Comet UI and try again."
+                    ) from exc
+                except requests.exceptions.RequestException as exc:
+                    raise Exception(
+                        f"Workspace {workspace_dst!r} does not exist and could not be "
+                        f"created automatically: {exc}. "
+                        f"Please create it via the Comet UI and try again."
+                    ) from exc
+            else:
+                raise Exception(
+                    f"{workspace_dst} does not exist; use --create-workspaces to "
+                    f"create it automatically, or create it via the Comet UI"
+                )
 
         if project_src == "panels":
             # experiment_src may be "*" or filename

cometx-3.6.6/cometx/cli/migrate_users.py

@@ -0,0 +1,393 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+# ****************************************
+#                              __
+#   _________  ____ ___  ___  / /__  __
+#  / ___/ __ \/ __ `__ \/ _ \/ __/ |/_/
+# / /__/ /_/ / / / / / /  __/ /__>  <
+# \___/\____/_/ /_/ /_/\___/\__/_/|_|
+#
+#
+#  Copyright (c) 2022 Cometx Development
+#      Team. All rights reserved.
+# ****************************************
+"""
+Migrate users into workspaces from a source environment to a destination
+environment based on a chargeback report.
+
+The chargeback report is fetched automatically from the source environment's
+admin API, or you can provide a local JSON file with --chargeback-report.
+
+Examples:
+
+$ cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY --dry-run
+$ cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY
+$ cometx migrate-users --api-key DEST_KEY --chargeback-report /path/to/report.json
+"""
+
+import argparse
+import base64
+import json
+import os
+import sys
+import urllib.parse
+
+import requests
+
+ADDITIONAL_ARGS = False
+
+
+def get_parser_arguments(parser):
+    parser.add_argument(
+        "--api-key",
+        help="API key for the destination environment (used to add workspace members). "
+        "Falls back to COMET_API_KEY environment variable if not provided.",
+        type=str,
+        default=None,
+    )
+    parser.add_argument(
+        "--url",
+        help="Base URL of the destination Comet environment (e.g. https://comet.example.com). "
+        "Required for self-hosted instances when the API key does not encode the server URL.",
+        type=str,
+        default=None,
+    )
+    parser.add_argument(
+        "--source-api-key",
+        help="API key for the source environment (used to fetch the chargeback report). "
+        "Required unless --chargeback-report is given.",
+        type=str,
+        default=None,
+    )
+    parser.add_argument(
+        "--source-url",
+        help="Base URL of the source Comet environment. "
+        "Required for self-hosted instances when the source API key does not encode the server URL.",
+        type=str,
+        default=None,
+    )
+    parser.add_argument(
+        "--chargeback-report",
+        help="Path to a local chargeback report JSON file (skips fetching from source environment)",
+        type=str,
+        default=None,
+    )
+    parser.add_argument(
+        "--create-workspaces",
+        help="Create workspaces on the destination environment if they don't already exist",
+        default=False,
+        action="store_true",
+    )
+    parser.add_argument(
+        "--dry-run",
+        help="Print what would happen without making any changes",
+        default=False,
+        action="store_true",
+    )
+    parser.add_argument(
+        "--failures-output",
+        help="Path to write failed operations JSON (default: bulk_add_failures_by_email.json)",
+        type=str,
+        default="bulk_add_failures_by_email.json",
+    )
+
+
+def _resolve_server_url(api_key, explicit_url=None):
+    """Return the server base URL.
+
+    Priority:
+    1. ``explicit_url`` (from --url / --source-url), if provided.
+    2. URL encoded inside the API key (new-style keys contain ``*<base64>``).
+    3. Error — no silent fallback to cloud URL.
+    """
+    if explicit_url:
+        parsed = urllib.parse.urlparse(explicit_url)
+        if parsed.scheme != "https":
+            print("[ERROR] --url/--source-url must use https://.")
+            sys.exit(1)
+        return explicit_url.rstrip("/")
+
+    if "*" in api_key:
+        try:
+            _, encoded = api_key.split("*", 1)
+            # base64 padding may be missing
+            padding = (4 - len(encoded) % 4) % 4
+            payload = json.loads(base64.b64decode(encoded + "=" * padding))
+            return payload["baseUrl"].rstrip("/")
+        except Exception:
+            pass  # Fall through to error below
+
+    print(
+        "[ERROR] Cannot determine the server URL from the API key. "
+        "Pass --url (destination) or --source-url (source) explicitly, "
+        "e.g. --url https://comet.example.com"
+    )
+    sys.exit(1)
+
+
+def _fetch_chargeback_report(server_url, source_api_key):
+    url = f"{server_url}/api/admin/chargeback/report"
+    print(f"Fetching chargeback report from {url}...")
+    resp = requests.get(
+        url,
+        headers={"Authorization": source_api_key},
+        timeout=30,
+    )
+    resp.raise_for_status()
+    return resp.json()
+
+
+def _load_chargeback_report(path):
+    print(f"Loading chargeback report from {path}...")
+    with open(path, "r") as f:
+        return json.load(f)
+
+
+def _get_existing_workspaces(dest_url, headers):
+    url = f"{dest_url}/api/rest/v2/workspaces"
+    resp = requests.get(url, headers=headers, timeout=15)
+    resp.raise_for_status()
+    data = resp.json()
+    # Handle {"workspaceNames": [...]} dict shape
+    if isinstance(data, dict):
+        names = data.get("workspaceNames", [])
+        if names and isinstance(names[0], dict):
+            return {ws["name"] for ws in names}
+        return set(names)
+    # Handle list of dicts with a "name" key
+    if data and isinstance(data[0], dict):
+        return {ws["name"] for ws in data}
+    # Handle flat list of strings
+    return set(data)
+
+
+def _create_workspace(dest_url, headers, workspace_name):
+    url = f"{dest_url}/api/rest/v2/write/workspace/new"
+    resp = requests.post(
+        url,
+        headers=headers,
+        json={"name": workspace_name},
+        timeout=15,
+    )
+    resp.raise_for_status()
+
+
+def _add_member(url, headers, email, workspace_name):
+    """Attempt to add a single member. Returns (status, error_info_or_None)."""
+    payload = {
+        "userEmail": email,
+        "workspaceName": workspace_name,
+        "admin": True,
+    }
+    try:
+        response = requests.post(
+            url,
+            headers=headers,
+            json=payload,
+            timeout=15,
+        )
+
+        if response.status_code == 200:
+            return "added", None
+
+        try:
+            error_msg = response.json()
+        except ValueError:
+            error_msg = response.text
+
+        is_already_member = (
+            isinstance(error_msg, dict)
+            and "already member of" in error_msg.get("msg", "")
+        )
+        if is_already_member:
+            return "already_member", None
+
+        return "failed", {
+            "status": response.status_code,
+            "error": error_msg,
+        }
+
+    except requests.exceptions.RequestException as e:
+        return "failed", {
+            "status": "exception",
+            "error": str(e),
+        }
+
+
+def migrate_users(parsed_args):
+    api_key = parsed_args.api_key or os.environ.get("COMET_API_KEY")
+    if not api_key:
+        print("[ERROR] No API key found. Set COMET_API_KEY or pass --api-key.")
+        sys.exit(1)
+
+    source_api_key = parsed_args.source_api_key
+    create_workspaces = parsed_args.create_workspaces
+    dry_run = parsed_args.dry_run
+    failures_output = parsed_args.failures_output
+
+    if not parsed_args.chargeback_report and not source_api_key:
+        print(
+            "[ERROR] --source-api-key is required when --chargeback-report is not provided."
+        )
+        sys.exit(1)
+
+    dest_url = _resolve_server_url(api_key, parsed_args.url)
+    print(f"Destination URL: {dest_url}")
+
+    if parsed_args.chargeback_report:
+        data = _load_chargeback_report(parsed_args.chargeback_report)
+    else:
+        source_url = _resolve_server_url(source_api_key, parsed_args.source_url)
+        print(f"Source URL: {source_url}")
+        if source_url == dest_url and source_api_key == api_key:
+            print(
+                "[WARNING] Source and destination URL and API key are identical. "
+                "Are you sure you want to migrate users to the same environment?"
+            )
+        try:
+            data = _fetch_chargeback_report(source_url, source_api_key)
+        except requests.exceptions.RequestException as e:
+            print(f"[ERROR] Failed to fetch chargeback report: {e}")
+            sys.exit(1)
+
+    workspaces = data.get("workspaces", [])
+    print(f"Found {len(workspaces)} workspaces\n")
+
+    headers = {
+        "Authorization": api_key,
+        "Content-Type": "application/json",
+    }
+
+    if create_workspaces and not dry_run:
+        print("Checking existing workspaces on destination...")
+        try:
+            existing = _get_existing_workspaces(dest_url, headers)
+        except requests.exceptions.RequestException as e:
+            print(f"[ERROR] Failed to list destination workspaces: {e}")
+            sys.exit(1)
+
+        ws_created = 0
+        for workspace in workspaces:
+            ws_name = workspace["name"]
+            if ws_name not in existing:
+                try:
+                    _create_workspace(dest_url, headers, ws_name)
+                    print(f"  [CREATED] Workspace '{ws_name}'")
+                    ws_created += 1
+                except requests.exceptions.RequestException as e:
+                    print(f"  [ERROR] Failed to create workspace '{ws_name}': {e}")
+        print(f"Created {ws_created} new workspace(s)\n")
+    elif create_workspaces and dry_run:
+        print("[DRY RUN] Would check/create workspaces on destination\n")
+
+    add_member_url = f"{dest_url}/api/rest/v2/write/add-workspace-member"
+
+    total_added = 0
+    total_skipped = 0
+    total_no_email = 0
+    total_already_member = 0
+    total_failed = 0
+    failures = []
+
+    for workspace in workspaces:
+        ws_name = workspace["name"]
+        members = workspace.get("members", [])
+
+        if not members:
+            print(f"[SKIP] Workspace '{ws_name}' — no members, skipping")
+            total_skipped += 1
+            continue
+
+        print(f"\n{'=' * 60}")
+        print(f"Workspace: {ws_name} ({len(members)} members)")
+        print(f"{'=' * 60}")
+
+        ws_added = 0
+        ws_no_email = 0
+        ws_already_member = 0
+        ws_failed = 0
+
+        for member in members:
+            email = member.get("email")
+
+            if not email:
+                ws_no_email += 1
+                total_no_email += 1
+                print(f"  [SKIP] '{member['userName']}' has no email, skipping")
+                continue
+
+            if dry_run:
+                print(f"  [DRY RUN] Would add '{email}' to '{ws_name}'")
+                ws_added += 1
+                total_added += 1
+                continue
+
+            status, error_info = _add_member(add_member_url, headers, email, ws_name)
+
+            if status == "added":
+                ws_added += 1
+                total_added += 1
+            elif status == "already_member":
+                ws_already_member += 1
+                total_already_member += 1
+            else:
+                print(
+                    f"  [FAIL] '{email}' -> '{ws_name}' "
+                    f"— {error_info['status']}: {error_info['error']}"
+                )
+                ws_failed += 1
+                total_failed += 1
+                failures.append(
+                    {
+                        "workspace": ws_name,
+                        "email": email,
+                        **error_info,
+                    }
+                )
+
+        if dry_run:
+            print(f"  [DRY RUN] Total: would add {ws_added}/{len(members)} users")
+        else:
+            parts = [f"  Added {ws_added}/{len(members)} users successfully"]
+            if ws_already_member:
+                parts.append(f"{ws_already_member} already a member")
+            if ws_no_email:
+                parts.append(f"{ws_no_email} skipped (no email)")
+            if ws_failed:
+                parts.append(f"{ws_failed} failed")
+            print(
+                parts[0]
+                + (" (" + ", ".join(parts[1:]) + ")" if len(parts) > 1 else "")
+            )
+
+    print(f"\n{'=' * 60}")
+    print("SUMMARY")
+    print(f"{'=' * 60}")
+    print(f"  Workspaces processed : {len(workspaces)}")
+    print(f"  Workspaces skipped   : {total_skipped} (no members)")
+    print(f"  Users added          : {total_added}")
+    print(f"  Already a member     : {total_already_member}")
+    print(f"  Skipped (no email)   : {total_no_email}")
+    print(f"  Failures             : {total_failed}")
+
+    if dry_run:
+        print(f"\n  *** DRY RUN MODE — no API calls were made ***")
+        print(f"  *** Remove --dry-run to execute for real ***")
+
+    if failures:
+        with open(failures_output, "w") as f:
+            json.dump(failures, f, indent=2)
+        print(f"\n  Failed operations saved to {failures_output}")
+
+
+def main(args):
+    parser = argparse.ArgumentParser(
+        description=__doc__, formatter_class=argparse.RawDescriptionHelpFormatter
+    )
+    get_parser_arguments(parser)
+    parsed_args = parser.parse_args(args)
+    migrate_users(parsed_args)
+
+
+if __name__ == "__main__":
+    main(sys.argv[1:])

{cometx-3.6.4 → cometx-3.6.6}/cometx.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cometx
-Version: 3.6.4
+Version: 3.6.6
 Summary: Python tools for Comet
 Home-page: https://github.com/comet-ml/cometx/
 Author: cometx development team
@@ -102,6 +102,7 @@ for ML management.
 * [cometx download](#cometx-download)
 * [cometx list](#cometx-list)
 * [cometx log](#cometx-log)
+* [cometx migrate-users](#cometx-migrate-users)
 * [cometx rename-duplicates](#cometx-rename-duplicates)
 * [cometx reproduce](#cometx-reproduce)
 * [cometx smoke-test](#cometx-smoke-test)
@@ -324,6 +325,7 @@ Not all combinations are possible:
 * `--symlink` - Instead of copying, create a link to an experiment in a project
 * `--sync` - Check to see if experiment name has been created first; if so, skip
 * `--path PATH` - Path to prepend to workspace_src when accessing files (supports ~ for home directory)
+* `--create-workspaces` - Attempt to create the destination workspace if it does not exist (requires appropriate API permissions)
 
 ### Using --path
 
@@ -385,6 +387,47 @@ Where TYPE is one of the following names:
 
 For more information, `cometx log --help`
 
+## cometx migrate-users
+
+This command migrates users into workspaces from a source Comet environment to a destination environment. It reads workspace membership from a chargeback report fetched from the source environment's admin API (or a pre-downloaded local file) and invites each user to the corresponding workspace in the destination environment by email. If a user with that email does not yet exist in the destination, they will be provisioned access once they sign up.
+
+```
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY [FLAGS ...]
+cometx migrate-users --api-key DEST_KEY --chargeback-report /path/to/report.json [FLAGS ...]
+```
+
+**Arguments:**
+* `--api-key API_KEY` - API key for the destination environment. Falls back to `COMET_API_KEY` if not provided.
+* `--url URL` - Base URL of the destination Comet environment (e.g. `https://comet.example.com`). Required for self-hosted instances when the API key does not encode the server URL.
+* `--source-api-key SOURCE_API_KEY` - API key for the source environment. Used to fetch the chargeback report. Required unless `--chargeback-report` is given.
+* `--source-url SOURCE_URL` - Base URL of the source Comet environment. Required for self-hosted source instances when the source API key does not encode the server URL.
+* `--chargeback-report PATH` - Path to a local chargeback report JSON file. When provided, `--source-api-key` is not required.
+
+### Flags
+
+* `--create-workspaces` - Create workspaces on the destination if they don't already exist (default: off)
+* `--dry-run` - Print what would happen without making any changes
+* `--failures-output PATH` - Path to write failed operations JSON (default: `bulk_add_failures_by_email.json`)
+
+### Examples
+
+```bash
+# Dry run — preview what would happen
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY --dry-run
+
+# Execute the migration
+cometx migrate-users --api-key DEST_KEY --source-api-key SOURCE_KEY
+
+# Use a local chargeback report file
+cometx migrate-users --api-key DEST_KEY --chargeback-report /tmp/chargeback_reports.json
+
+# Self-hosted environments — provide explicit URLs
+cometx migrate-users --api-key DEST_KEY --url https://comet.dest.example.com \
+    --source-api-key SOURCE_KEY --source-url https://comet.src.example.com
+```
+
+For more information, `cometx migrate-users --help`
+
 ## cometx rename-duplicates
 
 This command is used to rename duplicate experiments within projects. When multiple experiments share the same name in a project, this command renames the duplicates to NAME-1, NAME-2, etc. while avoiding conflicts with existing names.
@@ -535,12 +578,8 @@ cometx admin chargeback-report [YEAR-MONTH]
 **Examples:**
 ```
 cometx admin chargeback-report
-<<<<<<< Updated upstream
-cometx admin usage-report
-=======
 cometx admin chargeback-report 2024-09
 ```
->>>>>>> Stashed changes
 
 #### usage-report
 

{cometx-3.6.4 → cometx-3.6.6}/cometx.egg-info/SOURCES.txt

@@ -35,6 +35,7 @@ cometx/cli/delete_assets.py
 cometx/cli/download.py
 cometx/cli/list_command.py
 cometx/cli/log.py
+cometx/cli/migrate_users.py
 cometx/cli/rename_duplicates.py
 cometx/cli/reproduce.py
 cometx/cli/smoke_test.py