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()