style: ANSI art

Author: leonardo-martins-dev

src/brightdata/cli/banner.py

@@ -0,0 +1,100 @@
+"""
+ANSI art banner for Bright Data Python SDK CLI.
+"""
+
+import sys
+import os
+
+
+def _supports_color() -> bool:
+    """Check if terminal supports ANSI colors."""
+    # Check if we're in a terminal
+    if not hasattr(sys.stdout, 'isatty') or not sys.stdout.isatty():
+        return False
+    
+    # Windows 10+ supports ANSI colors
+    if sys.platform == "win32":
+        # Check if Windows version supports ANSI
+        try:
+            import ctypes
+            kernel32 = ctypes.windll.kernel32
+            # Enable ANSI escape sequences on Windows
+            kernel32.SetConsoleMode(kernel32.GetStdHandle(-11), 7)
+            return True
+        except:
+            return False
+    
+    # Check for common environment variables
+    if os.getenv("TERM") in ("xterm", "xterm-256color", "screen", "screen-256color"):
+        return True
+    
+    return False
+
+
+def get_banner() -> str:
+    """
+    Get ANSI art banner for Bright Data Python SDK.
+    
+    Returns:
+        Formatted banner string with colors
+    """
+    banner = """
+    
+    \033[1;33m██████╗ ██████╗ ██╗ ██████╗ ██╗  ██╗████████╗\033[0m
+    \033[1;33m██╔══██╗██╔══██╗██║██╔════╝ ██║  ██║╚══██╔══╝\033[0m
+    \033[1;33m██████╔╝██████╔╝██║██║  ███╗███████║   ██║   \033[0m
+    \033[1;33m██╔══██╗██╔══██╗██║██║   ██║██╔══██║   ██║   \033[0m
+    \033[1;33m██████╔╝██║  ██║██║╚██████╔╝██║  ██║   ██║   \033[0m
+    \033[1;33m╚═════╝ ╚═╝  ╚═╝╚═╝ ╚═════╝ ╚═╝  ╚═╝   ╚═╝   \033[0m
+    
+    \033[1;35m██████╗  █████╗ ████████╗ █████╗ \033[0m
+    \033[1;35m██╔══██╗██╔══██╗╚══██╔══╝██╔══██╗\033[0m
+    \033[1;35m██║  ██║███████║   ██║   ███████║\033[0m
+    \033[1;35m██║  ██║██╔══██║   ██║   ██╔══██║\033[0m
+    \033[1;35m██████╔╝██║  ██║   ██║   ██║  ██║\033[0m
+    \033[1;35m╚═════╝ ╚═╝  ╚═╝   ╚═╝   ╚═╝  ╚═╝\033[0m
+    
+    \033[1;32m██████╗ ██╗   ██╗████████╗██╗  ██╗ ██████╗ ███╗   ██╗\033[0m
+    \033[1;32m██╔══██╗╚██╗ ██╔╝╚══██╔══╝██║  ██║██╔═══██╗████╗  ██║\033[0m
+    \033[1;32m██████╔╝ ╚████╔╝    ██║   ███████║██║   ██║██╔██╗ ██║\033[0m
+    \033[1;32m██╔═══╝   ╚██╔╝     ██║   ██╔══██║██║   ██║██║╚██╗██║\033[0m
+    \033[1;32m██║        ██║      ██║   ██║  ██║╚██████╔╝██║ ╚████║\033[0m
+    \033[1;32m╚═╝        ╚═╝      ╚═╝   ╚═╝  ╚═╝ ╚═════╝ ╚═╝  ╚═══╝\033[0m
+    
+    \033[1;37m███████╗██████╗ ██╗  ██╗\033[0m
+    \033[1;37m██╔════╝██╔══██╗██║ ██╔╝\033[0m
+    \033[1;37m███████╗██║  ██║█████╔╝ \033[0m
+    \033[1;37m╚════██║██║  ██║██╔═██╗ \033[0m
+    \033[1;37m███████║██████╔╝██║  ██╗\033[0m
+    \033[1;37m╚══════╝╚═════╝ ╚═╝  ╚═╝\033[0m
+    
+    \033[1;93m🐍\033[0m
+    
+    """
+    return banner
+
+
+def print_banner() -> None:
+    """Print the banner to stdout with proper encoding and color support."""
+    # Enable color support on Windows
+    supports_color = _supports_color()
+    
+    banner = get_banner()
+    
+    # If no color support, strip ANSI codes
+    if not supports_color:
+        import re
+        # Remove ANSI escape sequences
+        ansi_escape = re.compile(r'\x1B(?:[@-Z\\-_]|\[[0-?]*[ -/]*[@-~])')
+        banner = ansi_escape.sub('', banner)
+    
+    # Ensure UTF-8 encoding for Windows compatibility
+    try:
+        if hasattr(sys.stdout, 'buffer') and sys.stdout.encoding != 'utf-8':
+            sys.stdout.buffer.write(banner.encode('utf-8'))
+            sys.stdout.buffer.write(b'\n')
+        else:
+            print(banner)
+    except (AttributeError, UnicodeEncodeError):
+        # Fallback: print without special characters
+        print(banner.encode('ascii', 'ignore').decode('ascii'))

src/brightdata/cli/main.py

@@ -6,14 +6,22 @@
 
 import click
 import sys
+import io
 
 from .commands import scrape_group, search_group
+from .banner import print_banner
+from .utils import handle_error
 
 
-@click.group()
+@click.group(invoke_without_command=True)
 @click.version_option(version="2.0.0", prog_name="brightdata")
+@click.option(
+    "--banner/--no-banner",
+    default=True,
+    help="Show/hide banner on startup"
+)
 @click.pass_context
-def cli(ctx: click.Context) -> None:
+def cli(ctx: click.Context, banner: bool) -> None:
     """
     Bright Data CLI - Command-line interface for Bright Data SDK.
     
@@ -27,6 +35,16 @@ def cli(ctx: click.Context) -> None:
     ctx.ensure_object(dict)
     # Store context for subcommands
     ctx.obj["api_key"] = None
+    
+    # Show banner when invoked without subcommand and not --help/--version
+    if ctx.invoked_subcommand is None and banner:
+        # Check if help or version was requested
+        import sys
+        if "--help" not in sys.argv and "--version" not in sys.argv:
+            print_banner()
+            click.echo()
+            click.echo("Run 'brightdata --help' to see available commands.")
+            click.echo()
 
 
 # Register command groups