Spaces:
Runtime error
Runtime error
phuochungus
commited on
Commit
•
0b4f36d
1
Parent(s):
d6a03dc
handle user not have profile
Browse files- 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,
|