bdd-trace 0.5.4__py3-none-any.whl → 0.5.5__py3-none-any.whl

bdd_trace/log.py

@@ -5,21 +5,25 @@ from queue import Queue
 
 from opentelemetry.sdk._logs import LoggingHandler
 
-from bdd_trace.trace import Profile
+from .trace import Profile, _get_profile
 
 _log_queue: Queue | None = None
 _log_listener: QueueListener | None = None
 
 
 def setup_logging(
-    profile: Profile,
     modules: list[str],
+    profile: Profile | None = None,
     level: str | int = INFO,
     enable_console_log: bool = True,
     trace_id_format: str = "%(asctime)s|%(levelname)s|%(name)s|%(otelTraceID)s|%(message)s",
     no_trace_id_format: str = "%(asctime)s|%(levelname)s|%(name)s|%(message)s",
     extra_handlers: list[Handler] | None = None,
 ) -> None:
+    profile = _get_profile(profile)
+    if profile is None:
+        raise ValueError("profile is not set")
+
     global _log_queue, _log_listener
     if _log_queue is not None or _log_listener is not None:
         raise RuntimeError("logging is already setup")

bdd_trace/trace.py

@@ -12,6 +12,7 @@ class Profile(str, Enum):
     PROD = "prod"
 
 
+_TRACE_PROFILE_KEY = "BDD_TRACE_PROFILE"
 _TRACES_EXPORTER_KEY = "traces_exporter"
 _METRICS_EXPORTER_KEY = "metrics_exporter"
 _LOGS_EXPORTER_KEY = "logs_exporter"
@@ -73,9 +74,9 @@ def init_trace(
     for key, value in kwargs.items():
         _set_env(key, value)
 
-    if profile:
+    if profile := _get_profile(profile):
         if profile == Profile.NO_TRACE:
-            logger.debug("NO_TRACE profile is set, skip auto instrumentation")
+            logger.info("NO_TRACE profile is set, skip auto instrumentation")
             return
         for key, value in _profile_config[profile].items():
             _set_env(key, value)
@@ -95,6 +96,18 @@ def init_trace(
     auto_instrumentation.initialize()
 
 
+def _get_profile(profile: Profile | None) -> Profile | None:
+    if profile:
+        return profile
+    profile_env = os.getenv(_TRACE_PROFILE_KEY)
+    if profile_env:
+        try:
+            return Profile(profile_env)
+        except ValueError:
+            raise ValueError(f"invalid profile from env {_TRACE_PROFILE_KEY}: {profile_env}")
+    return None
+
+
 def _set_env(key: str, value: str | None) -> None:
     if value is None:
         return

{bdd_trace-0.5.4.dist-info → bdd_trace-0.5.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: bdd-trace
-Version: 0.5.4
+Version: 0.5.5
 Summary: BDD项目日志和追踪SDK
 Author: 蔡涛
 Author-email: 蔡涛 <caitao@zhejianglab.cn>

bdd_trace-0.5.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+bdd_trace/__init__.py,sha256=X4Nx6VppPfXXXEe5oC_foryalbOk5f9HLOi50Nat8mY,154
+bdd_trace/log.py,sha256=pGDatl39U21DC0J8UMRgDv9M4InCa0536YXSZwb3dp4,3263
+bdd_trace/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+bdd_trace/trace.py,sha256=BozZyrqV3Vjl36Ay3A-JpkCAQkQLLjMfMx0I47Oop2M,4209
+bdd_trace-0.5.5.dist-info/WHEEL,sha256=F3mArEuDT3LDFEqo9fCiUx6ISLN64aIhcGSiIwtu4r8,79
+bdd_trace-0.5.5.dist-info/METADATA,sha256=T-nb99JNOGCq_Sksy0D6Jc0cbqa9QNJuynfOjRUMeAg,1585
+bdd_trace-0.5.5.dist-info/RECORD,,

{bdd_trace-0.5.4.dist-info → bdd_trace-0.5.5.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: uv 0.8.15
+Generator: uv 0.8.16
 Root-Is-Purelib: true
 Tag: py3-none-any

bdd_trace-0.5.4.dist-info/RECORD

@@ -1,7 +0,0 @@
-bdd_trace/__init__.py,sha256=X4Nx6VppPfXXXEe5oC_foryalbOk5f9HLOi50Nat8mY,154
-bdd_trace/log.py,sha256=HqkUiomCJlR3OsbHMSOYFe4mt5WitYcFhRqBmK9pBPM,3136
-bdd_trace/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bdd_trace/trace.py,sha256=wmzwkcHtV7CslLFKzpIFjSv-55WsqX28uPfsgYDvumA,3784
-bdd_trace-0.5.4.dist-info/WHEEL,sha256=Jb20R3Ili4n9P1fcwuLup21eQ5r9WXhs4_qy7VTrgPI,79
-bdd_trace-0.5.4.dist-info/METADATA,sha256=IuseK2ZM3uoiF4UMT2EXohHzYngc0gq8-L62CC6MTj4,1585
-bdd_trace-0.5.4.dist-info/RECORD,,