allow prompt_mfa to be async

matin · May 18, 2024 · ae1b425 · ae1b425
1 parent 316787d
commit ae1b425
Show file tree

Hide file tree

Showing 4 changed files with 1,172 additions and 2 deletions.
diff --git a/garth/sso.py b/garth/sso.py
@@ -1,3 +1,4 @@
+import asyncio
 import re
 import time
 from typing import Callable, Dict, Optional, Tuple
@@ -150,7 +151,10 @@ def handle_mfa(
     client: "http.Client", signin_params: dict, prompt_mfa: Callable
 ) -> None:
     csrf_token = get_csrf_token(client.last_resp.text)
-    mfa_code = prompt_mfa()
+    if asyncio.iscoroutinefunction(prompt_mfa):
+        mfa_code = asyncio.run(prompt_mfa())
+    else:
+        mfa_code = prompt_mfa()
     client.post(
         "sso",
         "/sso/verifyMFA/loginEnterMfaCode",

diff --git a/garth/version.py b/garth/version.py
@@ -1 +1 @@
-__version__ = "0.4.45"
+__version__ = "0.4.46"