fixes review comments

Author: rishabhpoddar

supertokens_python/recipe/session/asyncio/__init__.py

@@ -49,8 +49,8 @@
 
 
 async def create_new_session(
-    tenant_id: str,
     request: Any,
+    tenant_id: str,
     user_id: str,
     access_token_payload: Union[Dict[str, Any], None] = None,
     session_data_in_database: Union[Dict[str, Any], None] = None,

supertokens_python/recipe/session/syncio/__init__.py

@@ -39,8 +39,8 @@
 
 
 def create_new_session(
-    tenant_id: str,
     request: Any,
+    tenant_id: str,
     user_id: str,
     access_token_payload: Union[Dict[str, Any], None] = None,
     session_data_in_database: Union[Dict[str, Any], None] = None,

supertokens_python/recipe/thirdparty/api/implementation.py

@@ -12,7 +12,6 @@
 # License for the specific language governing permissions and limitations
 # under the License.
 from __future__ import annotations
-from supertokens_python.utils import utf_base64decode
 from base64 import b64decode
 import json
 

supertokens_python/recipe/thirdpartyemailpassword/asyncio/__init__.py

@@ -26,7 +26,7 @@
     CreateResetPasswordLinkUknownUserIdError,
     CreateResetPasswordLinkOkResult,
     SendResetPasswordEmailUnknownUserIdError,
-    SendResetPasswordEmailEmailOkResult
+    SendResetPasswordEmailEmailOkResult,
 )
 from supertokens_python.recipe.emailpassword.utils import get_password_reset_link
 
@@ -61,6 +61,7 @@ async def get_user_by_third_party_info(
         user_context,
     )
 
+
 async def thirdparty_manually_create_or_update_user(
     tenant_id: str,
     third_party_id: str,

supertokens_python/syncio/__init__.py

@@ -29,33 +29,40 @@
 
 
 def get_users_oldest_first(
+    tenant_id: str,
     limit: Union[int, None] = None,
     pagination_token: Union[str, None] = None,
     include_recipe_ids: Union[None, List[str]] = None,
     query: Union[None, Dict[str, str]] = None,
 ) -> UsersResponse:
     return sync(
         Supertokens.get_instance().get_users(
-            "public", "ASC", limit, pagination_token, include_recipe_ids, query
+            tenant_id, "ASC", limit, pagination_token, include_recipe_ids, query
         )
     )
 
 
 def get_users_newest_first(
+    tenant_id: str,
     limit: Union[int, None] = None,
     pagination_token: Union[str, None] = None,
     include_recipe_ids: Union[None, List[str]] = None,
     query: Union[None, Dict[str, str]] = None,
 ) -> UsersResponse:
     return sync(
         Supertokens.get_instance().get_users(
-            "public", "DESC", limit, pagination_token, include_recipe_ids, query
+            tenant_id, "DESC", limit, pagination_token, include_recipe_ids, query
         )
     )
 
 
-def get_user_count(include_recipe_ids: Union[None, List[str]] = None) -> int:
-    return sync(Supertokens.get_instance().get_user_count(include_recipe_ids))
+def get_user_count(
+    include_recipe_ids: Union[None, List[str]] = None,
+    tenant_id: Optional[str] = None,
+) -> int:
+    return sync(
+        Supertokens.get_instance().get_user_count(include_recipe_ids, tenant_id)
+    )
 
 
 def delete_user(user_id: str) -> None:

tests/Django/test_django.py

@@ -76,7 +76,7 @@ def get_cookies(response: HttpResponse) -> Dict[str, Any]:
 
 
 async def create_new_session_view(request: HttpRequest):
-    await create_new_session("public", request, "user_id")
+    await create_new_session(request, "public", "user_id")
     return JsonResponse({"foo": "bar"})
 
 
@@ -456,10 +456,12 @@ async def test_thirdparty_parsing_works(self):
 
         start_st()
 
-        state = b64encode(json.dumps({"redirectURI": "http://localhost:3000/redirect" }).encode()).decode()
+        state = b64encode(
+            json.dumps({"redirectURI": "http://localhost:3000/redirect"}).encode()
+        ).decode()
         code = "testing"
 
-        data = { "state": state, "code": code}
+        data = {"state": state, "code": code}
 
         request = self.factory.post(
             "/auth/callback/apple",
@@ -472,8 +474,11 @@ async def test_thirdparty_parsing_works(self):
         response = await temp
 
         self.assertEqual(response.status_code, 303)
-        self.assertEqual(response.content, b'')
-        self.assertEqual(response.headers['location'], f"http://localhost:3000/redirect?state={state.replace('=', '%3D')}&code={code}")
+        self.assertEqual(response.content, b"")
+        self.assertEqual(
+            response.headers["location"],
+            f"http://localhost:3000/redirect?state={state.replace('=', '%3D')}&code={code}",
+        )
 
     @pytest.mark.asyncio
     async def test_search_with_multiple_emails(self):

tests/Fastapi/test_fastapi.py

@@ -91,7 +91,7 @@ async def driver_config_client():
     @app.get("/login")
     async def login(request: Request):  # type: ignore
         user_id = "userId"
-        await create_new_session("public", request, user_id, {}, {})
+        await create_new_session(request, "public", user_id, {}, {})
         return {"userId": user_id}
 
     @app.post("/refresh")
@@ -135,12 +135,12 @@ async def custom_logout(request: Request):  # type: ignore
 
     @app.post("/create")
     async def _create(request: Request):  # type: ignore
-        await create_new_session("public", request, "userId", {}, {})
+        await create_new_session(request, "public", "userId", {}, {})
         return ""
 
     @app.post("/create-throw")
     async def _create_throw(request: Request):  # type: ignore
-        await create_new_session("public", request, "userId", {}, {})
+        await create_new_session(request, "public", "userId", {}, {})
         raise UnauthorisedError("unauthorised")
 
     return TestClient(app)

tests/Flask/test_flask.py

@@ -192,7 +192,7 @@ def t():  # type: ignore
     @app.route("/login")  # type: ignore
     def login():  # type: ignore
         user_id = "userId"
-        create_new_session("public", request, user_id, {}, {})
+        create_new_session(request, "public", user_id, {}, {})
 
         return jsonify({"userId": user_id, "session": "ssss"})
 
@@ -478,15 +478,20 @@ def test_thirdparty_parsing_works(driver_config_app: Any):
     start_st()
 
     test_client = driver_config_app.test_client()
-    state = b64encode(json.dumps({"redirectURI": "http://localhost:3000/redirect" }).encode()).decode()
+    state = b64encode(
+        json.dumps({"redirectURI": "http://localhost:3000/redirect"}).encode()
+    ).decode()
     code = "testing"
 
-    data = { "state": state, "code": code}
+    data = {"state": state, "code": code}
     res = test_client.post("/auth/callback/apple", data=data)
 
     assert res.status_code == 303
-    assert res.data == b''
-    assert res.headers["location"] == f"http://localhost:3000/redirect?state={state.replace('=', '%3D')}&code={code}"
+    assert res.data == b""
+    assert (
+        res.headers["location"]
+        == f"http://localhost:3000/redirect?state={state.replace('=', '%3D')}&code={code}"
+    )
 
 
 from flask.wrappers import Response
@@ -747,7 +752,7 @@ def test_api():  # type: ignore
     @app.route("/login")  # type: ignore
     def login():  # type: ignore
         user_id = "userId"
-        s = create_new_session("public", request, user_id, {}, {})
+        s = create_new_session(request, "public", user_id, {}, {})
         return jsonify({"user": s.get_user_id()})
 
     @app.route("/ping")  # type: ignore

tests/emailpassword/test_emailexists.py

@@ -50,7 +50,7 @@ async def driver_config_client():
     @app.get("/login")
     async def login(request: Request):  # type: ignore
         user_id = "userId"
-        await create_new_session("public", request, user_id, {}, {})
+        await create_new_session(request, "public", user_id, {}, {})
         return {"userId": user_id}
 
     @app.post("/refresh")

tests/emailpassword/test_emailverify.py

@@ -79,7 +79,7 @@ async def driver_config_client():
     @app.get("/login")
     async def login(request: Request):  # type: ignore
         user_id = "userId"
-        await create_new_session("public", request, user_id, {}, {})
+        await create_new_session(request, "public", user_id, {}, {})
         return {"userId": user_id}
 
     @app.post("/refresh")