Commit ca6a6b2d authored by xianyang's avatar xianyang

优化谷歌验证

parent 086dd8a3
...@@ -30,7 +30,7 @@ def login(form_data: UserLoginForm, db: Session = Depends(get_db)): ...@@ -30,7 +30,7 @@ def login(form_data: UserLoginForm, db: Session = Depends(get_db)):
@router.post("/goodleCode") @router.post("/goodleCode")
def goodle_code(data: GoogleCode, token=Depends(login_required), db: Session = Depends(get_db)): def goodle_code(data: GoogleCode, db: Session = Depends(get_db)):
user_data = get_user(db, data.username) user_data = get_user(db, data.username)
google_key = pyotp.random_base32(64) google_key = pyotp.random_base32(64)
google_img = get_qrcode(username=user_data.username, gtoken=google_key).decode('utf-8') google_img = get_qrcode(username=user_data.username, gtoken=google_key).decode('utf-8')
...@@ -38,7 +38,7 @@ def goodle_code(data: GoogleCode, token=Depends(login_required), db: Session = D ...@@ -38,7 +38,7 @@ def goodle_code(data: GoogleCode, token=Depends(login_required), db: Session = D
@router.post("/googleLogin") @router.post("/googleLogin")
def google_login(data: GoogleLogin, token=Depends(login_required), db: Session = Depends(get_db)): def google_login(data: GoogleLogin, db: Session = Depends(get_db)):
auth_info = authenticate_pwd(db=db, form_data=data) auth_info = authenticate_pwd(db=db, form_data=data)
if not auth_info.get("result"): if not auth_info.get("result"):
return HttpResultResponse(code=500, msg=auth_info.get("msg")) return HttpResultResponse(code=500, msg=auth_info.get("msg"))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment