21 lines
816 B
Python
21 lines
816 B
Python
from db import get_session
|
|
from models import User
|
|
from auth import get_password_hash
|
|
from config import DEFAULT_ADMIN_USERNAME, DEFAULT_ADMIN_PASSWORD
|
|
|
|
def init_default_admin():
|
|
"""创建默认管理员账号(如果不存在)"""
|
|
with get_session() as s:
|
|
admin = s.query(User).filter(User.username == DEFAULT_ADMIN_USERNAME).first()
|
|
if not admin:
|
|
admin = User(
|
|
username=DEFAULT_ADMIN_USERNAME,
|
|
hashed_password=get_password_hash(DEFAULT_ADMIN_PASSWORD),
|
|
is_admin=True,
|
|
is_active=True
|
|
)
|
|
s.add(admin)
|
|
s.commit()
|
|
print(f"✓ 创建默认管理员: {DEFAULT_ADMIN_USERNAME}")
|
|
else:
|
|
print(f"✓ 管理员账号已存在: {DEFAULT_ADMIN_USERNAME}") |