FastAPI（六十七）实战开发《在线课程学习系统》接口开发--用户 个人信息接口开发

def get_role_name(db:Session,id:id):
    return db.query(Role).filter(Role.id == id).first()

async def get_cure_user(request: Request, token: Optional[str] = Header(...), db: Session = Depends(get_db)) -> UsernameRole:
    credentials_exception = HTTPException(
        status_code=status.HTTP_401_UNAUTHORIZED,
        detail="验证失败"
    )
    credentials_FOR_exception = HTTPException(
        status_code=status.HTTP_403_FORBIDDEN,
        detail="用户未登录或者登陆token已经失效"
    )
    try:
        payload = jwt.decode(token, SECRET_KEY, algorithms=[ALGORITHM])
        username: str = payload.get("sub")
        if username is None:
            raise credentials_exception
        useris = await request.app.state.redis.get(username)
        if not useris and useris!=token:
            raise credentials_FOR_exception
        userrole = get_role_name(db,get_user_username(db, username).role).name
        user = UsernameRole(username=username,role=userrole)
        return user
    except JWTError:
        logger.error(traceback.format_exc())
        raise credentials_exception

class UsernameRole(UserBase):
    role:str

@usersRouter.get(path='/getcuruser',response_model=UserBase)
async def getcuruser(user:UsernameRole= Depends(get_cure_user),db: Session = Depends(get_db)):
    user_name=get_user_username(db,username=user.username)
    data={}
    data['username']=user_name.username
    data['sex']=user_name.sex
    data['age']=user_name.age
    if user.role=="学生":
        data['studentnum']=user_name.studentnum
    else:
        data['jobnum'] = user_name.jobnum
    return reponse(code=200,message='成功',data=data)