56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""数据迁移：归一化 users.role 字段。
|
||
|
|
|
||
|
|
把旧的角色值（enterprise/provider/government/investor/developer/service 等）
|
||
|
|
归一化为三种账号类型之一（operator/carrier/opc_member）。
|
||
|
|
|
||
|
|
用法：
|
||
|
|
cd code/server-core
|
||
|
|
uv run python scripts/db/normalize_user_roles.py
|
||
|
|
"""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
import sys
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
sys.path.insert(0, str(Path(__file__).resolve().parents[2]))
|
||
|
|
|
||
|
|
from sqlalchemy import select, update
|
||
|
|
from app.infrastructure.db import AsyncSessionLocal
|
||
|
|
from app.infrastructure.models import User
|
||
|
|
from app.domain.account_types import account_type
|
||
|
|
|
||
|
|
|
||
|
|
async def migrate() -> None:
|
||
|
|
async with AsyncSessionLocal() as session:
|
||
|
|
users = (await session.scalars(select(User))).all()
|
||
|
|
stats = {"total": 0, "changed": 0, "unchanged": 0}
|
||
|
|
changes = []
|
||
|
|
|
||
|
|
for u in users:
|
||
|
|
stats["total"] += 1
|
||
|
|
atype = account_type(u.role)
|
||
|
|
if u.role != atype:
|
||
|
|
changes.append((u.id, u.username, u.role, atype))
|
||
|
|
u.role = atype
|
||
|
|
stats["changed"] += 1
|
||
|
|
else:
|
||
|
|
stats["unchanged"] += 1
|
||
|
|
|
||
|
|
await session.commit()
|
||
|
|
|
||
|
|
print(f"归一化完成: {stats}")
|
||
|
|
print(f" - 总用户: {stats['total']}")
|
||
|
|
print(f" - 已变更: {stats['changed']}")
|
||
|
|
print(f" - 未变更: {stats['unchanged']}")
|
||
|
|
if changes:
|
||
|
|
print("\n变更明细:")
|
||
|
|
for uid, username, old, new in changes:
|
||
|
|
print(f" {uid} ({username}): {old} -> {new}")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
asyncio.run(migrate())
|