mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-15 17:30:13 +08:00
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
import click
|
||
import sys
|
||
from astrbot.core.config.default import VERSION
|
||
from .commands.cmd_init import init
|
||
from .commands.cmd_run import run
|
||
|
||
logo_tmpl = r"""
|
||
___ _______.___________..______ .______ ______ .___________.
|
||
/ \ / | || _ \ | _ \ / __ \ | |
|
||
/ ^ \ | (----`---| |----`| |_) | | |_) | | | | | `---| |----`
|
||
/ /_\ \ \ \ | | | / | _ < | | | | | |
|
||
/ _____ \ .----) | | | | |\ \----.| |_) | | `--' | | |
|
||
/__/ \__\ |_______/ |__| | _| `._____||______/ \______/ |__|
|
||
"""
|
||
|
||
|
||
@click.group()
|
||
@click.version_option(VERSION, prog_name="AstrBot")
|
||
def cli() -> None:
|
||
"""The AstrBot CLI"""
|
||
click.echo(logo_tmpl)
|
||
click.echo("Welcome to AstrBot CLI!")
|
||
click.echo(f"AstrBot version: {VERSION}")
|
||
|
||
|
||
@click.command()
|
||
@click.argument("command_name", required=False, type=str)
|
||
def help(command_name: str | None) -> None:
|
||
"""显示命令的帮助信息
|
||
|
||
如果提供了 COMMAND_NAME，则显示该命令的详细帮助信息。
|
||
否则，显示通用帮助信息。
|
||
"""
|
||
ctx = click.get_current_context()
|
||
if command_name:
|
||
# 查找指定命令
|
||
command = cli.get_command(ctx, command_name)
|
||
if command:
|
||
# 显示特定命令的帮助信息
|
||
click.echo(command.get_help(ctx))
|
||
else:
|
||
click.echo(f"Unknown command: {command_name}")
|
||
sys.exit(1)
|
||
else:
|
||
# 显示通用帮助信息
|
||
click.echo(cli.get_help(ctx))
|
||
|
||
|
||
cli.add_command(init)
|
||
cli.add_command(run)
|
||
cli.add_command(help)
|
||
|
||
if __name__ == "__main__":
|
||
cli() |