Update api/routes/auth.py

api/routes/auth.py

@@ -34,7 +34,8 @@ async def signup(data: SignupForm):
         "password": hashed_pw,
         "role": "patient",
         "created_at": datetime.utcnow().isoformat(),
-        "updated_at": datetime.utcnow().isoformat()
+        "updated_at": datetime.utcnow().isoformat(),
+        "device_token": data.device_token or ""  # Store device token if provided
     }
     
     try:
@@ -83,7 +84,8 @@ async def create_doctor(
         "specialty": data.specialty,
         "license_number": data.license_number,
         "created_at": datetime.utcnow().isoformat(),
-        "updated_at": datetime.utcnow().isoformat()
+        "updated_at": datetime.utcnow().isoformat(),
+        "device_token": data.device_token or ""  # Store device token if provided
     }
     
     try:
@@ -113,6 +115,14 @@ async def login(form_data: OAuth2PasswordRequestForm = Depends()):
             headers={"WWW-Authenticate": "Bearer"},
         )
     
+    # Update device token if provided in form_data (e.g., from frontend)
+    if hasattr(form_data, 'device_token') and form_data.device_token:
+        await users_collection.update_one(
+            {"email": user["email"]},
+            {"$set": {"device_token": form_data.device_token}}
+        )
+        logger.info(f"Device token updated for {form_data.username}")
+
     access_token = create_access_token(data={"sub": user["email"]})
     logger.info(f"Successful login for {form_data.username}")
     return {
@@ -123,7 +133,7 @@ async def login(form_data: OAuth2PasswordRequestForm = Depends()):
 
 @router.get("/me")
 async def get_me(request: Request, current_user: dict = Depends(get_current_user)):
-    logger.info(f"Fetching user profile for {current_user['email']}")
+    logger.info(f"Fetching user profile for {current_user['email']} at {datetime.utcnow().isoformat()}")
     print(f"Headers: {request.headers}")
     try:
         user = await users_collection.find_one({"email": current_user["email"]})
@@ -141,12 +151,13 @@ async def get_me(request: Request, current_user: dict = Depends(get_current_user
             "role": user.get("role", "patient"),
             "specialty": user.get("specialty"),
             "created_at": user.get("created_at"),
-            "updated_at": user.get("updated_at")
+            "updated_at": user.get("updated_at"),
+            "device_token": user.get("device_token", "")  # Include device token in response
         }
-        logger.info(f"User profile retrieved for {current_user['email']}")
+        logger.info(f"User profile retrieved for {current_user['email']} at {datetime.utcnow().isoformat()}")
         return response
     except Exception as e:
-        logger.error(f"Database error for user {current_user['email']}: {str(e)}")
+        logger.error(f"Database error for user {current_user['email']}: {str(e)} at {datetime.utcnow().isoformat()}")
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail=f"Database error: {str(e)}"