55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
|
|
#!/usr/bin/env python3
|
||
|
|
"""移除 test_item 表的 raw_data_fields 字段"""
|
||
|
|
|
||
|
|
import asyncio
|
||
|
|
from sqlalchemy.ext.asyncio import create_async_engine
|
||
|
|
from sqlalchemy import text
|
||
|
|
|
||
|
|
DB_CONFIG = {
|
||
|
|
'host': '123.57.81.127',
|
||
|
|
'port': 3306,
|
||
|
|
'user': 'cpy_admin',
|
||
|
|
'password': 'Tgzz2025+',
|
||
|
|
'database': 'ruoyi-fastapi'
|
||
|
|
}
|
||
|
|
|
||
|
|
DATABASE_URL = (
|
||
|
|
f"mysql+asyncmy://{DB_CONFIG['user']}:{DB_CONFIG['password']}@"
|
||
|
|
f"{DB_CONFIG['host']}:{DB_CONFIG['port']}/{DB_CONFIG['database']}"
|
||
|
|
)
|
||
|
|
|
||
|
|
async def remove_field():
|
||
|
|
engine = create_async_engine(DATABASE_URL, echo=False)
|
||
|
|
|
||
|
|
async with engine.connect() as conn:
|
||
|
|
print("=" * 80)
|
||
|
|
print("Removing raw_data_fields from test_item")
|
||
|
|
print("=" * 80)
|
||
|
|
|
||
|
|
# 检查字段是否存在
|
||
|
|
result = await conn.execute(text("DESCRIBE test_item"))
|
||
|
|
columns = result.fetchall()
|
||
|
|
column_names = [col[0] for col in columns]
|
||
|
|
|
||
|
|
if 'raw_data_fields' in column_names:
|
||
|
|
print("\n[1] Dropping raw_data_fields column...")
|
||
|
|
await conn.execute(text("ALTER TABLE test_item DROP COLUMN raw_data_fields"))
|
||
|
|
await conn.commit()
|
||
|
|
print(" [OK] raw_data_fields removed")
|
||
|
|
else:
|
||
|
|
print("\n[1] raw_data_fields does not exist")
|
||
|
|
|
||
|
|
# 验证
|
||
|
|
print("\n[2] Verification:")
|
||
|
|
result = await conn.execute(text("DESCRIBE test_item"))
|
||
|
|
columns = result.fetchall()
|
||
|
|
for col in columns:
|
||
|
|
print(f" {col[0]:<20} | {col[1]:<30}")
|
||
|
|
|
||
|
|
print("\n" + "=" * 80)
|
||
|
|
|
||
|
|
await engine.dispose()
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
asyncio.run(remove_field())
|