Fixes for return_on_mfa code

README.md

@@ -140,10 +140,10 @@ garth.login(email, password, prompt_mfa=lambda: input("Enter MFA code: "))
 For advanced use cases (like async handling), MFA can be handled separately:
 
 ```python
-result = garth.login(email, password, return_on_mfa=True)
-if isinstance(result, dict):  # MFA is required
+result1, result2 = garth.login(email, password, return_on_mfa=True)
+if result1 == "needs_mfa":  # MFA is required
     mfa_code = "123456"  # Get this from your custom MFA flow
-    oauth1, oauth2 = garth.resume_login(result['client_state'], mfa_code)
+    oauth1, oauth2 = garth.resume_login(result2, mfa_code)
 ```
 
 ### Configure

garth/http.py

@@ -161,10 +161,19 @@ def put(self, *args, **kwargs) -> Response:
         return self.request("PUT", *args, **kwargs)
 
     def login(self, *args, **kwargs):
-        self.oauth1_token, self.oauth2_token = sso.login(
+        oauth1, oauth2 = sso.login(
             *args, **kwargs, client=self
         )
-
+        self.oauth1_token = oauth1
+        self.oauth2_token = oauth2
+        return oauth1, oauth2
+
+    def resume_login(self, *args, **kwargs):
+        self.oauth1_token, self.oauth2_token = sso.resume_login(
+            *args, **kwargs, # client=self
+        )
+        return self.oauth1_token, self.oauth2_token
+
     def refresh_oauth2(self):
         assert self.oauth1_token, "OAuth1 token is required for OAuth2 refresh"
         # There is a way to perform a refresh of an OAuth2 token, but it

garth/sso.py

@@ -141,13 +141,10 @@ def login(
     # Handle MFA
     if "MFA" in title:
         if return_on_mfa or prompt_mfa is None:
-            return {
-                "needs_mfa": True,
-                "client_state": {
-                    "csrf_token": csrf_token,
-                    "signin_params": SIGNIN_PARAMS,
-                    "client": client,
-                },
+            return "needs_mfa", {
+                "csrf_token": csrf_token,
+                "signin_params": SIGNIN_PARAMS,
+                "client": client,
             }
 
         handle_mfa(client, SIGNIN_PARAMS, prompt_mfa)

garth/version.py

@@ -1 +1 @@
-__version__ = "0.5.2"
+__version__ = "0.5.3"