isolate 0.13.9__tar.gz → 0.13.10__tar.gz

{isolate-0.13.9 → isolate-0.13.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.13.9
+Version: 0.13.10
 Summary: Managed isolated environments for Python
 Author-email: Features & Labels <hello@fal.ai>
 Project-URL: Issues, https://github.com/fal-ai/isolate/issues

{isolate-0.13.9 → isolate-0.13.10}/src/isolate/_isolate_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.13.9'
-__version_tuple__ = version_tuple = (0, 13, 9)
+__version__ = version = '0.13.10'
+__version_tuple__ = version_tuple = (0, 13, 10)

{isolate-0.13.9 → isolate-0.13.10}/src/isolate/logger.py

@@ -1,15 +1,29 @@
 import json
+import os
 
 
 # NOTE: we probably should've created a proper `logging.getLogger` here,
 # but it handling `source` would be not trivial, so we are better off
 # just keeping it simple for now.
 class IsolateLogger:
+    def __init__(self):
+        self.log_labels = {}
+        raw = os.getenv("ISOLATE_LOG_LABELS")
+        if raw:
+            labels = json.loads(raw)
+            for key, value in labels.items():
+                if value.startswith("$"):
+                    expanded = os.getenv(value[1:])
+                else:
+                    expanded = value
+                self.log_labels[key] = expanded
+
     def log(self, level, message, source):
         record = {
             "isolate_source": source.name,
             "level": level.name,
             "message": message,
+            **self.log_labels,
         }
         print(json.dumps(record))
 

{isolate-0.13.9 → isolate-0.13.10}/src/isolate.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: isolate
-Version: 0.13.9
+Version: 0.13.10
 Summary: Managed isolated environments for Python
 Author-email: Features & Labels <hello@fal.ai>
 Project-URL: Issues, https://github.com/fal-ai/isolate/issues

{isolate-0.13.9 → isolate-0.13.10}/src/isolate.egg-info/SOURCES.txt

@@ -73,6 +73,7 @@ tests/test_concurrency.py
 tests/test_connections.py
 tests/test_isolate.py
 tests/test_log.py
+tests/test_logger.py
 tests/test_serialization.py
 tests/test_server.py
 tools/Dockerfile

isolate-0.13.10/tests/test_logger.py

@@ -0,0 +1,22 @@
+import json
+
+from isolate.logger import IsolateLogger
+
+
+def test_logger(monkeypatch):
+    labels = {
+        "foo": "$MYENVVAR1",
+        "bar": "$MYENVVAR2",
+        "baz": "baz",
+        "qux": "qux",
+    }
+    monkeypatch.setenv("MYENVVAR1", "myenvvar1")
+    monkeypatch.setenv("MYENVVAR2", "myenvvar2")
+    monkeypatch.setenv("ISOLATE_LOG_LABELS", json.dumps(labels))
+    logger = IsolateLogger()
+    assert logger.log_labels == {
+        "foo": "myenvvar1",
+        "bar": "myenvvar2",
+        "baz": "baz",
+        "qux": "qux",
+    }