feat: console output option (#1477)

Co-authored-by: Alex Hall <[email protected]>

Author: stevenh

logfire/_internal/config.py

@@ -113,6 +113,8 @@
 )
 
 if TYPE_CHECKING:
+    from typing import TextIO
+
     from opentelemetry._events import EventLoggerProvider
 
     from .main import Logfire
@@ -159,6 +161,8 @@ class ConsoleOptions:
 
     show_project_link: bool = True
     """Whether to print the URL of the Logfire project after initialization."""
+    output: TextIO | None = None
+    """The output stream to write console output to (default: stdout)."""
 
 
 @dataclass
@@ -894,6 +898,7 @@ def add_span_processor(span_processor: SpanProcessor) -> None:
                     include_tags=self.console.include_tags,
                     verbose=self.console.verbose,
                     min_log_level=self.console.min_log_level,
+                    output=self.console.output,
                 )
                 add_span_processor(SimpleSpanProcessor(console_span_exporter))
                 log_record_processors.append(SimpleLogRecordProcessor(ConsoleLogExporter(console_span_exporter)))

tests/test_configure.py

@@ -7,6 +7,7 @@
 import threading
 from collections.abc import Iterable, Sequence
 from contextlib import ExitStack
+from io import StringIO
 from pathlib import Path
 from time import sleep, time
 from typing import Any
@@ -881,6 +882,19 @@ def test_config_serializable_console_false():
     assert GLOBAL_CONFIG.console is False
 
 
+def test_config_console_output_set():
+    output = StringIO()
+    logfire.configure(send_to_logfire=False, console=logfire.ConsoleOptions(output=output))
+    assert isinstance(GLOBAL_CONFIG.console, logfire.ConsoleOptions)
+    assert GLOBAL_CONFIG.console.output is output
+
+    deserialize_config(serialize_config())
+    assert isinstance(GLOBAL_CONFIG.console, logfire.ConsoleOptions)
+    assert GLOBAL_CONFIG.console.output is output
+    logfire.info('test')
+    assert 'test' in output.getvalue()
+
+
 def test_sanitize_project_name():
     assert sanitize_project_name('foo') == 'foo'
     assert sanitize_project_name('FOO') == 'foo'