finalize user_communicator

rendercv/user_communicator.py

@@ -1,45 +1,49 @@
 import time
 from typing import Callable
 
-from rich import print
-from rich.console import Console, Group
-from rich.panel import Panel
-from rich.live import Live
-from rich.progress import (
-    BarColumn,
-    Progress,
-    ProgressColumn,
-    TextColumn,
-    Task,
-)
-from rich.text import Text
-
-import pydantic
-
-error_console = Console(stderr=True)
+import rich
+import rich.console
+import rich.panel
+import rich.live
+import rich.table
+import rich.text
+import rich.progress
 
 
-def print_rendercv_graphics():
+console = rich.console.Console()
+
+
+def welcome():
     """Print a welcome message to the terminal."""
-    print("Welcome to [bold blue]RenderCV[/bold blue]!")
-    print("Documentation: [link=https://sinaatalay.github.io/rendercv/]")
-    print("Source code: [link=https://github.com/sinaatalay/rendercv/]")
-    print("Report bugs: [link=https://github.com/sinaatalay/rendercv/issues/]")
+    table = rich.table.Table(
+        title="\nWelcome to [bold]Render[dodger_blue3]CV[/dodger_blue3][/bold]!",
+        title_justify="center",
+    )
+
+    table.add_column("Title", style="magenta")
+    table.add_column("Link", style="cyan", justify="right", no_wrap=True)
+
+    table.add_row("Documentation", "https://sinaatalay.github.io/rendercv/")
+    table.add_row("Source code", "https://github.com/sinaatalay/rendercv/")
+    table.add_row("Bug reports", "https://github.com/sinaatalay/rendercv/issues/")
+    table.add_row("Feature requests", "https://github.com/sinaatalay/rendercv/issues/")
+
+    console.print(table, justify="center")
 
 
 def warning(text):
     """Print a warning message to the terminal."""
-    print(f"[bold yellow]{text}[/bold yellow]")
+    console.print(f"[bold yellow]{text}[/bold yellow]")
 
 
 def error(text):
     """Print an error message to the terminal."""
-    error_console.print(f"[bold red]{text}[/bold red]")
+    console.print(f"[bold red]{text}[/bold red]")
 
 
 def information(text):
     """Print an information message to the terminal."""
-    print(f"[bold blue]{text}[/bold blue]")
+    console.print(f"[bold green]{text}")
 
 
 def time_the_event_below(event_name: str) -> Callable:
@@ -91,28 +95,35 @@ def handle_exceptions(function: Callable) -> Callable:
     return wrapper
 
 
-class LiveProgressReporter(Live):
+class LiveProgressReporter(rich.live.Live):
+    """This class is a wrapper around `rich.live.Live` that provides the live progress
+    reporting functionality.
+
+    Args:
+        number_of_steps (int): The number of steps to be finished.
+    """
+
     def __init__(self, number_of_steps: int):
-        class TimeElapsedColumn(ProgressColumn):
-            def render(self, task: "Task") -> Text:
+        class TimeElapsedColumn(rich.progress.ProgressColumn):
+            def render(self, task: "rich.progress.Task") -> rich.text.Text:
                 elapsed = task.finished_time if task.finished else task.elapsed
                 if elapsed is None:
-                    return Text("--.-", style="progress.elapsed")
+                    return rich.text.Text("--.-", style="progress.elapsed")
                 delta = f"{elapsed:.1f} s"
-                return Text(str(delta), style="progress.elapsed")
+                return rich.text.Text(str(delta), style="progress.elapsed")
 
-        self.step_progress = Progress(
-            TimeElapsedColumn(), TextColumn("{task.description}")
+        self.step_progress = rich.progress.Progress(
+            TimeElapsedColumn(), rich.progress.TextColumn("{task.description}")
         )
 
-        self.overall_progress = Progress(
+        self.overall_progress = rich.progress.Progress(
             TimeElapsedColumn(),
-            BarColumn(),
-            TextColumn("{task.description}"),
+            rich.progress.BarColumn(),
+            rich.progress.TextColumn("{task.description}"),
         )
 
-        self.group = Group(
-            Panel(Group(self.step_progress)),
+        self.group = rich.console.Group(
+            rich.panel.Panel(rich.console.Group(self.step_progress)),
             self.overall_progress,
         )
 
@@ -122,32 +133,44 @@ class LiveProgressReporter(Live):
         self.overall_progress.update(
             self.overall_task_id,
             description=(
-                f"[bold #AAAAAA](0 out of {self.number_of_steps} steps finished)"
+                f"[bold #AAAAAA]({self.current_step} out of"
+                f" {self.number_of_steps} steps finished)"
             ),
         )
         super().__init__(self.group)
 
     def __enter__(self) -> "LiveProgressReporter":
+        """Overwrite the `__enter__` method for the correct return type."""
         self.start(refresh=self._renderable is not None)
         return self
 
     def start_a_step(self, step_name: str):
+        """Start a step and update the progress bars."""
         self.current_step_name = step_name
         self.current_step_id = self.step_progress.add_task(
             f"{self.current_step_name} has started."
         )
 
     def finish_the_current_step(self):
+        """Finish the current step and update the progress bars."""
         self.step_progress.stop_task(self.current_step_id)
         self.step_progress.update(
             self.current_step_id, description=f"{self.current_step_name} has finished."
         )
-        self.overall_progress.update(self.overall_task_id, advance=1)
         self.current_step += 1
+        self.overall_progress.update(
+            self.overall_task_id,
+            description=(
+                f"[bold #AAAAAA]({self.current_step} out of"
+                f" {self.number_of_steps} steps finished)"
+            ),
+            advance=1,
+        )
         if self.current_step == self.number_of_steps:
             self.end()
 
     def end(self):
+        """End the live progress reporting."""
         self.overall_progress.update(
             self.overall_task_id,
             description="[bold green]Your CV is rendered!",