Added taskification to action builder

Author: Joel Collins

examples/builder.py

@@ -43,7 +43,9 @@ def cleanup():
     "/dictionary",
 )
 labthing.add_view(
-    action_from(my_component.average_data, description="Take an averaged measurement"),
+    action_from(
+        my_component.average_data, description="Take an averaged measurement", task=True
+    ),
     "/average",
 )
 

labthings/server/view/builder.py

@@ -1,3 +1,4 @@
+from labthings.core.tasks import taskify
 from labthings.server.types import (
     value_to_field,
     data_dict_to_schema,
@@ -7,6 +8,7 @@
     ThingProperty,
     PropertySchema,
     ThingAction,
+    marshal_task,
     use_args,
     Doc,
 )
@@ -72,12 +74,19 @@ def _put(self, args):
     return generated_class
 
 
-def action_from(function, name: str = None, description=None):
+def action_from(function, name: str = None, description=None, task=False):
 
     # Create a class name
     if not name:
         name = f"Action_{function.__name__}"
 
+    # Create schema
+    action_schema = function_signature_to_schema(function)
+
+    # Handle taskification
+    if task:
+        function = taskify(function)
+
     # Create inner functions
     def _post(self, args):
         return function(**args)
@@ -86,9 +95,12 @@ def _post(self, args):
     generated_class = type(name, (View, object), {"post": _post})
 
     # Add decorators for arguments etc
-    action_schema = function_signature_to_schema(function)
+
     generated_class.post = use_args(action_schema)(generated_class.post)
 
+    if task:
+        generated_class.post = marshal_task(generated_class.post)
+
     generated_class = ThingAction(generated_class)
 
     if description: