构建模块化的FastAPI应用: 从用户认证到角色控制

实现了用户身份验证及角色授权的基本功能。具体来说,当用户尝试访问某些资源时,系统会首先验证用户的身份,然后根据用户的角色来决定是否允许访问特定资源。例如,普通用户只能访问自己的信息,而管理员可以访问额外的管理界面。这种机制保证了系统的安全性,并且可以根据需要灵活地扩展不同的角色和权限。
在这里插入图片描述

  1. 用户模型定义:定义了一个用户模型 User,包含用户名和角色信息。此外还定义了一个 UserInDB 模型,包含用户密码,用于表示数据库中的用户信息。

  2. 用户身份验证:实现了一个依赖项 get_current_user,通过请求头部的 token 字段来确定当前用户的身份。如果 token 对应的用户存在于模拟数据库 FAKE_USERS_DB 中,则返回该用户对象;否则抛出异常,提示用户不存在或 Token 无效。

  3. 管理员权限检查:定义了一个依赖项 get_admin_user,基于 get_current_user 函数来进一步确认当前用户是否具有管理员权限。如果不是管理员,则抛出异常,提示需要管理员权限。

  4. 用户相关API:在 users 路由模块中定义了几个与用户相关的路径操作。这些操作都需要通过 get_current_user 依赖项来验证当前用户的合法性。例如,read_users 方法返回当前用户的列表,read_user_me 方法返回当前用户的信息,而 read_user 方法则返回指定用户名对应的用户信息,前提是当前用户与请求的用户名匹配。

  5. 管理员相关API:在 admin 路由模块中定义了管理员专属的路径操作。这些操作需要通过 get_admin_user 依赖项来确保只有管理员可以访问。例如,admin_dashboard 方法返回管理员的信息,只有管理员用户可以调用这个端点。

用户角色验证功能

在 FastAPI 中,通常会使用 Pydantic 模型来定义请求和响应的数据结构。为了实现一个简单的用户角色验证功能,我们可以创建一个模型来表示用户,并包含一个字段来存储用户的角色信息。然后,我们可以在依赖项中使用这个模型来进行角色检查。

首先,我们需要安装必要的依赖:

pip install fastapi pydantic uvicorn

接下来,我们更新文件结构并添加新的模型文件:

更新后的文件结构

.
├── app
│   ├── __init__.py
│   ├── main.py
│   ├── dependencies.py
│   ├── models.py  # 新增模型文件
│   ├── routers
│   │   ├── __init__.py
│   │   ├── items.py
│   │   └── users.py
│   └── internal
│       ├── __init__.py
│       └── admin.py

文件内容

  • app/models.py

    用户模型定义。

    from pydantic import BaseModel
    
    class User(BaseModel):
        username: str
        role: str  # 角色字段
    
    class 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值