__init__.py

"""Self-service CLI command group for v2 REST API.

Registers self-service sub-groups per entity under
``backend.ai v2 my {entity} {command}``.
"""

from __future__ import annotations

import click

from ai.backend.common.cli import LazyGroup


@click.group()
def my() -> None:
    """Self-service commands for the current user."""


@my.group(cls=LazyGroup, import_name="ai.backend.client.cli.v2.my.keypair:keypair")
def keypair() -> None:
    """My keypair commands."""


@my.group(cls=LazyGroup, import_name="ai.backend.client.cli.v2.my.role:role")
def role() -> None:
    """My role commands."""


@my.group(
    cls=LazyGroup,
    import_name="ai.backend.client.cli.v2.my.login_history:login_history",
)
def login_history() -> None:
    """My login history commands."""


@my.group(
    cls=LazyGroup,
    import_name="ai.backend.client.cli.v2.my.login_session:login_session",
)
def login_session() -> None:
    """My login session commands."""


@my.group(cls=LazyGroup, import_name="ai.backend.client.cli.v2.my.export:export")
def export() -> None:
    """My export commands."""


@my.group(cls=LazyGroup, import_name="ai.backend.client.cli.v2.my.session:session")
def session() -> None:
    """My session commands."""


@my.group(cls=LazyGroup, import_name="ai.backend.client.cli.v2.my.deployment:deployment")
def deployment() -> None:
    """My deployment commands."""


@my.group(
    cls=LazyGroup,
    import_name="ai.backend.client.cli.v2.my.resource_allocation:resource_allocation",
    name="resource-allocation",
)
def resource_allocation() -> None:
    """My resource allocation commands."""


@my.group(
    cls=LazyGroup,
    import_name="ai.backend.client.cli.v2.my.resource_policy:resource_policy",
    name="resource-policy",
)
def resource_policy() -> None:
    """My resource policy commands."""


@my.group(
    cls=LazyGroup,
    import_name="ai.backend.client.cli.v2.my.storage_host:storage_host",
    name="storage-host",
)
def storage_host() -> None:
    """My storage host commands."""