"""Logging utilities.""" import logging import sys from datetime import datetime from typing import Optional from rich.console import Console from rich.logging import RichHandler from src.config import get_settings def setup_logging(level: Optional[str] = None) -> None: """ Set up application logging with rich formatting. Args: level: Log level (DEBUG, INFO, WARNING, ERROR). Defaults to settings. """ settings = get_settings() log_level = level or settings.log_level # Create rich console console = Console() # Configure root logger logging.basicConfig( level=log_level, format="%(message)s", datefmt="[%X]", handlers=[ RichHandler( console=console, rich_tracebacks=True, show_path=False, ) ], ) # Reduce noise from third-party libraries logging.getLogger("httpx").setLevel(logging.WARNING) logging.getLogger("httpcore").setLevel(logging.WARNING) logging.getLogger("openai").setLevel(logging.WARNING) logging.getLogger("web3").setLevel(logging.WARNING) def get_logger(name: str) -> logging.Logger: """ Get a logger instance. Args: name: Logger name (usually __name__) Returns: Logger instance """ return logging.getLogger(name) class WhaleWatcherLogger: """Custom logger for whale watcher with formatted output.""" def __init__(self): self.console = Console() self.logger = logging.getLogger("whale_watcher") def whale_detected( self, amount: float, side: str, price: float, market: str, ) -> None: """Log a whale trade detection.""" self.console.print( f"\n[bold cyan]{'='*60}[/bold cyan]\n" f"[bold yellow]🐋 WHALE TRADE DETECTED![/bold yellow]\n" f"[bold cyan]{'='*60}[/bold cyan]\n" f"[green]Amount:[/green] ${amount:,.2f} USDC\n" f"[green]Side:[/green] {side}\n" f"[green]Price:[/green] {price:.4f}\n" f"[green]Market:[/green] {market}\n" f"[green]Time:[/green] {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}\n" f"[bold cyan]{'='*60}[/bold cyan]\n" ) def report_generated(self, market: str) -> None: """Log that a report was generated.""" self.console.print( f"\n[bold magenta]{'='*60}[/bold magenta]\n" f"[bold magenta]📊 ANALYSIS REPORT GENERATED[/bold magenta]\n" f"[bold magenta]{'='*60}[/bold magenta]\n" f"[green]Market:[/green] {market[:50]}...\n" f"[bold magenta]{'='*60}[/bold magenta]\n" ) def monitoring_started(self, market_count: int, interval: int = 0, min_trade_size: float = 1000, min_price: float = 0.2, max_price: float = 0.8) -> None: """Log monitoring start.""" self.console.print( f"\n[bold green]{'='*60}[/bold green]\n" f"[bold green]🚀 WHALE WATCHER STARTED (RTDS WebSocket)[/bold green]\n" f"[bold green]{'='*60}[/bold green]\n" f"[green]Monitored Markets:[/green] {market_count}\n" f"[green]Mode:[/green] Real-time WebSocket (zero missed trades)\n" f"[green]Min Trade Size:[/green] ${min_trade_size:,.0f} USD\n" f"[green]Price Range:[/green] {min_price} - {max_price}\n" f"[bold green]{'='*60}[/bold green]\n" ) def error(self, message: str) -> None: """Log an error.""" self.console.print(f"[bold red]❌ ERROR:[/bold red] {message}") def info(self, message: str) -> None: """Log an info message.""" self.console.print(f"[blue]â„šī¸[/blue] {message}") def separator(self) -> None: """Print a separator line.""" self.console.print(f"[dim]{'─'*60}[/dim]")