代码提交

2026-09-21 19:03:31 +08:00
parent 34d656492c
commit 97986d9a1a
217 changed files with 12042 additions and 0 deletions
@@ -0,0 +1,70 @@
+"""初始化数据库：建库 + 建表 + 建初始管理员。
+
+用法：
+    python -m app.scripts.init_db            # 建表（不会删数据）
+    python -m app.scripts.init_db --drop     # 先删表再建（危险，会清空数据）
+"""
+
+from __future__ import annotations
+
+import argparse
+import sys
+
+from app.core.config import settings
+from app.core.database import SessionLocal, engine, ensure_database_exists
+from app.core.security import hash_password
+from app.dao.account_dao import AccountDao
+from app.model import Account, Base, Role
+
+
+def init(drop: bool = False) -> None:
+    ensure_database_exists()
+    if drop:
+        confirm = input("这会删除 wolin 库里的全部表和数据，输入 yes 继续：")
+        if confirm.strip().lower() != "yes":
+            print("已取消")
+            return
+        Base.metadata.drop_all(bind=engine)
+        print("已删除所有表")
+
+    Base.metadata.create_all(bind=engine)
+    tables = ", ".join(sorted(Base.metadata.tables.keys()))
+    print(f"建表完成：{tables}")
+
+    with SessionLocal() as db:
+        if AccountDao.get_by_username(db, settings.ADMIN_USERNAME, with_deleted=True) is None:
+            db.add(
+                Account(
+                    username=settings.ADMIN_USERNAME,
+                    hashed_password=hash_password(settings.ADMIN_PASSWORD),
+                    real_name="系统管理员",
+                    role=Role.ADMIN.value,
+                )
+            )
+            db.commit()
+            print(f"初始管理员：{settings.ADMIN_USERNAME} / {settings.ADMIN_PASSWORD}")
+        else:
+            print(f"管理员 {settings.ADMIN_USERNAME} 已存在，跳过")
+
+        if AccountDao.get_by_username(db, "viewer", with_deleted=True) is None:
+            db.add(
+                Account(
+                    username="viewer",
+                    hashed_password=hash_password("viewer123"),
+                    real_name="只读访客",
+                    role=Role.VIEWER.value,
+                )
+            )
+            db.commit()
+            print("只读账号：viewer / viewer123")
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="初始化沃林学生管理系统数据库")
+    parser.add_argument("--drop", action="store_true", help="先删除所有表再重建（会丢数据）")
+    args = parser.parse_args()
+    try:
+        init(drop=args.drop)
+    except Exception as exc:  # noqa: BLE001
+        print(f"初始化失败：{exc}", file=sys.stderr)
+        raise