Infer staging regions from write tokens (#1037)

Author: alexmojaki

logfire/_internal/config.py

@@ -1748,6 +1748,12 @@ def get_base_url_from_token(token: str) -> str:
     region = 'us'
     if match := PYDANTIC_LOGFIRE_TOKEN_PATTERN.match(token):
         region = match.group('region')
+
+        if region == 'stagingus':
+            return 'https://logfire-us.pydantic.info'
+        elif region == 'stagingeu':
+            return 'https://logfire-eu.pydantic.info'
+
         if region not in REGIONS:
             region = 'us'
 

tests/test_configure.py

@@ -50,6 +50,7 @@
     LogfireConfig,
     LogfireCredentials,
     _get_token_repr,  # type: ignore
+    get_base_url_from_token,
     sanitize_project_name,
 )
 from logfire._internal.exporters.console import ConsoleLogExporter, ShowParentsConsoleSpanExporter
@@ -2272,3 +2273,8 @@ def export(self, spans: Sequence[ReadableSpan]) -> SpanExportResult:
 
     assert exporter.export([]) == SpanExportResult.FAILURE
     assert not caplog.messages
+
+
+def test_staging_token_regions():
+    assert get_base_url_from_token('pylf_v1_stagingeu_123456') == 'https://logfire-eu.pydantic.info'
+    assert get_base_url_from_token('pylf_v1_stagingus_123456') == 'https://logfire-us.pydantic.info'