### 在 FastAPI 应用中通过数据库迁移或脚本添加默认角色 为了实现将默认角色（如 `superuser` 和 `adminuser`）一次性添加到角色表中的需求，可以采用以下方法： #### 方法一：利用 Alembic 数据库迁移工具 Alembic 是一种常用的 SQLAlchemy 数据库迁移工具。可以通过创建一个新的迁移版本，在升级过程中执行 SQL 或 Python 脚本来初始化数据。 1. **定义模型** 首先确保在 FastAPI 中已经定义了一个用于存储角色的角色表模型。例如： ```python from sqlalchemy import Column, Integer, String from sqlalchemy.ext.declarative import declarative_base Base = declarative_base() class Role(Base): __tablename__ = "roles" id = Column(Integer, primary_key=True, index=True) name = Column(String, unique=True, nullable=False) ``` 2. **创建新的迁移文件** 使用 Alembic 创建一个新的迁移文件，并在其中写入插入默认角色的逻辑。 ```bash alembic revision --autogenerate -m "Add default roles" ``` 3. **编辑迁移文件** 修改生成的迁移文件，加入插入默认角色的代码。例如： ```python def upgrade(): # 添加默认角色 op.bulk_insert( "roles", [ {"name": "superuser"}, {"name": "adminuser"} ] ) def downgrade(): # 删除默认角色 op.execute("DELETE FROM roles WHERE name IN ('superuser', 'adminuser')") ``` 4. **运行迁移** 将应用程序部署到 Debian 后，运行以下命令完成迁移： ```bash alembic upgrade head ``` --- #### 方法二：使用自定义脚本 如果不想依赖于 Alembic，则可以直接编写一个独立的 Python 脚本，手动向数据库中插入默认角色。 1. **编写脚本** 编写一个名为 `add_default_roles.py` 的脚本，内容如下： ```python from sqlalchemy.orm import sessionmaker from sqlalchemy import create_engine from models import Role # 假设模型已导入 DATABASE_URL = "postgresql://username:password@localhost/db_name" engine = create_engine(DATABASE_URL) SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine) def add_default_roles(): db = SessionLocal() try: superuser_role = Role(name="superuser") adminuser_role = Role(name="adminuser") db.add(superuser_role) db.add(adminuser_role) db.commit() # 提交事务 except Exception as e: db.rollback() raise e finally: db.close() if __name__ == "__main__": add_default_roles() ``` 2. **运行脚本** 在 Debian 上部署完成后，运行该脚本即可： ```bash python add_default_roles.py ``` --- #### 方法三：结合环境变量与启动钩子 可以在应用启动时自动检测并初始化默认角色。这种方法适合动态环境中频繁部署的应用程序。 1. **修改主入口文件** 在 FastAPI 主入口文件中增加一段初始化逻辑： ```python from fastapi import FastAPI from sqlalchemy.orm import sessionmaker from sqlalchemy import create_engine from models import Role app = FastAPI() DATABASE_URL = "postgresql://username:password@localhost/db_name" engine = create_engine(DATABASE_URL) SessionLocal = sessionmaker(bind=engine) @app.on_event("startup") async def startup_event(): db = SessionLocal() existing_roles = {role.name for role in db.query(Role).all()} required_roles = {"superuser", "adminuser"} missing_roles = required_roles - existing_roles if missing_roles: new_roles = [Role(name=name) for name in missing_roles] db.add_all(new_roles) db.commit() ``` 2. **注意事项** 此方法会在每次应用重启时检查是否存在缺失的角色，并补充它们。因此适用于需要高可用性和一致性的场景[^3]。 --- ### 总结 以上三种方式各有优劣，具体选择取决于实际需求和技术栈偏好。推荐优先考虑基于 Alembic 的解决方案，因为它更符合现代开发实践，能够更好地管理数据库结构变更和初始数据加载过程。