42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
||
"""创建默认管理员账号 admin@example.com / admin123，便于前端登录。"""
|
||
import asyncio
|
||
import os
|
||
import sys
|
||
|
||
# 在导入 th_agenter 前加载 .env
|
||
from pathlib import Path
|
||
_root = Path(__file__).resolve().parents[1]
|
||
sys.path.insert(0, str(_root))
|
||
os.chdir(_root)
|
||
|
||
from dotenv import load_dotenv
|
||
load_dotenv()
|
||
|
||
# 若未设置，可在此指定（或通过环境变量传入）
|
||
# os.environ.setdefault("DATABASE_URL", "mysql+aiomysql://root:xxx@localhost:3306/allm?charset=utf8mb4")
|
||
|
||
from th_agenter.db.database import AsyncSessionFactory
|
||
from th_agenter.services.user import UserService
|
||
from utils.util_schemas import UserCreate
|
||
|
||
|
||
async def main():
|
||
async with AsyncSessionFactory() as session:
|
||
svc = UserService(session)
|
||
exists = await svc.get_user_by_email("admin@example.com")
|
||
if exists:
|
||
print("admin@example.com 已存在，跳过创建")
|
||
return
|
||
user = await svc.create_user(UserCreate(
|
||
username="admin",
|
||
email="admin@example.com",
|
||
password="admin123",
|
||
full_name="Admin",
|
||
))
|
||
print(f"已创建管理员: {user.username} / admin@example.com / admin123")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(main())
|