trying to avoid daisy_logs artifact after running tests

daisy/client.py

@@ -1,5 +1,6 @@
 from .block import BlockStatus
 from .context import Context
+from .logging import set_log_basedir
 from .messages import (
     AcquireBlock,
     BlockFailed,
@@ -56,6 +57,7 @@ def __init__(self, context=None):
             self.context = Context.from_env()
         logger.debug("Client context: %s", self.context)
 
+        set_log_basedir(self.context["logdir"])
         self.host = self.context["hostname"]
         self.port = int(self.context["port"])
         self.worker_id = int(self.context["worker_id"])

daisy/context.py

@@ -1,6 +1,7 @@
 import copy
 import logging
 import os
+from .logging import get_log_basedir
 
 logger = logging.getLogger(__name__)
 
@@ -11,7 +12,7 @@ class Context:
 
     def __init__(self, **kwargs):
 
-        self.__dict = dict(**kwargs)
+        self.__dict = dict(logdir=get_log_basedir(), **kwargs)
 
     def copy(self):
 

daisy/logging.py

@@ -22,10 +22,16 @@ def set_log_basedir(path):
     else:
         LOG_BASEDIR = None
 
+def get_log_basedir():
+    """Get the base directory for logging."""
+    global LOG_BASEDIR
+    return LOG_BASEDIR
+
 
 def get_worker_log_basename(worker_id, task_id=None):
     """Get the basename of log files for individual workers."""
-
+    global LOG_BASEDIR
+
     if LOG_BASEDIR is None:
         return None
 

daisy/server.py

@@ -2,6 +2,7 @@
 from .block import BlockStatus
 from .block_bookkeeper import BlockBookkeeper
 from .context import Context
+from .logging import get_log_basedir
 from .messages import (
     AcquireBlock,
     BlockFailed,

daisy/task_worker_pools.py

@@ -1,4 +1,5 @@
 from .context import Context
+from .logging import get_log_basedir
 from .server_observer import ServerObserver
 from .worker_pool import WorkerPool
 import logging

tests/conf.py

@@ -0,0 +1,8 @@
+import pytest
+
+from daisy.logging import set_log_basedir, get_log_basedir
+
+
+@pytest.fixture(autouse=True)
+def logdir(tmp_path):
+    set_log_basedir(tmp_path / "daisy_logs")