feat(cli): 部分完成 nyahome init 和 check 命令
This commit is contained in:
+124
-1
@@ -3,9 +3,11 @@
|
||||
避免在此文件中引用 router 和 service 模块内的代码。
|
||||
"""
|
||||
|
||||
import os
|
||||
from typing import Annotated
|
||||
|
||||
import typer
|
||||
from rich.table import Table
|
||||
|
||||
from nyahome import __version__
|
||||
from nyahome.cli.cli import console
|
||||
@@ -57,7 +59,7 @@ def run() -> None:
|
||||
host=os.getenv("NYAHOME_UVICORN_HOST", "0.0.0.0"),
|
||||
port=int(os.getenv("NYAHOME_UVICORN_PORT", "9000")),
|
||||
timeout_graceful_shutdown=2,
|
||||
log_config="logging.yaml",
|
||||
log_config=".nyahome/logging.yaml",
|
||||
log_level="debug",
|
||||
)
|
||||
|
||||
@@ -79,6 +81,127 @@ def openapi(
|
||||
console.print(f"[cyan]已经保存 openapi.json 到 {path} 。[/cyan]")
|
||||
|
||||
|
||||
@app.command()
|
||||
def init() -> None:
|
||||
"""
|
||||
交互式初始化 NyaHome。
|
||||
"""
|
||||
from dotenv import set_key
|
||||
from rich.prompt import Confirm, IntPrompt, Prompt
|
||||
|
||||
from nyahome.cli.cli import DATA_DIR, ENV_PATH, LOGGING_YAML, db_driver_available, db_type_allowlist
|
||||
from nyahome.cli.cli_check import LOGGING_YAML_CONTENT
|
||||
|
||||
console.print("\n准备初始化 NyaHome。")
|
||||
|
||||
# 1.数据目录初始化
|
||||
if DATA_DIR.is_dir():
|
||||
console.print("\n1.数据目录 [cyan].nyahome[/cyan] 已存在，跳过创建。")
|
||||
else:
|
||||
DATA_DIR.mkdir()
|
||||
console.print("\n1.已创建数据目录 [cyan].nyahome[/cyan]。")
|
||||
|
||||
# 2.日志配置文件初始化
|
||||
if LOGGING_YAML.is_file():
|
||||
console.print("\n2.日志配置文件 [cyan]logging.yaml[/cyan] 已存在，跳过创建。")
|
||||
if Confirm.ask("需要[yellow]覆盖其至默认值[/yellow]吗？", default=False):
|
||||
with open(LOGGING_YAML, "w") as f:
|
||||
f.write(LOGGING_YAML_CONTENT)
|
||||
console.print("已覆盖至默认值。")
|
||||
else:
|
||||
with open(LOGGING_YAML, "w") as f:
|
||||
f.write(LOGGING_YAML_CONTENT)
|
||||
console.print("\n2.已创建日志配置文件 [cyan]logging.yaml[/cyan]。")
|
||||
|
||||
# 3.环境变量初始化
|
||||
console.print("\n3.一些必须的环境变量需要设置。")
|
||||
if Confirm.ask("\n设置[yellow]数据库连接[/yellow]（环境变量）？", default=True):
|
||||
db_type = Prompt.ask(
|
||||
"NYAHOME_DB_TYPE - 数据库协议", default="sqlite", choices=db_type_allowlist, console=console
|
||||
)
|
||||
al = db_driver_available.get(db_type, [])
|
||||
db_driver = Prompt.ask("NYAHOME_DB_DRIVER - 数据库驱动库", default=al[0], choices=al, console=console)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_TYPE", db_type)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_DRIVER", db_driver)
|
||||
console.print("已设置数据库类型和驱动程序。")
|
||||
if db_type == "sqlite":
|
||||
console.print("采用 [cyan]sqlite[/cyan] 数据库，无需再额外配置。")
|
||||
else:
|
||||
console.print("接下来，需要继续设置数据库的连接凭证。")
|
||||
db_name = Prompt.ask("NYAHOME_DB_NAME - 数据库名称", default="nyahome", console=console)
|
||||
db_user = Prompt.ask("NYAHOME_DB_USER - 数据库用户", default="nyahome", console=console)
|
||||
db_password = Prompt.ask("NYAHOME_DB_PASSWORD - 密码", default="nyahome", console=console)
|
||||
db_host = Prompt.ask("NYAHOME_DB_HOST - 主机名", default="localhost", console=console)
|
||||
db_port = Prompt.ask("NYAHOME_DB_PORT - 端口", default="3006", console=console)
|
||||
if db_password == "nyahome":
|
||||
console.print("[yellow]使用了默认数据库密码。如果是生产环境，建议更换。[/yellow]")
|
||||
set_key(ENV_PATH, "NYAHOME_DB_NAME", db_name)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_USER", db_user)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_PASSWORD", db_password)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_HOST", db_host)
|
||||
set_key(ENV_PATH, "NYAHOME_DB_PORT", db_port)
|
||||
console.print("已设置数据库连接凭证。")
|
||||
if Confirm.ask("\n设置 [yellow]uvicorn[/yellow] 启动配置？", default=True):
|
||||
un_host = Prompt.ask("NYAHOME_UVICORN_HOST - 绑定主机名", default="0.0.0.0", console=console)
|
||||
un_port = IntPrompt.ask("NYAHOME_UVICORN_PORT - 绑定端口", default=9000, console=console)
|
||||
un_reload = Confirm.ask("NYAHOME_UVICORN_RELOAD - 自动重载", default=False, console=console)
|
||||
if un_reload:
|
||||
console.print("[yellow]启用了 uvicorn reload。如果是生产环境，建议关闭。[/yellow]")
|
||||
set_key(ENV_PATH, "NYAHOME_UVICORN_HOST", un_host)
|
||||
set_key(ENV_PATH, "NYAHOME_UVICORN_PORT", str(un_port))
|
||||
set_key(ENV_PATH, "NYAHOME_UVICORN_RELOAD", "true" if un_reload else "false")
|
||||
console.print("已设置 uvicorn 启动配置。")
|
||||
|
||||
|
||||
@app.command()
|
||||
def check() -> None:
|
||||
"""
|
||||
详细自检查环境变量与设置，得到检查报告，可能有用。
|
||||
"""
|
||||
from dotenv import load_dotenv
|
||||
|
||||
from nyahome.cli.cli import DATA_DIR, ENV_PATH
|
||||
from nyahome.cli.cli_check import (
|
||||
check_database_connector,
|
||||
check_database_type,
|
||||
check_nyahome_status,
|
||||
check_uvicorn,
|
||||
cw,
|
||||
)
|
||||
|
||||
def _(step: int, description: str) -> str:
|
||||
return f"\n[cyan]> Step {step}[/cyan]: {description}"
|
||||
|
||||
console.print(_(1, "检查可用的数据库驱动程序"))
|
||||
table1 = Table(title="数据库驱动库")
|
||||
table1.add_column("驱动库", style="cyan")
|
||||
table1.add_column("状态与描述")
|
||||
database_connectors = check_database_connector()
|
||||
for key, value in database_connectors.items():
|
||||
table1.add_row(key, value)
|
||||
console.print(table1)
|
||||
|
||||
console.print(_(2, "检查环境变量"))
|
||||
load_dotenv(ENV_PATH)
|
||||
check_database_type(os.environ)
|
||||
check_uvicorn(os.environ)
|
||||
|
||||
console.print(_(3, "检查 NyaHome 安装模式与运行环境"))
|
||||
check_nyahome_status()
|
||||
|
||||
console.print(_(4, "检查 NyaHome 数据目录可用性"))
|
||||
if not DATA_DIR.is_dir():
|
||||
cw.warning("NyaHome 数据目录 .nyahome 不存在。")
|
||||
else:
|
||||
if not (DATA_DIR / "logging.yaml").is_file():
|
||||
cw.warning(".nyahome/logging.yaml 日志配置文件不存在。")
|
||||
if not (DATA_DIR / "contents").is_dir():
|
||||
cw.warning(".nyahome/contents 上传目录不存在。")
|
||||
cw.info("可以运行 [cyan]nyahome init[/cyan] 命令来重新初始化数据目录。")
|
||||
|
||||
console.print(f"\n[yellow]完成自检，共有 {cw.counter} 个警告。[/yellow]")
|
||||
|
||||
|
||||
app.add_typer(env_app, name="env", no_args_is_help=True, help="设置 NyaHome 应用的环境变量。")
|
||||
app.add_typer(aii_app, name="aii", no_args_is_help=True, help="添加、设置、修改 AI 提供商和模型。")
|
||||
|
||||
|
||||
Reference in New Issue
Block a user