Add service API for using main thread queue

Author: ctrueden

src/appose/service.py

@@ -109,16 +109,20 @@ def start(self) -> None:
         self._stderr_thread.start()
         self._monitor_thread.start()
 
-    def task(self, script: str, inputs: Args | None = None) -> "Task":
+    def task(
+        self, script: str, inputs: Args | None = None, queue: str | None = None
+    ) -> "Task":
         """
         Create a new task, passing the given script to the worker for execution.
         :param script:
             The script for the worker to execute in its environment.
         :param inputs:
             Optional list of key/value pairs to feed into the script as inputs.
+        :param queue:
+            Optional queue target. Pass "main" to queue to worker's main thread.
         """
         self.start()
-        return Task(self, script, inputs)
+        return Task(self, script, inputs, queue)
 
     def close(self) -> None:
         """
@@ -298,12 +302,17 @@ class Task:
     """
 
     def __init__(
-        self, service: Service, script: str, inputs: Args | None = None
+        self,
+        service: Service,
+        script: str,
+        inputs: Args | None = None,
+        queue: str | None = None,
     ) -> None:
         self.uuid = uuid4().hex
         self.service = service
         self.script = script
         self.inputs: Args = {}
+        self.queue: str | None = queue
         if inputs is not None:
             self.inputs.update(inputs)
         self.outputs: Args = {}
@@ -323,7 +332,7 @@ def start(self) -> "Task":
 
             self.status = TaskStatus.QUEUED
 
-        args = {"script": self.script, "inputs": self.inputs}
+        args = {"script": self.script, "inputs": self.inputs, "queue": self.queue}
         self._request(RequestType.EXECUTE, args)
 
         return self

tests/test_appose.py

@@ -64,6 +64,11 @@ def sqrt_age(age):
 task.outputs["result"] = sqrt_age(age)
 """
 
+main_thread_check = """
+import threading
+task.outputs["thread"] = threading.current_thread().name
+"""
+
 
 def test_groovy():
     env = appose.system()
@@ -105,6 +110,22 @@ def test_scope():
         assert result == 10
 
 
+def test_main_thread_queue():
+    env = appose.system()
+    with env.python() as service:
+        task = service.task(main_thread_check, queue="main")
+        task.start()
+        task.wait_for()
+        thread = task.outputs.get("thread")
+        assert thread == "MainThread"
+
+        task = service.task(main_thread_check)
+        task.start()
+        task.wait_for()
+        thread = task.outputs.get("thread")
+        assert thread != "MainThread"
+
+
 def execute_and_assert(service: Service, script: str):
     task = service.task(script)