wip (sso): fix message display logic

graphistry · Nov 6, 2024 · 9283e8a · 9283e8a
1 parent 412e944
commit 9283e8a
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/graphistry/pygraphistry.py b/graphistry/pygraphistry.py
@@ -2476,7 +2476,7 @@ def sso_repeat_get_token(repeat: int = 20, wait: int = 5):
         return
 
     @staticmethod
-    def sso_wait_for_token_html_display(repeat: int = 20, wait: int = 5, fail_silently: bool = True):
+    def sso_wait_for_token_html_display(repeat: int = 20, wait: int = 5, fail_silently: bool = False):
         """Get the JWT token for SSO login and display corresponding message in HTML
         Get the JWT token for SSO login and display corresponding message in HTML
         """
@@ -2486,7 +2486,12 @@ def sso_wait_for_token_html_display(repeat: int = 20, wait: int = 5, fail_silent
             if not PyGraphistry.sso_repeat_get_token(repeat, wait):
                 msg_html = f'{msg_html}<br /><strong>Failed to get token after {repeat*wait} seconds .... </strong>'
                 if not fail_silently:
-                    raise Exception(f"Failed to get token after {repeat*wait} seconds ....")
+                    raise Exception(f"Failed to get token after {repeat*wait} seconds. Please re-run the process") 
+                else:
+                    msg_html = f'{msg_html}<br /><strong>Got token</strong>'
+                    display(HTML(msg_html))
+                    return
+
             msg_html = f'{msg_html}<br /><strong>Got token</strong>'
             display(HTML(msg_html))
         else: