From 6d2ae4ff14148f4d33b69add86af8865ca3c76e6 Mon Sep 17 00:00:00 2001
From: monosans <hsyqixco@protonmail.com>
Date: Wed, 6 Nov 2024 15:09:38 +0300
Subject: [PATCH] fix: use global rich.Console object

---
 proxy_scraper_checker/__main__.py | 6 +++---
 proxy_scraper_checker/logs.py     | 9 ++-------
 2 files changed, 5 insertions(+), 10 deletions(-)

diff --git a/proxy_scraper_checker/__main__.py b/proxy_scraper_checker/__main__.py
index ab94ae7..e4bf2f8 100644
--- a/proxy_scraper_checker/__main__.py
+++ b/proxy_scraper_checker/__main__.py
@@ -3,7 +3,7 @@ from __future__ import annotations
 
 from . import logs
 
-_console, _logs_listener = logs.configure()
+_logs_listener = logs.configure()
 
 import asyncio
 import logging
@@ -11,6 +11,7 @@ import sys
 from typing import TYPE_CHECKING
 
 import aiofiles
+import rich
 from aiohttp import ClientSession, TCPConnector
 from rich.progress import BarColumn, MofNCompleteColumn, Progress, TextColumn
 from rich.table import Table
@@ -113,7 +114,6 @@ async def main() -> None:
                 TextColumn("[green]{task.fields[col2]}"),
                 BarColumn(),
                 MofNCompleteColumn(),
-                console=_console,
                 transient=True,
             ) as progress:
                 scrape = scraper.scrape_all(
@@ -147,7 +147,7 @@ async def main() -> None:
             if settings.check_website:
                 storage.remove_unchecked()
             count_after_checking = storage.get_count()
-            _console.print(
+            rich.print(
                 get_summary_table(
                     before=count_before_checking, after=count_after_checking
                 )
diff --git a/proxy_scraper_checker/logs.py b/proxy_scraper_checker/logs.py
index 02f9323..6c8c698 100644
--- a/proxy_scraper_checker/logs.py
+++ b/proxy_scraper_checker/logs.py
@@ -8,25 +8,20 @@ from typing import TYPE_CHECKING
 if TYPE_CHECKING:
     from typing import Any
 
-    from rich.console import Console
 
-
-def configure() -> tuple[Console, logging.handlers.QueueListener]:
+def configure() -> logging.handlers.QueueListener:
     log_queue: queue.Queue[Any] = queue.Queue()
 
     logging.root.setLevel(logging.INFO)
     logging.root.addHandler(logging.handlers.QueueHandler(log_queue))
 
     # Start logging before importing rich for the first time
-    from rich.console import Console  # noqa: PLC0415
     from rich.logging import RichHandler  # noqa: PLC0415
 
-    console = Console()
     stream_handler = RichHandler(
-        console=console,
         omit_repeated_times=False,
         show_path=False,
         log_time_format=logging.Formatter.default_time_format,
     )
 
-    return console, logging.handlers.QueueListener(log_queue, stream_handler)
+    return logging.handlers.QueueListener(log_queue, stream_handler)