From 60d3f7a3924bdd73b3bd73f31da182c7f5e0131d Mon Sep 17 00:00:00 2001
From: Sourcery AI <>
Date: Sun, 22 Jan 2023 15:09:08 +0000
Subject: [PATCH] 'Refactored by Sourcery'

---
 gotrue/_sync/gotrue_client.py | 35 +++++++++++++++++------------------
 1 file changed, 17 insertions(+), 18 deletions(-)

diff --git a/gotrue/_sync/gotrue_client.py b/gotrue/_sync/gotrue_client.py
index 36d9f274..43e41079 100644
--- a/gotrue/_sync/gotrue_client.py
+++ b/gotrue/_sync/gotrue_client.py
@@ -372,8 +372,7 @@ def get_user(self, jwt: Union[str, None] = None) -> UserResponse:
         `get_user()` will attempt to get the `jwt` from the current session.
         """
         if not jwt:
-            session = self.get_session()
-            if session:
+            if session := self.get_session():
                 jwt = session.access_token
         return self._request("GET", "user", jwt=jwt, xform=parse_user_response)
 
@@ -535,15 +534,15 @@ def _enroll(self, params: MFAEnrollParams) -> AuthMFAEnrollResponse:
         return response
 
     def _challenge(self, params: MFAChallengeParams) -> AuthMFAChallengeResponse:
-        session = self.get_session()
-        if not session:
+        if session := self.get_session():
+            return self._request(
+                "POST",
+                f"factors/{params.get('factor_id')}/challenge",
+                jwt=session.access_token,
+                xform=AuthMFAChallengeResponse.parse_obj,
+            )
+        else:
             raise AuthSessionMissingError()
-        return self._request(
-            "POST",
-            f"factors/{params.get('factor_id')}/challenge",
-            jwt=session.access_token,
-            xform=AuthMFAChallengeResponse.parse_obj,
-        )
 
     def _challenge_and_verify(
         self,
@@ -579,15 +578,15 @@ def _verify(self, params: MFAVerifyParams) -> AuthMFAVerifyResponse:
         return response
 
     def _unenroll(self, params: MFAUnenrollParams) -> AuthMFAUnenrollResponse:
-        session = self.get_session()
-        if not session:
+        if session := self.get_session():
+            return self._request(
+                "DELETE",
+                f"factors/{params.get('factor_id')}",
+                jwt=session.access_token,
+                xform=AuthMFAUnenrollResponse.parse_obj,
+            )
+        else:
             raise AuthSessionMissingError()
-        return self._request(
-            "DELETE",
-            f"factors/{params.get('factor_id')}",
-            jwt=session.access_token,
-            xform=AuthMFAUnenrollResponse.parse_obj,
-        )
 
     def _list_factors(self) -> AuthMFAListFactorsResponse:
         response = self.get_user()