FastAPI（六十五）实战开发《在线课程学习系统》接口开发--用户注册接口开发

from pydantic import BaseModel
from typing import Optional

class UserBase(BaseModel):
    username: str


class UserCreate(UserBase):
    """
    请求模型验证：
    username:
    password:
    """
    password: str
    role: int
    jobnum: Optional[int] = None
    studentnum: Optional[int] = None
    sex: str = "男"
    age: int

from sqlalchemy.orm import Session
from models.models import *
from models.schemas import *
def get_user(db: Session, user_id: int):
    return db.query(User).filter(User.id == user_id,User.status==False).first()
# 新建用户
def db_create_user(db: Session, user: UserCreate):
    roles = db.query(Role).filter(Role.name == user.role).first()
    db_user = User(**user.dict())
    db_user.role=roles.id
    db.add(db_user)
    db.commit()  # 提交保存到数据库中
    db.refresh(db_user)  # 刷新
    return db_user

def get_user_username(db: Session, username: str):
    return db.query(User).filter(User.username == username,User.status==False).first()

1.校验参数是否合规
2.查询用户名是否存在
3.密码加密
4.保存到数据库

from fastapi import APIRouter, Request
from fastapi import Depends, HTTPException, Header
from models.crud import *
from models.get_db import get_db
from jose import JWTError, jwt
from passlib.context import CryptContext
from config import *
from  common.jsontools import *
from  common.logs import logger
usersRouter = APIRouter()
pwd_context = CryptContext(schemes=["bcrypt"], deprecated="auto")

def verify_password(plain_password, hashed_password):
    return pwd_context.verify(plain_password, hashed_password)

def get_password_hash(password):
    return pwd_context.hash(password)
# 新建用户
@usersRouter.post("/create", tags=["users"])
def create_user(user: UserCreate, db: Session = Depends(get_db)):
    
    logger.info("创建用户")
    if len(user.username)<8 or len(user.username)>16:
        return reponse(code=100106,message="用户名长度应该是8-16位",data="")
    if user.age<18:
        return reponse(code=100103, message="年纪大小不符合", data="")
    if (user.role == "学生" and user.studentnum is None) or (user.role == "教师" and  user.jobnum is None) or (
            user.role not in ["教师", '学生']):
        return reponse(code=100102, message="身份和对应号不匹配", data="")
    db_crest = get_user_username(db, user.username)
    if db_crest:
        return reponse(code=100104, message="用户名重复", data="")
    try:
        user.password = get_password_hash(user.password)
    except Exception as e:
        logger.exception(e)
        return reponse(code=100105, data="", message="密码加密失败")
    try:
        user=db_create_user(db=db, user=user)
        logger.success("创建用户成功")
        return reponse(code=200,data={'user':user.username},message="success")
    except Exception as e:
        logger.exception(e)
        return reponse(code=100101, data="", message="注册失败")

from fastapi import FastAPI
from routers.user import usersRouter
app = FastAPI()
app.include_router(usersRouter, prefix="/user", tags=['users'])

if __name__ == '__main__':
    import uvicorn
    uvicorn.run(app='main:app', host="127.0.0.1", port=8000, reload=True, debug=True)