chores: Format

Author: KShivendu

supertokens_python/recipe/session/access_token.py

@@ -155,7 +155,9 @@ def validate_access_token_structure(payload: Dict[str, Any], version: int) -> No
 
         if version >= 4:
             if not isinstance(payload.get("tId"), str):
-                raise Exception("Access token does not contain all the information. Maybe the structure has changed?")
+                raise Exception(
+                    "Access token does not contain all the information. Maybe the structure has changed?"
+                )
 
     elif (
         not isinstance(payload.get("sessionHandle"), str)

supertokens_python/recipe/session/asyncio/__init__.py

@@ -426,18 +426,21 @@ async def revoke_session(
 
 
 async def revoke_all_sessions_for_user(
-    user_id: str, tenant_id: Optional[str] = None, user_context: Union[None, Dict[str, Any]] = None
+    user_id: str,
+    tenant_id: Optional[str] = None,
+    user_context: Union[None, Dict[str, Any]] = None,
 ) -> List[str]:
     if user_context is None:
         user_context = {}
     return await SessionRecipe.get_instance().recipe_implementation.revoke_all_sessions_for_user(
-        user_id, tenant_id or DEFAULT_TENANT_ID,
-        tenant_id is None, user_context
+        user_id, tenant_id or DEFAULT_TENANT_ID, tenant_id is None, user_context
     )
 
 
 async def get_all_session_handles_for_user(
-    user_id: str, tenant_id: Optional[str], user_context: Union[None, Dict[str, Any]] = None
+    user_id: str,
+    tenant_id: Optional[str],
+    user_context: Union[None, Dict[str, Any]] = None,
 ) -> List[str]:
     if user_context is None:
         user_context = {}

supertokens_python/recipe/session/interfaces.py

@@ -41,7 +41,13 @@
 
 
 class SessionObj:
-    def __init__(self, handle: str, user_id: str, user_data_in_jwt: Dict[str, Any], tenant_id: str):
+    def __init__(
+        self,
+        handle: str,
+        user_id: str,
+        user_data_in_jwt: Dict[str, Any],
+        tenant_id: str,
+    ):
         self.handle = handle
         self.user_id = user_id
         self.user_data_in_jwt = user_data_in_jwt
@@ -76,7 +82,9 @@ def __init__(
         self.user_id = user_id
         self.session_data_in_database = session_data_in_database
         self.expiry = expiry
-        self.custom_claims_in_access_token_payload = custom_claims_in_access_token_payload
+        self.custom_claims_in_access_token_payload = (
+            custom_claims_in_access_token_payload
+        )
         self.time_created = time_created
         self.tenant_id = tenant_id
 
@@ -208,13 +216,21 @@ async def revoke_session(
 
     @abstractmethod
     async def revoke_all_sessions_for_user(
-        self, user_id: str, tenant_id: str, revoke_across_all_tenants: Optional[bool], user_context: Dict[str, Any]
+        self,
+        user_id: str,
+        tenant_id: str,
+        revoke_across_all_tenants: Optional[bool],
+        user_context: Dict[str, Any],
     ) -> List[str]:
         pass
 
     @abstractmethod
     async def get_all_session_handles_for_user(
-        self, user_id: str, tenant_id: str, fetch_across_all_tenants: Optional[bool], user_context: Dict[str, Any]
+        self,
+        user_id: str,
+        tenant_id: str,
+        fetch_across_all_tenants: Optional[bool],
+        user_context: Dict[str, Any],
     ) -> List[str]:
         pass
 

supertokens_python/recipe/session/recipe_implementation.py

@@ -75,7 +75,7 @@ async def create_new_session(
             disable_anti_csrf is True,
             access_token_payload,
             session_data_in_database,
-            tenant_id
+            tenant_id,
         )
         log_debug_message("createNewSession: Finished")
 
@@ -97,7 +97,7 @@ async def create_new_session(
             payload,
             None,
             True,
-            tenant_id
+            tenant_id,
         )
 
         return new_session
@@ -327,14 +327,26 @@ async def revoke_session(
         return await session_functions.revoke_session(self, session_handle)
 
     async def revoke_all_sessions_for_user(
-        self, user_id: str, tenant_id: Optional[str], revoke_across_all_tenants: Optional[bool], user_context: Dict[str, Any]
+        self,
+        user_id: str,
+        tenant_id: Optional[str],
+        revoke_across_all_tenants: Optional[bool],
+        user_context: Dict[str, Any],
     ) -> List[str]:
-        return await session_functions.revoke_all_sessions_for_user(self, user_id, tenant_id, revoke_across_all_tenants)
+        return await session_functions.revoke_all_sessions_for_user(
+            self, user_id, tenant_id, revoke_across_all_tenants
+        )
 
     async def get_all_session_handles_for_user(
-        self, user_id: str, tenant_id: Optional[str], fetch_across_all_tenants: Optional[bool], user_context: Dict[str, Any]
+        self,
+        user_id: str,
+        tenant_id: Optional[str],
+        fetch_across_all_tenants: Optional[bool],
+        user_context: Dict[str, Any],
     ) -> List[str]:
-        return await session_functions.get_all_session_handles_for_user(self, user_id, tenant_id, fetch_across_all_tenants)
+        return await session_functions.get_all_session_handles_for_user(
+            self, user_id, tenant_id, fetch_across_all_tenants
+        )
 
     async def revoke_multiple_sessions(
         self, session_handles: List[str], user_context: Dict[str, Any]
@@ -467,6 +479,6 @@ async def regenerate_access_token(
             response["session"]["handle"],
             response["session"]["userId"],
             response["session"]["userDataInJWT"],
-            response["session"]["tenantId"]
+            response["session"]["tenantId"],
         )
         return RegenerateAccessTokenOkResult(session, access_token_obj)

supertokens_python/recipe/session/session_functions.py

@@ -101,7 +101,7 @@ async def create_new_session(
     disable_anti_csrf: bool,
     access_token_payload: Union[None, Dict[str, Any]],
     session_data_in_database: Union[None, Dict[str, Any]],
-    tenant_id: Optional[str]
+    tenant_id: Optional[str],
 ) -> CreateOrRefreshAPIResponse:
     if session_data_in_database is None:
         session_data_in_database = {}
@@ -263,7 +263,7 @@ async def get_session(
                 access_token_info["userId"],
                 access_token_info["userData"],
                 access_token_info["expiryTime"],
-                access_token_info["tenantId"]
+                access_token_info["tenantId"],
             )
         )
 
@@ -302,7 +302,9 @@ async def get_session(
                         "expiryTime"
                     ]  # if the token didn't pass validation, but we got here, it means it was a v2 token that we didn't have the key cached for.
                 ),  # This will throw error if others are none and 'expiryTime' key doesn't exist in the payload
-                response["session"].get("tenantId") or (access_token_info or {}).get("tenantId") or DEFAULT_TENANT_ID,
+                response["session"].get("tenantId")
+                or (access_token_info or {}).get("tenantId")
+                or DEFAULT_TENANT_ID,
             ),
             GetSessionAPIResponseAccessToken(
                 response["accessToken"]["token"],
@@ -387,32 +389,33 @@ async def refresh_session(
 
 
 async def revoke_all_sessions_for_user(
-    recipe_implementation: RecipeImplementation, user_id: str,
-    tenant_id: Optional[str], revoke_across_all_tenants: Optional[bool],
+    recipe_implementation: RecipeImplementation,
+    user_id: str,
+    tenant_id: Optional[str],
+    revoke_across_all_tenants: Optional[bool],
 ) -> List[str]:
     if tenant_id is None:
         tenant_id = DEFAULT_TENANT_ID
 
     response = await recipe_implementation.querier.send_post_request(
         NormalisedURLPath(f"{tenant_id}/recipe/session/remove"),
-        {
-            "userId": user_id,
-            "revokeAcrossAllTenants": revoke_across_all_tenants
-        }
+        {"userId": user_id, "revokeAcrossAllTenants": revoke_across_all_tenants},
     )
     return response["sessionHandlesRevoked"]
 
 
 async def get_all_session_handles_for_user(
-    recipe_implementation: RecipeImplementation, user_id: str,
-    tenant_id: Optional[str], fetch_across_all_tenants: Optional[bool]
+    recipe_implementation: RecipeImplementation,
+    user_id: str,
+    tenant_id: Optional[str],
+    fetch_across_all_tenants: Optional[bool],
 ) -> List[str]:
     if tenant_id is None:
         tenant_id = DEFAULT_TENANT_ID
 
     response = await recipe_implementation.querier.send_get_request(
         NormalisedURLPath(f"{tenant_id}/recipe/session/user"),
-        {"userId": user_id, "fetchAcrossAllTenants": fetch_across_all_tenants}
+        {"userId": user_id, "fetchAcrossAllTenants": fetch_across_all_tenants},
     )
     return response["sessionHandles"]
 
@@ -431,8 +434,7 @@ async def revoke_multiple_sessions(
     recipe_implementation: RecipeImplementation, session_handles: List[str]
 ) -> List[str]:
     response = await recipe_implementation.querier.send_post_request(
-        NormalisedURLPath("/recipe/session/remove"),
-        {"sessionHandles": session_handles}
+        NormalisedURLPath("/recipe/session/remove"), {"sessionHandles": session_handles}
     )
     return response["sessionHandlesRevoked"]
 
@@ -481,6 +483,6 @@ async def get_session_information(
             response["expiry"],
             response["userDataInJWT"],
             response["timeCreated"],
-            response["tenantId"]
+            response["tenantId"],
         )
     return None

supertokens_python/recipe/session/syncio/__init__.py

@@ -187,7 +187,9 @@ def revoke_session(
 
 
 def revoke_all_sessions_for_user(
-    user_id: str, tenant_id: Optional[str] = None, user_context: Union[None, Dict[str, Any]] = None
+    user_id: str,
+    tenant_id: Optional[str] = None,
+    user_context: Union[None, Dict[str, Any]] = None,
 ) -> List[str]:
     from supertokens_python.recipe.session.asyncio import (
         revoke_all_sessions_for_user as async_revoke_all_sessions_for_user,
@@ -197,13 +199,17 @@ def revoke_all_sessions_for_user(
 
 
 def get_all_session_handles_for_user(
-    user_id: str, tenant_id: Optional[str] = None, user_context: Union[None, Dict[str, Any]] = None
+    user_id: str,
+    tenant_id: Optional[str] = None,
+    user_context: Union[None, Dict[str, Any]] = None,
 ) -> List[str]:
     from supertokens_python.recipe.session.asyncio import (
         get_all_session_handles_for_user as async_get_all_session_handles_for_user,
     )
 
-    return sync(async_get_all_session_handles_for_user(user_id, tenant_id, user_context))
+    return sync(
+        async_get_all_session_handles_for_user(user_id, tenant_id, user_context)
+    )
 
 
 def revoke_multiple_sessions(