improve pretty print

Author: thomass-dev

skore-hub-project/src/skore_hub_project/authentication/login.py

@@ -4,6 +4,7 @@
 from logging import getLogger
 
 from rich.align import Align
+from rich.live import Live
 from rich.panel import Panel
 
 from skore_hub_project import console
@@ -31,8 +32,8 @@ def login(*, timeout: int = 600) -> None:
         logger.debug(f"Already logged in {URI} with {type(credentials)}.")
         console.print(
             Panel(
-                Align.center("[blink]Already logged in."),
-                title="[cyan]Login to [bold]Skore Hub",
+                Align.center("Already logged in."),
+                title="[cyan]Login to [b]Skore Hub",
                 border_style="cyan",
                 padding=1,
             )
@@ -43,36 +44,25 @@ def login(*, timeout: int = 600) -> None:
     try:
         credentials = APIKey()
     except KeyError:
-        console.print(
-            Panel(
-                Align.center(
-                    "Falling back to interactive authentication for the session.\n"
-                    "We recommend that you set up an API key via [url](coming soon) "
-                    "and use it to log in."
-                ),
-                title="[cyan]Login to [bold]Skore Hub",
-                subtitle="[dark_orange bold]API key not detected",
-                border_style="dark_orange",
-                padding=1,
-            )
-        )
+        with Live(console=console, auto_refresh=False) as live:
+            credentials = Token(timeout=timeout, live=live)
 
-        credentials = Token(timeout=timeout)
-
-        console.print(
-            Panel(
-                Align.center("[blink]Login successful."),
-                title="[cyan]Login to [b]Skore Hub",
-                border_style="cyan",
-                padding=1,
+            live.update(
+                Panel(
+                    Align.center(
+                        "Login successful using [b]interactive authentication."
+                    ),
+                    title="[cyan]Login to [bold]Skore Hub",
+                    border_style="cyan",
+                    padding=1,
+                )
             )
-        )
+            live.refresh()
     else:
         console.print(
             Panel(
-                Align.center("[blink]Login successful."),
+                Align.center("Login successful using [b]API key."),
                 title="[cyan]Login to [bold]Skore Hub",
-                subtitle="[cyan bold]API key detected",
                 border_style="cyan",
                 padding=1,
             )

skore-hub-project/src/skore_hub_project/authentication/token.py

@@ -5,10 +5,16 @@
 from datetime import datetime, timezone
 from threading import RLock
 from time import sleep
+from typing import cast
 from urllib.parse import urljoin
 from webbrowser import open as open_webbrowser
 
 from httpx import HTTPStatusError, TimeoutException
+from rich.align import Align
+from rich.console import Group
+from rich.live import Live
+from rich.panel import Panel
+from rich.text import Text
 
 from skore_hub_project import console
 from skore_hub_project.authentication.uri import URI
@@ -196,17 +202,28 @@ class Token:
     Refresh the token on-the-fly if necessary.
     """
 
-    def __init__(self, *, timeout: int = 600) -> None:
+    def __init__(self, *, timeout: int = 600, live: Live | None = None) -> None:
         url, device_code, user_code = get_oauth_device_login()
-
-        console.print(
-            (
-                f"Opening browser for interactive authentication; "
-                f"if this fails, please visit:\n{url}"
+        panel = Panel(
+            Align.center(
+                "[b]API key not detected.[/b]\n\n"
+                "Starting interactive authentication for the session.\n"
+                "[i]We recommend that you set up an API key via url (coming soon) and "
+                "use it to log in.[/i]\n\n"
+                "Opening browser for interactive authentication; if this fails, "
+                f"please visit:\n[link={url}]{url}[/link]"
             ),
-            soft_wrap=True,
+            title="[cyan]Login to [bold]Skore Hub",
+            border_style="cyan",
+            padding=1,
         )
 
+        if live:
+            live.update(panel)
+            live.refresh()
+        else:
+            console.print(panel, soft_wrap=True)
+
         open_webbrowser(url)
 
         get_oauth_device_code_probe(device_code, timeout=timeout)