databricks: fix register to display SSO login link prior to timer expiration, fixes: #539

graphistry/pygraphistry.py

@@ -2,6 +2,7 @@
 from typing_extensions import Literal
 from graphistry.Plottable import Plottable
 from graphistry.privacy import Mode, Privacy
+from IPython.display import DisplayObject, TextDisplayObject
 
 """Top-level import of class PyGraphistry as "Graphistry". Used to connect to the Graphistry server and then create a base plotter."""
 import calendar, gzip, io, json, os, numpy as np, pandas as pd, requests, sys, time, warnings
@@ -99,6 +100,19 @@ def strtobool(val: Any) -> bool:
     else:
         raise ValueError("invalid truth value %r" % (val,))
 
+# used to print Login link databricks register() call 
+class Markdown(TextDisplayObject):
+
+    def __init__(self,TextDisplayObject):
+        import markdown as md
+
+        #converting markdown to html
+        self.html = md.markdown(TextDisplayObject)
+
+    def _repr_html_(self):
+        return self.html
+
+
 class PyGraphistry(object):
     _config = _get_initial_config()
     _tag = util.fingerprint()
@@ -270,7 +284,8 @@ def _handle_auth_url(auth_url, sso_timeout, sso_opt_into_type):
         if in_ipython() or in_databricks() or sso_opt_into_type == 'display':  # If run in notebook, just display the HTML
             # from IPython.core.display import HTML
             from IPython.display import display, HTML
-            display(HTML(f'<a href="{auth_url}" target="_blank">Login SSO</a>'))
+            display(HTML(f'<a href="{auth_url}" target="_blank">old: Login SSO</a>'))
+            display(Markdown(f'[new: Login SSO]({auth_url})'))
             print("Please click the above URL to open browser to login")
             print(f"If you cannot see the URL, please open browser, browse to this URL: {auth_url}")
             print("Please close browser tab after SSO login to back to notebook")