phuochungus commited on
Commit
0b4f36d
1 Parent(s): d6a03dc

handle user not have profile

Browse files
Files changed (1) hide show
  1. app/dependencies.py +4 -0
app/dependencies.py CHANGED
@@ -2,6 +2,7 @@ from fastapi import Depends, HTTPException, status
2
  from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
3
  from firebase_admin import auth
4
  from firebase_admin.auth import ExpiredIdTokenError, InvalidIdTokenError
 
5
 
6
  security = HTTPBearer()
7
 
@@ -11,6 +12,9 @@ def get_current_user(
11
  ):
12
  try:
13
  payload = auth.verify_id_token(credentials.credentials)
 
 
 
14
  except ExpiredIdTokenError as e:
15
  raise HTTPException(
16
  status_code=status.HTTP_401_UNAUTHORIZED,
 
2
  from fastapi.security import HTTPAuthorizationCredentials, HTTPBearer
3
  from firebase_admin import auth
4
  from firebase_admin.auth import ExpiredIdTokenError, InvalidIdTokenError
5
+ from . import db
6
 
7
  security = HTTPBearer()
8
 
 
12
  ):
13
  try:
14
  payload = auth.verify_id_token(credentials.credentials)
15
+ user_doc_ref = db.collection("user").document(payload["sub"]).get()
16
+ if user_doc_ref.exists:
17
+ raise HTTPException(status_code=400, detail="User profile not found")
18
  except ExpiredIdTokenError as e:
19
  raise HTTPException(
20
  status_code=status.HTTP_401_UNAUTHORIZED,