Atualizando retorno do endpoint

fga-eps-mds · Dec 2, 2023 · fa51135 · fa51135
1 parent 363fda5
commit fa51135
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/src/controller/authController.py b/src/controller/authController.py
@@ -67,11 +67,19 @@ async def login(data: authSchema.UserLogin, db: Session = Depends(get_db)):
 @auth.post("/login/social")
 async def login_social(user: authSchema.UserSocial, db: Session = Depends(get_db)):
 
-    existing_user = userRepository.get_user_by_email_social(db, user.email)
+    existing_user = userRepository.get_user_by_email(db, user.email)
     if existing_user is None:
         new_user = userRepository.create_user_social(db, user.name, user.email)
         access_token = security.create_access_token(data={"id": new_user.id, "email": new_user.email, "role": new_user.role})
-        return JSONResponse(status_code=200, content={"access_token": access_token, "token_type": "bearer", "is_new_user": True})
+        return JSONResponse(
+    status_code=200, 
+    content={
+        "access_token": access_token, 
+        "token_type": "bearer", 
+        "is_new_user": True, 
+        "user_id": new_user.id
+    }
+)
     else:
         access_token = security.create_access_token(data={"id": existing_user.id, "email": existing_user.email, "role": existing_user.role})
         refresh_token = security.create_refresh_token(data={ "id": existing_user.id })