Update main.py

avgupta456 · Nov 30, 2024 · eeb0f55 · eeb0f55
1 parent 5b10c99
commit eeb0f55
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/src/processing/wrapped/main.py b/backend/src/processing/wrapped/main.py
@@ -14,8 +14,11 @@ async def query_wrapped_user(
 ) -> Tuple[bool, Optional[WrappedPackage]]:
     start_date, end_date = date(year, 1, 1), date(year, 12, 31)
     user: Optional[PublicUserModel] = await db_get_public_user(user_id)
-    access_token = None if user is None else user.access_token
-    private_access = False if user is None else user.private_access or False
+    private_access = False
+    access_token = None
+    if user is not None and user.private_access:
+        private_access = True
+        access_token = user.access_token
     user_package: UserPackage = await query_user(
         user_id,
         access_token,