Neo
779b2f6d52
- 清理 1155 个已删除的历史文件（废弃 prompt_logs、tmp、旧 ops 脚本） - export/ 数据文件从 git 移除（已在 .gitignore） - demo-miniprogram 从 tmp/ 移入 apps/，添加 CLAUDE.md 注解 - DDL 合并：完整 schema 定义填充到 db/*/schemas/（从 docs/database/ddl/ 复制） - 39 个 v1 迁移脚本归档到 db/_archived/migrations_v1_merged/ - 4 个迁移变更类 BD_Manual 文档归档到 docs/database/_archived/ - .gitignore 补充 .vite/ 和 apps/*.zip - settings.json 添加 effortLevel 默认配置 - scripts/ops/ 新增运维脚本入库 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
"""一次性迁移：admin_users 表新增 roles 字段。"""
|
||
import os
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
from dotenv import load_dotenv
|
||
|
||
load_dotenv(Path(__file__).resolve().parents[2] / ".env")
|
||
|
||
dsn = os.environ.get("TEST_APP_DB_DSN") or os.environ.get("APP_DB_DSN", "").replace("zqyy_app", "test_zqyy_app")
|
||
if not dsn:
|
||
print("ERROR: 无法获取测试库 DSN", file=sys.stderr)
|
||
sys.exit(1)
|
||
|
||
import psycopg2
|
||
|
||
conn = psycopg2.connect(dsn)
|
||
try:
|
||
with conn.cursor() as cur:
|
||
cur.execute("""
|
||
ALTER TABLE admin_users
|
||
ADD COLUMN IF NOT EXISTS roles text[] NOT NULL DEFAULT '{site_admin}';
|
||
""")
|
||
cur.execute("COMMENT ON COLUMN admin_users.roles IS '用户角色列表，如 site_admin / tenant_admin';")
|
||
conn.commit()
|
||
print("OK: admin_users.roles 列已添加")
|
||
|
||
# 验证
|
||
with conn.cursor() as cur:
|
||
cur.execute("SELECT id, username, roles FROM admin_users LIMIT 5;")
|
||
for row in cur.fetchall():
|
||
print(f" id={row[0]} username={row[1]} roles={row[2]}")
|
||
finally:
|
||
conn.close()
|