mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
28 lines
676 B
Python
28 lines
676 B
Python
import io
|
|
import logging
|
|
from contextlib import asynccontextmanager
|
|
|
|
|
|
@asynccontextmanager
|
|
async def capture_logs(logger_name, level=logging.ERROR):
|
|
logger = logging.getLogger(logger_name)
|
|
|
|
# Store original handlers and level
|
|
original_handlers = logger.handlers[:]
|
|
original_level = logger.level
|
|
|
|
# Set up capture
|
|
log_capture = io.StringIO()
|
|
handler = logging.StreamHandler(log_capture)
|
|
handler.setLevel(level)
|
|
|
|
logger.handlers = [handler]
|
|
logger.setLevel(level)
|
|
|
|
try:
|
|
yield log_capture
|
|
finally:
|
|
# Restore original configuration
|
|
logger.handlers = original_handlers
|
|
logger.setLevel(original_level)
|