new branch
This commit is contained in:
39
backend/scripts/create_admin.py
Normal file
39
backend/scripts/create_admin.py
Normal file
@@ -0,0 +1,39 @@
|
||||
import asyncio
|
||||
import sys
|
||||
|
||||
sys.path.insert(0, ".")
|
||||
|
||||
from app.db.session import async_session_factory
|
||||
from app.models.user import User
|
||||
from app.core.security import get_password_hash
|
||||
|
||||
|
||||
async def create_admin():
|
||||
async with async_session_factory() as session:
|
||||
# 检查用户是否已存在
|
||||
from sqlalchemy import select
|
||||
|
||||
result = await session.execute(select(User).where(User.username == "linkong"))
|
||||
existing_user = result.scalar_one_or_none()
|
||||
|
||||
if existing_user:
|
||||
print(f"用户 linkong 已存在,更新密码...")
|
||||
existing_user.set_password("LK12345678")
|
||||
existing_user.role = "super_admin"
|
||||
else:
|
||||
print("创建管理员用户...")
|
||||
user = User(
|
||||
username="linkong",
|
||||
email="linkong@example.com",
|
||||
password_hash=get_password_hash("LK12345678"),
|
||||
role="super_admin",
|
||||
is_active=True,
|
||||
)
|
||||
session.add(user)
|
||||
|
||||
await session.commit()
|
||||
print("完成!")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(create_admin())
|
||||
Reference in New Issue
Block a user