FastAPI-Users 项目教程：如何通过编程方式创建用户-CSDN博客

本文链接：https://blog.csdn.net/gitblog_00383/article/details/148506084

FastAPI-Users 项目教程：如何通过编程方式创建用户

前言

在 FastAPI-Users 项目中，我们通常会通过 REST API 端点来创建用户。但在某些场景下，我们需要在代码中以编程方式直接创建用户，比如初始化脚本、测试用例或后台任务等。本文将详细介绍如何在 FastAPI-Users 项目中实现这一功能。

理解核心概念

1. 依赖注入与手动管理

FastAPI 的强大之处在于其依赖注入系统，它能自动处理各种依赖关系。但在编程方式创建用户时，我们处于依赖注入系统之外，需要手动管理所有依赖对象的生命周期。

2. UserManager 类

UserManager 是 FastAPI-Users 中负责用户管理的核心类，它封装了用户创建、验证等操作。在常规 API 调用中，FastAPI 会自动为我们实例化这个类，但在编程方式中我们需要手动处理。

实现步骤详解

第一步：将依赖转换为上下文管理器

在 FastAPI 中，依赖通常定义为生成器函数（使用 yield）。为了在普通代码中使用这些依赖，我们需要将它们转换为上下文管理器。

from contextlib import asynccontextmanager
from fastapi_users.manager import UserManager
from .dependencies import get_user_db, get_user_manager

# 将依赖转换为上下文管理器
get_user_db_context = asynccontextmanager(get_user_db)
get_user_manager_context = asynccontextmanager(get_user_manager)

这种转换使我们能够使用更直观的 with...as 语法来管理资源。

第二步：编写用户创建函数

现在我们可以编写一个完整的用户创建函数。这个函数需要：

手动管理所有依赖的生命周期
正确处理数据库连接
使用 UserManager 创建用户

async def create_user(email: str, password: str, is_superuser: bool = False):
    async with get_user_db_context() as user_db:
        async with get_user_manager_context(user_db) as user_manager:
            user = await user_manager.create(
                {
                    "email": email,
                    "password": password,
                    "is_superuser": is_superuser
                }
            )
            print(f"User created: {user.id}")
            return user

第三步：处理额外依赖

如果你的项目中有额外的依赖（如自定义的 UserManager 或数据库配置），需要确保在上下文管理器中正确传递这些依赖：

async def create_user_with_custom_deps(email: str, password: str, custom_setting: str):
    async with get_db_context(custom_setting) as db:
        async with get_user_db_context(db) as user_db:
            async with get_user_manager_context(user_db, other_dep=value) as user_manager:
                return await user_manager.create(
                    {"email": email, "password": password}
                )

实际应用示例

在脚本中使用

import asyncio

async def main():
    user = await create_user(
        email="admin@example.com",
        password="securepassword",
        is_superuser=True
    )
    print(f"Created admin user with ID: {user.id}")

if __name__ == "__main__":
    asyncio.run(main())

在测试中使用

import pytest

@pytest.mark.asyncio
async def test_user_creation():
    user = await create_user("test@example.com", "testpass")
    assert user.email == "test@example.com"