96 lines
3.5 KiB
Python
96 lines
3.5 KiB
Python
|
"""初始化 jinrong_agent 库：依次执行 01-共用底座 + 02-agent专用 两份 DDL。
|
||
|
|
|
||
|
|
用法（密码与 core/reset.ps1 同一套，二选一）：
|
||
|
|
# 方式一：环境变量（避免命令行明文）
|
||
|
|
$env:MYSQL_PWD = "你的密码"; python scripts/init_agent_db.py
|
||
|
|
# 方式二：命令行传参
|
||
|
|
python scripts/init_agent_db.py --mysql-password 123456
|
||
|
|
# 可选参数
|
||
|
|
python scripts/init_agent_db.py --mysql-host 127.0.0.1 --mysql-port 3306 --mysql-user root
|
||
|
|
|
||
|
|
说明：
|
||
|
|
- 全新机器首次执行即可建齐：底座 11 张（01）+ Agent 专用 7 张（02）；
|
||
|
|
- CREATE DATABASE 带 IF NOT EXISTS；表已存在时 MySQL 报 1050，脚本提示后安全跳过，
|
||
|
|
不会 DROP 任何数据（如需彻底重建，请手动 DROP DATABASE jinrong_agent 后重跑）；
|
||
|
|
- 游客审计依赖 jinrong_agent.audit_log（在 01 中），客服归档/备注依赖 02 中的表。
|
||
|
|
"""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import argparse
|
||
|
|
import os
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
import pymysql
|
||
|
|
from pymysql.constants import CLIENT
|
||
|
|
from pymysql.err import OperationalError
|
||
|
|
|
||
|
|
ROOT = Path(__file__).resolve().parent.parent
|
||
|
|
SQL_FILES = [
|
||
|
|
("01-共用底座", ROOT / "docs/项目框架设计/表设计/01-mysql-共用底座.sql"),
|
||
|
|
("02-agent专用", ROOT / "docs/项目框架设计/表设计/02-mysql-agent专用.sql"),
|
||
|
|
]
|
||
|
|
|
||
|
|
ER_TABLE_EXISTS = 1050
|
||
|
|
|
||
|
|
|
||
|
|
def run_sql_file(cur, label: str, path: Path) -> None:
|
||
|
|
if not path.exists():
|
||
|
|
print(f"[ERROR] SQL 文件不存在: {path}")
|
||
|
|
sys.exit(1)
|
||
|
|
print(f">> 执行 {label}: {path.name}")
|
||
|
|
sql = path.read_text(encoding="utf-8")
|
||
|
|
try:
|
||
|
|
cur.execute(sql)
|
||
|
|
while cur.nextset():
|
||
|
|
pass
|
||
|
|
print(f" {label} 执行完成")
|
||
|
|
except OperationalError as e:
|
||
|
|
# 1050 = 表已存在：非全新环境重跑的预期情况，安全跳过，不破坏数据
|
||
|
|
if e.args and e.args[0] == ER_TABLE_EXISTS:
|
||
|
|
print(f" [SKIP] {label} 部分表已存在（1050），跳过已建表；如需重建请先 DROP DATABASE jinrong_agent")
|
||
|
|
else:
|
||
|
|
raise
|
||
|
|
|
||
|
|
|
||
|
|
def main() -> None:
|
||
|
|
parser = argparse.ArgumentParser(description="初始化 jinrong_agent 库（01 底座 + 02 Agent 专用）")
|
||
|
|
parser.add_argument("--mysql-host", default=os.getenv("MYSQL_HOST", "127.0.0.1"))
|
||
|
|
parser.add_argument("--mysql-port", type=int, default=int(os.getenv("MYSQL_PORT", "3306")))
|
||
|
|
parser.add_argument("--mysql-user", default=os.getenv("MYSQL_USER", "root"))
|
||
|
|
parser.add_argument("--mysql-password", default=os.getenv("MYSQL_PWD", ""),
|
||
|
|
help="MySQL 密码；也可用 MYSQL_PWD 环境变量")
|
||
|
|
args = parser.parse_args()
|
||
|
|
|
||
|
|
if not args.mysql_password:
|
||
|
|
print("[ERROR] 缺少密码：请用 --mysql-password 传参，或预设 MYSQL_PWD 环境变量")
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
conn = pymysql.connect(
|
||
|
|
host=args.mysql_host,
|
||
|
|
port=args.mysql_port,
|
||
|
|
user=args.mysql_user,
|
||
|
|
password=args.mysql_password,
|
||
|
|
charset="utf8mb4",
|
||
|
|
client_flag=CLIENT.MULTI_STATEMENTS,
|
||
|
|
autocommit=True,
|
||
|
|
)
|
||
|
|
try:
|
||
|
|
with conn.cursor() as cur:
|
||
|
|
for label, path in SQL_FILES:
|
||
|
|
run_sql_file(cur, label, path)
|
||
|
|
|
||
|
|
cur.execute("USE jinrong_agent")
|
||
|
|
cur.execute("SHOW TABLES")
|
||
|
|
tables = [r[0] for r in cur.fetchall()]
|
||
|
|
print(f"\njinrong_agent 现有 {len(tables)} 张表:")
|
||
|
|
for t in tables:
|
||
|
|
print(f" - {t}")
|
||
|
|
finally:
|
||
|
|
conn.close()
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|