[ Web Proxy ]
URL:
Viewing: https://raw.githubusercontent.com/Gwill/feast/master/sdk/python/feast/cli/projects.py [Back]  [Original]

import click
import yaml

from feast import utils
from feast.cli.cli_options import tagsOption
from feast.errors import FeastObjectNotFoundException, ProjectNotFoundException
from feast.repo_operations import create_feature_store


@click.group(name="projects")
def projects_cmd():
    """
    Access projects
    """
    pass


@projects_cmd.command("describe")
@click.argument("name", type=click.STRING)
@click.pass_context
def project_describe(ctx: click.Context, name: str):
    """
    Describe a project
    """
    store = create_feature_store(ctx)
    try:
        project = store.get_project(name)
    except FeastObjectNotFoundException as e:
        print(e)
        exit(1)
    print(
        yaml.dump(
            yaml.safe_load(str(project)),
            default_flow_style=False,
            sort_keys=False,
        )
    )


@projects_cmd.command("current_project")
@click.pass_context
def project_current(ctx: click.Context):
    """
    Returns the current project configured with FeatureStore object
    """
    store = create_feature_store(ctx)
    try:
        project = store.get_project(name=None)
    except FeastObjectNotFoundException as e:
        print(e)
        exit(1)
    print(
        yaml.dump(
            yaml.safe_load(str(project)),
            default_flow_style=False,
            sort_keys=False,
        )
    )


@projects_cmd.command("delete")
@click.argument("name", type=click.STRING)
@click.option(
    "-y",
    "--yes",
    is_flag=True,
    default=False,
    help="Skip confirmation prompt and delete immediately.",
)
@click.pass_context
def project_delete(ctx: click.Context, name: str, yes: bool):
    """
    Delete a project and all its resources from the registry.
    """
    store = create_feature_store(ctx)

    if not yes:
        click.confirm(
            f"Are you sure you want to delete project '{name}'? "
            "This will remove all associated resources from the registry.",
            abort=True,
        )

    try:
        store.delete_project(name)
    except (FeastObjectNotFoundException, ProjectNotFoundException) as e:
        print(str(e))
        raise SystemExit(1)

    print(f"Project '{name}' deleted successfully.")


@projects_cmd.command(name="list")
@tagsOption
@click.pass_context
def project_list(ctx: click.Context, tags: list[str]):
    """
    List all projects
    """
    store = create_feature_store(ctx)
    table = []
    tags_filter = utils.tags_list_to_dict(tags)
    for project in store.list_projects(tags=tags_filter):
        table.append([project.name, project.description, project.tags, project.owner])
    from tabulate import tabulate

    print(
        tabulate(
            table,
            headers=["NAME", "DESCRIPTION", "TAGS", "OWNER"],
            tablefmt="plain",
        )
    )

Web Proxy Viewer  |  New URL  |  Original Page