fix(database): upgrade error in pg database

Co-Authored-By: Aqr-K <95741669+Aqr-K@users.noreply.github.com>

database/versions/e2dbe1421fa4_2_0_3.py

@@ -21,28 +21,58 @@ depends_on = None
 
 
 def upgrade() -> None:
-    # ### commands auto generated by Alembic - please adjust! ###
-    # 支持订阅自定义媒体类别和过滤规则组、自定义识别词
-    with contextlib.suppress(Exception):
+    conn = op.get_bind()
+    inspector = sa.inspect(conn)
+
+    # 检查并添加 downloadhistory.media_category
+    dh_columns = inspector.get_columns('downloadhistory')
+    if not any(c['name'] == 'media_category' for c in dh_columns):
         op.add_column('downloadhistory', sa.Column('media_category', sa.String(), nullable=True))
+
+    # 检查并添加 subscribe 表的列
+    sub_columns = inspector.get_columns('subscribe')
+    if not any(c['name'] == 'custom_words' for c in sub_columns):
         op.add_column('subscribe', sa.Column('custom_words', sa.String(), nullable=True))
+    if not any(c['name'] == 'media_category' for c in sub_columns):
         op.add_column('subscribe', sa.Column('media_category', sa.String(), nullable=True))
+    if not any(c['name'] == 'filter_groups' for c in sub_columns):
         op.add_column('subscribe', sa.Column('filter_groups', sa.JSON(), nullable=True))
-    # 将String转换为JSON类型
-    with contextlib.suppress(Exception):
-        op.alter_column('subscribe', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('downloadhistory', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('mediaserveritem', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('message', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('plugindata', 'value', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('site', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('sitestatistic', 'note', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('systemconfig', 'value', existing_type=sa.String(), type_=sa.JSON())
-        op.alter_column('userconfig', 'value', existing_type=sa.String(), type_=sa.JSON())
-    # 清空用户配置表中不兼容的数据
+
+    # 定义需要检查和转换的表和列
+    columns_to_alter = {
+        'subscribe': 'note',
+        'downloadhistory': 'note',
+        'mediaserveritem': 'note',
+        'message': 'note',
+        'plugindata': 'value',
+        'site': 'note',
+        'sitestatistic': 'note',
+        'systemconfig': 'value',
+        'userconfig': 'value'
+    }
+
+    for table, column_name in columns_to_alter.items():
+        try:
+            cols = inspector.get_columns(table)
+            # 找到对应的列信息
+            target_col = next((c for c in cols if c['name'] == column_name), None)
+            # 如果列存在且类型不是JSON，则进行修改
+            if target_col and not isinstance(target_col['type'], sa.JSON):
+                # PostgreSQL需要指定USING子句来处理类型转换
+                if conn.dialect.name == 'postgresql':
+                    op.alter_column(table, column_name,
+                                    existing_type=sa.String(),
+                                    type_=sa.JSON(),
+                                    postgresql_using=f'"{column_name}"::json')
+                else:
+                    op.alter_column(table, column_name,
+                                    existing_type=sa.String(),
+                                    type_=sa.JSON())
+        except Exception as e:
+            print(f"Could not alter column {column_name} in table {table}: {e}")
+
     with SessionFactory() as db:
         UserConfig.truncate(db)
-    # ### end Alembic commands ###
 
 
 def downgrade() -> None: