|
10 | 10 | import time |
11 | 11 | import asyncio |
12 | 12 | import os |
| 13 | +import logging |
| 14 | +import sys |
13 | 15 |
|
14 | 16 |
|
15 | 17 | from codewiki.cli.utils.progress import ProgressTracker |
@@ -64,6 +66,51 @@ def __init__( |
64 | 66 | cluster_model=config.get('cluster_model', ''), |
65 | 67 | base_url=config.get('base_url', '') |
66 | 68 | ) |
| 69 | + |
| 70 | + # Configure backend logging |
| 71 | + self._configure_backend_logging() |
| 72 | + |
| 73 | + def _configure_backend_logging(self): |
| 74 | + """Configure backend logger for CLI use.""" |
| 75 | + # Get backend logger (parent of all backend modules) |
| 76 | + backend_logger = logging.getLogger('codewiki.src.be') |
| 77 | + |
| 78 | + # Remove existing handlers to avoid duplicates |
| 79 | + backend_logger.handlers.clear() |
| 80 | + |
| 81 | + if self.verbose: |
| 82 | + # In verbose mode, show DEBUG and above |
| 83 | + backend_logger.setLevel(logging.DEBUG) |
| 84 | + |
| 85 | + # Create console handler with formatting |
| 86 | + console_handler = logging.StreamHandler(sys.stdout) |
| 87 | + console_handler.setLevel(logging.DEBUG) |
| 88 | + |
| 89 | + # Create formatter with timestamp and minimal formatting for cleaner output |
| 90 | + formatter = logging.Formatter( |
| 91 | + '[%(asctime)s] %(message)s', |
| 92 | + datefmt='%H:%M:%S' |
| 93 | + ) |
| 94 | + console_handler.setFormatter(formatter) |
| 95 | + |
| 96 | + # Add handler to logger |
| 97 | + backend_logger.addHandler(console_handler) |
| 98 | + else: |
| 99 | + # In non-verbose mode, suppress backend logs (use WARNING level to hide INFO/DEBUG) |
| 100 | + backend_logger.setLevel(logging.WARNING) |
| 101 | + |
| 102 | + # Create console handler for warnings and errors only |
| 103 | + console_handler = logging.StreamHandler(sys.stderr) |
| 104 | + console_handler.setLevel(logging.WARNING) |
| 105 | + |
| 106 | + # Simple formatter for warnings/errors |
| 107 | + formatter = logging.Formatter('%(levelname)s: %(message)s') |
| 108 | + console_handler.setFormatter(formatter) |
| 109 | + |
| 110 | + backend_logger.addHandler(console_handler) |
| 111 | + |
| 112 | + # Prevent propagation to root logger to avoid duplicate messages |
| 113 | + backend_logger.propagate = False |
67 | 114 |
|
68 | 115 | def generate(self) -> DocumentationJob: |
69 | 116 | """ |
|
0 commit comments