This commit is contained in:
@@ -77,8 +77,8 @@ def callback(code: str, session: Session = Depends(get_session)):
|
||||
response = requests.post(TOKEN_URL, data=data, headers=headers)
|
||||
if response.status_code != 200:
|
||||
raise HTTPException(
|
||||
status_code=400,
|
||||
detail=messages.failtogettoken
|
||||
status_code=404,
|
||||
detail=messages.Messages.not_found('token')
|
||||
)
|
||||
|
||||
token_data = response.json()
|
||||
@@ -154,26 +154,26 @@ def verify_token(token: str):
|
||||
)
|
||||
return decoded
|
||||
except jwt.ExpiredSignatureError:
|
||||
raise HTTPException(status_code=401, detail=messages.tokenexipired)
|
||||
raise HTTPException(status_code=401, detail=messages.Messages.tokenexipired)
|
||||
except jwt.InvalidTokenError:
|
||||
raise HTTPException(status_code=401, detail=messages.invalidtoken)
|
||||
raise HTTPException(status_code=401, detail=messages.Messages.invalidtoken)
|
||||
|
||||
|
||||
def get_current_user(request: Request, session: Session = Depends(get_session)):
|
||||
access_token = request.cookies.get('access_token')
|
||||
if not access_token:
|
||||
raise HTTPException(status_code=401, detail=messages.notauthenticated)
|
||||
raise HTTPException(status_code=401, detail=messages.Messages.notauthenticated)
|
||||
payload = verify_token(access_token)
|
||||
if not payload:
|
||||
raise HTTPException(status_code=401, detail='aze')
|
||||
email = payload.get('email')
|
||||
|
||||
if not email:
|
||||
raise HTTPException(status_code=401, detail=messages.notauthenticated)
|
||||
raise HTTPException(status_code=401, detail=messages.Messages.notauthenticated)
|
||||
|
||||
user = session.exec(select(User).where(User.email == email)).first()
|
||||
if not user:
|
||||
raise HTTPException(status_code=401, detail=messages.usernotfound)
|
||||
raise HTTPException(status_code=401, detail=messages.Messages.not_found('user'))
|
||||
return user
|
||||
|
||||
@router.post('/refresh')
|
||||
@@ -191,8 +191,8 @@ def refresh_token(refresh_token: Annotated[str | None, Cookie()] = None):
|
||||
result = requests.post(TOKEN_URL, data=data, headers=headers)
|
||||
if result.status_code != 200:
|
||||
raise HTTPException(
|
||||
status_code=400,
|
||||
detail=messages.failtogettoken
|
||||
status_code=404,
|
||||
detail=messages.Messages.not_found('token')
|
||||
)
|
||||
|
||||
token_data = result.json()
|
||||
|
||||
Reference in New Issue
Block a user