fix: upgrade startup script controls

This commit is contained in:
linkong
2026-03-27 11:13:01 +08:00
parent 30a29a6e34
commit bf2c4a172d
7 changed files with 402 additions and 73 deletions

View File

@@ -60,6 +60,28 @@ async def seed_default_datasources(session: AsyncSession):
await session.commit()
async def ensure_default_admin_user(session: AsyncSession):
from app.core.security import get_password_hash
from app.models.user import User
result = await session.execute(
text("SELECT id FROM users WHERE username = 'admin'")
)
if result.fetchone():
return
session.add(
User(
username="admin",
email="admin@planet.local",
password_hash=get_password_hash("admin123"),
role="super_admin",
is_active=True,
)
)
await session.commit()
async def init_db():
import app.models.user # noqa: F401
import app.models.gpu_cluster # noqa: F401
@@ -68,6 +90,7 @@ async def init_db():
import app.models.datasource # noqa: F401
import app.models.datasource_config # noqa: F401
import app.models.alert # noqa: F401
import app.models.bgp_anomaly # noqa: F401
import app.models.collected_data # noqa: F401
import app.models.system_setting # noqa: F401
@@ -125,3 +148,4 @@ async def init_db():
async with async_session_factory() as session:
await seed_default_datasources(session)
await ensure_default_admin_user(session)