expose Run.wd variable via scheduler API

Signed-off-by: Brad Bondurant <[email protected]>

artiq/frontend/artiq_master.py

@@ -123,6 +123,7 @@ def ccb_issue(service, *args, **kwargs):
         "scheduler_request_termination": scheduler.request_termination,
         "scheduler_get_status": scheduler.get_status,
         "scheduler_check_pause": scheduler.check_pause,
+        "scheduler_get_wd": scheduler.get_wd,
         "ccb_issue": ccb_issue,
     })
     experiment_db.scan_repository_async()

artiq/master/scheduler.py

@@ -479,3 +479,11 @@ def check_pause(self, rid):
                     return False
                 return r.priority_key() > run.priority_key()
         raise KeyError("RID not found")
+
+    def get_wd(self, rid):
+        """Returns the ``wd`` attribute of the run with the specified RID."""
+        for pipeline in self._pipelines.values():
+            if rid in pipeline.pool.runs:
+                run = pipeline.pool.runs[rid]
+                return run.wd
+        raise KeyError("RID not found")

artiq/master/worker_impl.py

@@ -124,6 +124,13 @@ def submit(self, pipeline_name=None, expid=None, priority=None, due_date=None, f
         make_parent_action("scheduler_request_termination"))
     get_status = staticmethod(make_parent_action("scheduler_get_status"))
 
+    _get_wd = staticmethod(make_parent_action("scheduler_get_wd"))
+    @host_only
+    def get_wd(self, rid=None):
+        if rid is None:
+            rid = self.rid
+        return self._get_wd(rid)
+
 
 class CCB:
     issue = staticmethod(make_parent_action("ccb_issue"))