fastapi jwt 部分

安装需要的包

# 安装jwt及pydantic相关依赖包
pip install passlib
pip install pydantic[email]
pip install python-multipart
pip install bcrypt
pip install jwt
pip install python-jose

哈希并校验密码

# 1、创建对象，进行哈希和校验密码
from passlib.context import CryptContext
# 使用的算法是Bcrypt
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")


# 2、哈希密码：hash(password)
def get_password_hash(password):
    """
    哈希来自用户的密码
    :param password: 原密码
    :return: 哈希后的密码
    """
    return pwd_context.hash(password)
# 哈希后的密码
# $2b$12$sErK932BEaLyIisz30PubepN7w91RLwkISWbAFYgUgoIqh8goJLEW

# 3、校验密码：verify(plain_password, hashed_password)

def verify_password(plain_password, hashed_password):
    """
    校验接收的密码是否与存储的哈希值匹配
    :param plain_password: 原密码
    :param hashed_password: 哈希后的密码
    :return: 返回值为bool类型，校验成功返回True,反之False
    """
    return pwd_context.verify(plain_password, hashed_password)

JWT令牌验证加密相关配置，及数据模型创建

###############
# JWT令牌验证加密相关配置，及数据模型创建
###############
# 1、生成一个随机的密钥，用于对JWT令牌进行签名 windos 或linux 操作系统环境下执行
# openssl rand -hex 32
# 4a2ca2b5dac83fcfcdc97aeab073ee4bf9f99c3e116345e69c874941da247027
# 2、创建密钥变量
SECRET_KEY = "4a2ca2b5dac83fcfcdc97aeab073ee4bf9f99c3e116345e69c874941da247027"
# 3、创建用于设定JWT令牌签名算法的变量 -- 算法
ALGORITHM = "HS256"
# 4、创建设置令牌过期时间变量（单位：分钟）
ACCESS_TOKEN_EXPIRE_MINUTES = 30

# 5、模拟数据库用户信息
users_db = {
    "zhang san":{
        "id": 1,
        "username": "zhang san",
        "email": "zhang_san@user.com",
        "hashed_password": "$2b$12$nmpXZ.WdL3fbSDMC2UaAWuOK5f4vf08Wx71bgnQxJUsIt3qKKzuKe",
        "disabled": False,
    },
    "li si":{
        "id": 2,
        "username": "li si",
        "email": "li_si@user.com",
        "hashed_password": "$2b$12$nmpXZ.WdL3fbSDMC2UaAWuOK5f4vf08Wx71bgnQxJUsIt3qKKzuKe",
        "disabled": True,
    }
}

# 6、创建所需模型类

from typing import Optional,Union
from pydantic import BaseModel, EmailStr

# 1) 用户模型基类
class UserBase(BaseModel):
    id: Optional[int] = False
    username: str = None
    # pip install pydantic[email] 使用email验证的时候需要增加这个库
    email: Optional[EmailStr] = None
    disabled: Union[bool, None] = None

# 2）登录成功返回token模型
class TokenModel(BaseModel):
    access_token: str = None
    token_type:Union[str, None] = None

# 3）用户密码
class UserInDB(UserBase):
    hashed_password: str

查看登录用户是否在数据库，及创建用户token

#################
# 查看登录用户是否在数据库，及创建用户token
#################
from datetime import datetime, timed