签入版本

2026-09-16 12:00:58 +08:00
commit 20a2f846ff
842 changed files with 93035 additions and 0 deletions
@@ -0,0 +1,117 @@
+# +======================================================================
+# | 模块: Alembic 迁移环境
+# | 说明: 配置数据库连接与模型 metadata，支持离线/在线两种迁移模式
+# +======================================================================
+"""Alembic 迁移环境。
+
+- 数据库 URL 取自 src.config.database.DATABASE_URL
+- 自动聚合 src.modules.*.models 中注册的模型 metadata，支持 autogenerate
+"""
+
+# ----------------------------------------------------------------------
+# 导入依赖
+# 涵盖：模块搜索路径、日志配置、路径处理、SQLAlchemy 引擎与连接池、
+#       Alembic 上下文
+# ----------------------------------------------------------------------
+import sys
+from logging.config import fileConfig
+from pathlib import Path
+
+from sqlalchemy import create_engine, pool
+
+from alembic import context
+
+# ============================================================
+# 将 src 目录加入 sys.path
+# ============================================================
+# 将 src 目录加入 sys.path，确保可直接 import 项目代码
+ROOT = Path(__file__).resolve().parent.parent
+SRC = ROOT / "src"
+if str(SRC) not in sys.path:
+    sys.path.insert(0, str(SRC))
+
+from config.database import DATABASE_URL
+
+# ============================================================
+# 聚合全部模型 metadata
+# ============================================================
+# 聚合全部模型 metadata（导入各模块 models 即完成注册）
+from extensions.sqlalchemy import Base
+import modules.auth.models
+import modules.user.models
+import modules.user_role.models
+import modules.role.models
+import modules.role_menu.models
+import modules.menu.models
+import modules.dept.models
+import modules.category.models
+import modules.article.models
+import modules.param.models
+import modules.file_template.models
+import modules.job.models
+import modules.dictionary.dict.models
+import modules.dictionary.dict_item.models
+import modules.configuration.config.models
+import modules.configuration.config_item.models
+import modules.level.models
+import modules.position.models
+import modules.city.models
+import modules.notice.models
+import modules.link.models
+import modules.operation_log.models
+import modules.login_log.models
+import modules.job_log.models
+import modules.example.models
+
+# ============================================================
+# 读取 Alembic 配置
+# ============================================================
+# 读取 alembic.ini 中的配置
+config = context.config
+
+# 存在配置文件时加载其日志配置
+if config.config_file_name is not None:
+    fileConfig(config.config_file_name)
+
+# 目标 metadata（autogenerate 依据）
+target_metadata = Base.metadata
+
+# 数据库 URL
+database_url = DATABASE_URL
+
+
+# ======================================================================
+# 离线模式：仅生成 SQL 脚本
+# ======================================================================
+def run_migrations_offline() -> None:
+    """离线模式：仅生成 SQL 脚本。"""
+    context.configure(
+        url=database_url,
+        target_metadata=target_metadata,
+        literal_binds=True,
+        dialect_opts={"paramstyle": "named"},
+    )
+    with context.begin_transaction():
+        context.run_migrations()
+
+
+# ======================================================================
+# 在线模式：直接执行迁移
+# ======================================================================
+def run_migrations_online() -> None:
+    """在线模式：直接执行迁移。"""
+    # NullPool：迁移结束后即释放连接，无需连接池
+    connectable = create_engine(database_url, poolclass=pool.NullPool)
+    with connectable.connect() as connection:
+        context.configure(connection=connection, target_metadata=target_metadata)
+        with context.begin_transaction():
+            context.run_migrations()
+
+
+# ============================================================
+# 入口：按模式执行
+# ============================================================
+if context.is_offline_mode():
+    run_migrations_offline()
+else:
+    run_migrations_online()