如何使用Flask-Migrate迁移时隔离两个本地PostgreSQL数据库

出现这个问题的核心原因是当前环境变量的DATABASE_URL没有指定具体数据库名，导致Flask-Migrate连接到了默认数据库（大概率是databaseA），或者项目B意外加载了项目A的模型。以下是具体解决方法：

1. 明确指定项目B的数据库连接地址 #

方法一：在项目B的配置中直接指定 #

在项目B的Flask配置文件（比如config.py）或app初始化代码里，硬编码数据库连接（适合测试环境，生产环境推荐用环境变量）：

app.config['SQLALCHEMY_DATABASE_URI'] = 'postgresql://localhost/databaseB'

注意：如果你的PostgreSQL需要用户名密码，补全格式为：postgresql://username:password@localhost/databaseB

方法二：临时覆盖环境变量执行命令 #

在执行migrate命令前，临时设置当前会话的环境变量，优先级高于全局环境变量：

• Linux/macOS终端：

  DATABASE_URL=postgresql://localhost/databaseB flask db migrate -m "Initial migration"
  

• Windows命令提示符（cmd）：

  set DATABASE_URL=postgresql://localhost/databaseB && flask db migrate -m "Initial migration"
  

• Windows PowerShell：

  $env:DATABASE_URL="postgresql://localhost/databaseB"; flask db migrate -m "Initial migration"
  

方法三：用.env文件管理项目专属环境变量 #

适合长期维护，避免每次手动设置：

1. 安装python-dotenv：

   pip install python-dotenv
   

2. 在项目B的根目录创建.env文件，写入：

   DATABASE_URL=postgresql://localhost/databaseB
   

3. 在项目B的app入口文件（比如app.py）开头添加：

   from dotenv import load_dotenv
   load_dotenv()  # 加载当前目录的.env文件
   

之后执行flask db命令时，会自动读取这个专属的环境变量，不受全局环境变量影响。

2. 检查是否意外导入了项目A的模型 #

Flask-Migrate是根据当前Flask应用加载的SQLAlchemy模型生成迁移脚本的，如果项目B的代码里不小心导入了项目A的模型，就会把A的表操作也包含进来。检查所有import语句，确保没有引入项目A的模型文件。

3. 验证数据库连接是否正确 #

可以在项目B里加一段测试代码，确认当前连接的是databaseB：

from sqlalchemy import create_engine
from your_app_module import app  # 替换成你的app模块名

engine = create_engine(app.config['SQLALCHEMY_DATABASE_URI'])
with engine.connect() as conn:
    result = conn.execute("SELECT current_database();")
    print("当前连接的数据库：", result.fetchone()[0])

运行这段代码，如果输出是databaseB，说明连接正确。

内容的提问来源于stack exchange，提问作者JGV