tabsdata-cli 2.0.0__cp312-abi3-win_amd64.whl

tabsdata_cli-2.0.0.data/purelib/tabsdatak/cli/td/user_group.py

@@ -0,0 +1,200 @@
+#
+# Copyright 2026 Tabsdata Inc.
+#
+
+
+import rich_click as click
+from rich.table import Table
+
+from tabsdatak.cli._registry import cli_command
+from tabsdatak.cli.cli_utils import (
+    echo_more_results,
+    format_option,
+    hint_common_solutions,
+    logical_prompt,
+    make_console,
+    max_result_option,
+    render,
+    take,
+    verify_login_or_prompt,
+)
+
+
+@cli_command
+@click.group()
+def user():
+    """User management commands"""
+
+
+@user.command()
+@click.option("--name", "-n", help="Name of the user to create.")
+@click.option("--full-name", "-f", help="Full name of the user.")
+@click.option("--email", "-e", help="Email of the user.")
+@click.option(
+    "--password",
+    "-p",
+    help="Password for the user. Will be prompted for it if not provided.",
+)
+@click.option("--disabled", is_flag=True, default=False, help="Disable the user.")
+@click.pass_context
+def create(
+    ctx: click.Context,
+    name: str,
+    full_name: str,
+    email: str,
+    password: str,
+    disabled: bool,
+):
+    """Create a new user"""
+    verify_login_or_prompt(ctx)
+    click.echo("Creating a new user")
+    click.echo("-" * 10)
+    name = name or logical_prompt(ctx, "Name of the user")
+    if password is None:
+        password = logical_prompt(ctx, "Password for the user", hide_input=True)
+        second_password = logical_prompt(
+            ctx, "Please re-enter the password", hide_input=True
+        )
+        if password != second_password:
+            raise click.ClickException("Passwords do not match.")
+    full_name = full_name or logical_prompt(
+        ctx, "Full name of the user", default_value=name
+    )
+    email = email or logical_prompt(ctx, "Email of the user", default_value="")
+    try:
+        ctx.obj["tabsdataserver"].create_user(
+            name,
+            password,
+            full_name=full_name,
+            email=email if email else None,
+            enabled=not disabled,
+        )
+        click.echo("User created successfully")
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to create user: {e}")
+
+
+@user.command()
+@click.option("--name", "-n", help="Name of the user to delete.")
+@click.option(
+    "--confirm",
+    help="Write 'delete' to confirm deletion. Will be prompted for it if not provided.",
+)
+@click.pass_context
+def delete(ctx: click.Context, name: str, confirm: str):
+    """Delete a user by name"""
+    verify_login_or_prompt(ctx)
+    name = name or logical_prompt(ctx, "Name of the user to delete")
+    click.echo(f"Deleting user: {name}")
+    click.echo("-" * 10)
+    confirm = confirm or logical_prompt(ctx, "Please type 'delete' to confirm deletion")
+    if confirm != "delete":
+        raise click.ClickException(
+            "Deletion not confirmed. The confirmation word is 'delete'."
+        )
+    try:
+        ctx.obj["tabsdataserver"].delete_user(name)
+        click.echo("User deleted successfully")
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to delete user: {e}")
+
+
+@user.command()
+@format_option
+@click.option("--name", "-n", help="Name of the user to display.")
+@click.pass_context
+def get(ctx: click.Context, name: str):
+    """Display a user by name"""
+    verify_login_or_prompt(ctx)
+    name = name or logical_prompt(ctx, "Name of the user to display")
+    try:
+        user = ctx.obj["tabsdataserver"].get_user(name)
+
+        def _table():
+            table = Table(title="User")
+            table.add_column("Name", style="cyan", no_wrap=True)
+            table.add_column("Full name")
+            table.add_column("Email")
+            table.add_column("Enabled")
+            table.add_row(user.name, user.full_name, user.email, str(user.enabled))
+            click.echo()
+            make_console().print(table)
+            click.echo()
+
+        render(ctx, data=lambda: user._data, table=_table)
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to display user: {e}")
+
+
+@user.command()
+@format_option
+@max_result_option
+@click.pass_context
+def list(ctx: click.Context):
+    """List all users"""
+    verify_login_or_prompt(ctx)
+    try:
+        list_of_users, truncated = take(
+            ctx, ctx.obj["tabsdataserver"].list_users_generator()
+        )
+
+        def _table():
+            table = Table(title="Users")
+            table.add_column("Name", style="cyan", no_wrap=True)
+            table.add_column("Full name")
+            table.add_column("Email")
+            table.add_column("Enabled")
+            for user in list_of_users:
+                table.add_row(user.name, user.full_name, user.email, str(user.enabled))
+            click.echo()
+            make_console().print(table)
+            click.echo(f"Number of users: {len(list_of_users)}")
+            click.echo()
+
+        render(ctx, data=lambda: [u._data for u in list_of_users], table=_table)
+        echo_more_results(ctx, truncated)
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to list users: {e}")
+
+
+@user.command()
+@click.option("--name", "-n", help="Name of the user to update.")
+@click.option("--full-name", "-f", help="New full name of the user.")
+@click.option("--email", "-e", help="New email of the user.")
+@click.option("--password", "-p", help="New password for the user.")
+@click.option(
+    "--enabled",
+    type=bool,
+    default=None,
+    help="Indicate if the user is enabled or disabled",
+)
+@click.pass_context
+def update(
+    ctx: click.Context,
+    name: str,
+    full_name: str,
+    email: str,
+    password: str,
+    enabled: bool,
+):
+    """Update a user by name"""
+    verify_login_or_prompt(ctx)
+    name = name or logical_prompt(ctx, "Name of the user to update")
+    click.echo(f"Updating user: {name}")
+    click.echo("-" * 10)
+    try:
+        ctx.obj["tabsdataserver"].update_user(
+            name,
+            full_name=full_name,
+            email=email,
+            password=password,
+            enabled=enabled,
+        )
+        click.echo("User updated successfully")
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to update user: {e}")

