48 lines
1.9 KiB
Python
48 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""为已有数据库添加新列（traffic、countries）并将部分列改为可空。仅需运行一次。"""
|
||
import sys
|
||
from app import app, db
|
||
from sqlalchemy import text
|
||
|
||
def run():
|
||
with app.app_context():
|
||
engine = db.engine
|
||
dialect = engine.dialect.name
|
||
if dialect == "mysql":
|
||
with engine.connect() as conn:
|
||
for col, spec in [
|
||
("traffic", "VARCHAR(64) NULL"),
|
||
("countries", "VARCHAR(255) NULL"),
|
||
]:
|
||
try:
|
||
conn.execute(text("ALTER TABLE vps_plans ADD COLUMN {} {}".format(col, spec)))
|
||
conn.commit()
|
||
print("Added column:", col)
|
||
except Exception as e:
|
||
if "Duplicate column" in str(e) or "already exists" in str(e).lower():
|
||
print("Column", col, "already exists, skip.")
|
||
else:
|
||
print("Add", col, ":", e)
|
||
for col, spec in [
|
||
("region", "VARCHAR(128) NULL"),
|
||
("vcpu", "INT NULL"),
|
||
("memory_gb", "INT NULL"),
|
||
("storage_gb", "INT NULL"),
|
||
("bandwidth_mbps", "INT NULL"),
|
||
("price_cny", "DOUBLE NULL"),
|
||
("price_usd", "DOUBLE NULL"),
|
||
]:
|
||
try:
|
||
conn.execute(text("ALTER TABLE vps_plans MODIFY COLUMN {} {}".format(col, spec)))
|
||
conn.commit()
|
||
print("Modified column:", col)
|
||
except Exception as e:
|
||
print("Modify", col, ":", e)
|
||
else:
|
||
print("Only MySQL migration is implemented. For SQLite, recreate the DB or add columns manually.")
|
||
print("Done.")
|
||
|
||
if __name__ == "__main__":
|
||
run()
|
||
sys.exit(0)
|