Docker部署FastAPI应用，如何在容器启动时创建初始数据库用户？

问题背景 #

我用Docker构建了基于FastAPI的Web应用，包含web和db两个容器：

• db容器启动时会创建web_dev和web_test数据库，但数据表由web容器启动时的迁移操作生成
• 应用中创建用户的接口受保护，需要在容器启动完成后自动创建第一个初始用户

最佳实现方案：在Web容器的启动流程中添加初始化步骤 #

因为数据表是Web容器执行迁移后才存在的，所以不能放在DB容器的初始化脚本里（此时表还未创建），应该在Web容器中，数据库迁移完成后、启动Uvicorn服务前执行创建初始用户的操作。

具体步骤 #

1. 编写初始用户创建脚本
   在项目根目录创建create_initial_user.py（路径可自定义），示例代码如下：

import os
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from app.models import User  # 替换为你的User模型实际路径
from app.core.security import get_password_hash  # 替换为你的密码哈希实现

# 从环境变量读取数据库连接信息
DATABASE_URL = os.getenv("DATABASE_URL")

engine = create_engine(DATABASE_URL)
SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)

def create_initial_user():
    db = SessionLocal()
    try:
        # 检查初始用户是否已存在，避免重复创建
        existing_user = db.query(User).filter(User.email == "admin@example.com").first()
        if not existing_user:
            hashed_pwd = get_password_hash("admin123")  # 替换为你的初始密码
            new_user = User(
                email="admin@example.com",
                hashed_password=hashed_pwd,
                is_superuser=True
                # 补充你的User模型所需的其他字段
            )
            db.add(new_user)
            db.commit()
            print("✅ 初始用户创建成功")
        else:
            print("ℹ️ 初始用户已存在，跳过创建步骤")
    finally:
        db.close()

if __name__ == "__main__":
    create_initial_user()

2. 修改Web容器的start脚本
   更新compose/local/start脚本，增加迁移执行和用户初始化步骤：

#!/bin/bash

set -o errexit
set -o pipefail
set -o nounset

# 执行数据库迁移（替换为你实际使用的迁移命令，这里以Alembic为例）
alembic upgrade head

# 执行初始用户创建脚本
python create_initial_user.py

# 启动FastAPI服务
uvicorn app.main:app --reload --reload-dir app --host 0.0.0.0

3. 确保脚本可访问（可选）
   如果将create_initial_user.py放在compose/local/目录下，需要在Web容器的Dockerfile中添加复制命令：

COPY ./compose/local/create_initial_user.py /app/create_initial_user.py

为什么不适合放在DB容器？ #

DB容器/docker-entrypoint-initdb.d/下的脚本仅在**数据库第一次初始化（数据卷为空）**时执行，且此时Web容器尚未执行迁移，数据表不存在，直接执行插入用户的SQL会报错。而Web容器的启动流程是在DB就绪后执行，迁移完成后表已存在，完全适配用户初始化的时机。

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