mirror of
https://github.com/langbot-app/LangBot.git
synced 2026-06-02 12:05:54 +00:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import sqlalchemy
|
||
from .. import migration
|
||
|
||
|
||
@migration.migration_class(6)
|
||
class DBMigratePluginInstallSource(migration.DBMigration):
|
||
"""插件安装来源"""
|
||
|
||
async def upgrade(self):
|
||
"""升级"""
|
||
# 查询表结构获取所有列名（异步执行 SQL）
|
||
result = await self.ap.persistence_mgr.execute_async(sqlalchemy.text('PRAGMA table_info(plugin_settings);'))
|
||
# fetchall() 是同步方法，无需 await
|
||
columns = [row[1] for row in result.fetchall()]
|
||
|
||
# 检查并添加 install_source 列
|
||
if 'install_source' not in columns:
|
||
await self.ap.persistence_mgr.execute_async(
|
||
sqlalchemy.text(
|
||
"ALTER TABLE plugin_settings ADD COLUMN install_source VARCHAR(255) NOT NULL DEFAULT 'github'"
|
||
)
|
||
)
|
||
|
||
# 检查并添加 install_info 列
|
||
if 'install_info' not in columns:
|
||
await self.ap.persistence_mgr.execute_async(
|
||
sqlalchemy.text("ALTER TABLE plugin_settings ADD COLUMN install_info JSON NOT NULL DEFAULT '{}'")
|
||
)
|
||
|
||
async def downgrade(self):
|
||
"""降级"""
|
||
pass
|