feat 订阅支持选择站点范围

alembic/gen.py

@@ -0,0 +1,19 @@
+import importlib
+from pathlib import Path
+
+from alembic.config import Config as AlembicConfig
+from alembic.command import revision as alembic_revision
+
+from app.core.config import settings
+
+# 导入模块，避免建表缺失
+for module in Path(__file__).with_name("models").glob("*.py"):
+    importlib.import_module(f"app.db.models.{module.stem}")
+
+db_version = input("请输入版本号：")
+db_location = settings.CONFIG_PATH / 'user.db'
+script_location = settings.ROOT_PATH / 'alembic'
+alembic_cfg = AlembicConfig()
+alembic_cfg.set_main_option('script_location', str(script_location))
+alembic_cfg.set_main_option('sqlalchemy.url', f"sqlite:///{db_location}")
+alembic_revision(alembic_cfg, db_version, True)

alembic/versions/9f4edd55c2d4_1_0_0.py

@@ -0,0 +1,32 @@
+"""1.0.0
+
+Revision ID: 9f4edd55c2d4
+Revises: 
+Create Date: 2023-07-13 12:27:26.402317
+
+"""
+from alembic import op
+import sqlalchemy as sa
+
+
+# revision identifiers, used by Alembic.
+revision = '9f4edd55c2d4'
+down_revision = None
+branch_labels = None
+depends_on = None
+
+
+def upgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    try:
+        with op.batch_alter_table("subscribe") as batch_op:
+            batch_op.add_column(sa.Column('sites', sa.Text, nullable=True))
+    except Exception as e:
+        pass
+    # ### end Alembic commands ###
+
+
+def downgrade() -> None:
+    # ### commands auto generated by Alembic - please adjust! ###
+    pass
+    # ### end Alembic commands ###