| from dataclasses import dataclass |
| from functools import wraps |
| from textwrap import dedent |
| from typing import Any, Callable, Optional |
|
|
| import click |
|
|
|
|
| def require_workspace(): |
| def require_workspace_inner(f: Callable[..., None]): |
| @wraps(f) |
| @click.pass_context |
| def new_func(ctx: click.Context, *args: Any, **kwargs: Any): |
| if ctx.obj.is_workspace_valid: |
| return ctx.invoke(f, *args, **kwargs) |
| else: |
| click.echo( |
| "The current directory is not a valid Task Weaver project directory. " |
| "There needs to be a `taskweaver-config.json` in the root of the project directory. " |
| "Please change the working directory to a valid project directory or initialize a new one. " |
| "Refer to --help for more information.", |
| ) |
| ctx.exit(1) |
|
|
| return new_func |
|
|
| return require_workspace_inner |
|
|
|
|
| @dataclass |
| class CliContext: |
| workspace: Optional[str] |
| workspace_param: Optional[str] |
| is_workspace_valid: bool |
| is_workspace_empty: bool |
|
|
|
|
| def get_ascii_banner() -> str: |
| return dedent( |
| r""" |
| ========================================================= |
| _____ _ _ __ |
| |_ _|_ _ ___| | _ | | / /__ ____ __ _____ _____ |
| | |/ _` / __| |/ /| | /| / / _ \/ __ `/ | / / _ \/ ___/ |
| | | (_| \__ \ < | |/ |/ / __/ /_/ /| |/ / __/ / |
| |_|\__,_|___/_|\_\|__/|__/\___/\__,_/ |___/\___/_/ |
| ========================================================= |
| """, |
| ).strip() |
|
|
