28 lines
705 B
Python
28 lines
705 B
Python
"""将 auth.users.status 列的 DEFAULT 值从 'pending' 改为 'new'"""
|
|
import os
|
|
import psycopg2
|
|
from dotenv import load_dotenv
|
|
|
|
load_dotenv()
|
|
dsn = os.environ["APP_DB_DSN"]
|
|
conn = psycopg2.connect(dsn)
|
|
conn.autocommit = True
|
|
cur = conn.cursor()
|
|
|
|
print(f"连接: {dsn.split('@')[1] if '@' in dsn else dsn}")
|
|
|
|
cur.execute("ALTER TABLE auth.users ALTER COLUMN status SET DEFAULT 'new'")
|
|
print("已将 auth.users.status DEFAULT 改为 'new'")
|
|
|
|
# 验证
|
|
cur.execute("""
|
|
SELECT column_default
|
|
FROM information_schema.columns
|
|
WHERE table_schema = 'auth' AND table_name = 'users' AND column_name = 'status'
|
|
""")
|
|
row = cur.fetchone()
|
|
print(f"验证 DEFAULT: {row[0]}")
|
|
|
|
conn.close()
|
|
print("完成")
|