tabsdata_cli-2.0.0.data/purelib/tabsdatak/cli/td/worker_group.py

@@ -0,0 +1,185 @@
+#
+# Copyright 2026 Tabsdata Inc.
+#
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING, List
+
+import rich_click as click
+
+from tabsdatak.cli._registry import cli_command
+from tabsdatak.cli.cli_utils import (
+    echo_more_results,
+    format_option,
+    hint_common_solutions,
+    make_console,
+    max_result_option,
+    output_format,
+    render,
+    show_hint,
+    take,
+    verify_login_or_prompt,
+)
+from tabsdatak.cli.td.execution_utils import (
+    echo_function_run_status,
+    echo_plan_status,
+    echo_transaction_status,
+    emit_worker_logs,
+    obtain_list_worker_filters,
+    span_projects,
+    workers_table,
+)
+from tabsdatak.sdk.status_utils.worker import WORKER_VALID_USER_PROVIDED_STATUSES
+
+if TYPE_CHECKING:
+    from tabsdatak.sdk.tabsdata_server import TabsdataServer
+
+
+@cli_command
+@click.group()
+def worker():
+    """Worker management commands"""
+
+
+@worker.command(name="list")
+@click.option("--proj", type=str, help="Name of the project to filter workers by.")
+@click.option(
+    "--status",
+    type=str,
+    multiple=True,
+    help=(
+        "Filter workers by status. The possible statuses are "
+        f"'{WORKER_VALID_USER_PROVIDED_STATUSES}'. This field is "
+        "case-insensitive. "
+    ),
+)
+@click.option(
+    "--coll",
+    type=str,
+    help="Collection of the function to which the workers belong.",
+)
+@click.option(
+    "--fn", type=str, help="Name of the function to which the workers belong."
+)
+@click.option("--plan", type=str, help="ID of the plan to filter workers by.")
+@click.option(
+    "--plan-name",
+    help=(
+        "A plan name wildcard to match. "
+        "For example, 'my_plan*' will match all plans "
+        "with names starting with 'my_plan'."
+    ),
+)
+@click.option("--fnrun", type=str, help="ID of the function run to filter workers by.")
+@click.option("--trx", type=str, help="ID of the transaction to filter workers by.")
+@format_option
+@max_result_option
+@click.pass_context
+def list_workers(
+    ctx: click.Context,
+    proj: str,
+    status: List[str],
+    coll: str,
+    fn: str,
+    plan: str,
+    plan_name: str,
+    fnrun: str,
+    trx: str,
+):
+    """List workers across every project you can view."""
+    verify_login_or_prompt(ctx)
+    try:
+        server: TabsdataServer = ctx.obj["tabsdataserver"]
+        span_projects(server)
+        request_filter = obtain_list_worker_filters(
+            status=status,
+            exe_name=plan_name,
+            execution_id=plan,
+            fn=fn,
+            fnrun=fnrun,
+            collection=coll,
+            trx=trx,
+            project=proj,
+        )
+        if output_format(ctx) == "table":
+            if plan:
+                echo_plan_status(server, plan)
+            if trx:
+                echo_transaction_status(server, trx)
+            if fnrun:
+                echo_function_run_status(server, fnrun)
+        list_of_workers, truncated = take(
+            ctx, server.list_workers_generator(filter=request_filter)
+        )
+
+        if output_format(ctx) != "table":
+            render(
+                ctx,
+                data=[w._data for w in list_of_workers],
+                table=lambda: None,
+            )
+            echo_more_results(ctx, truncated)
+            return
+
+        click.echo()
+        console = make_console()
+        console.print(workers_table(list_of_workers))
+        click.echo(f"Number of workers: {len(list_of_workers)}")
+        echo_more_results(ctx, truncated)
+        click.echo()
+        show_hint(
+            ctx,
+            "Read logs with 'tdk worker logs --worker <WORKER-ID>', or "
+            "'tdk fnrun logs --fnrun <RUN-ID>' for a function run's workers.",
+        )
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to list workers: {e}")
+
+
+@worker.command()
+@click.option("--worker", required=True, help="ID of the worker whose logs to show.")
+@click.option(
+    "--file",
+    help=(
+        "Path of the file where the logs will be saved, or '-' to write them to "
+        "stdout. Any folders provided before the file must already exist."
+    ),
+)
+@format_option
+@click.pass_context
+def logs(ctx: click.Context, worker: str, file: str):
+    """
+    Show the logs of a worker.
+    """
+    verify_login_or_prompt(ctx)
+    try:
+        server: TabsdataServer = ctx.obj["tabsdataserver"]
+        emit_worker_logs(ctx, server, worker, file)
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to show logs: {e}")
+
+
+@worker.command()
+@click.option("--worker", required=True, help="ID of the worker to recover.")
+@click.pass_context
+def recover(ctx: click.Context, worker: str):
+    """
+    Recover the transaction a worker belongs to: the transaction's failed or
+    on-hold function runs are re-scheduled.
+    """
+    verify_login_or_prompt(ctx)
+    try:
+        server: TabsdataServer = ctx.obj["tabsdataserver"]
+        span_projects(server)
+        target = server.get_worker(worker)
+        trx = target.transaction.id
+        click.echo(f"Recovering transaction '{trx}' of worker '{worker}'")
+        click.echo("-" * 10)
+        server.recover_transaction(trx)
+        click.echo("Transaction recovered successfully")
+    except Exception as e:
+        hint_common_solutions(ctx, e)
+        raise click.ClickException(f"Failed to recover worker: {e}")