Restructured labthings

openflexure_microscope/common/flask_labthings/views/plugins.py

@@ -0,0 +1,64 @@
+"""
+Top-level representation of attached and enabled plugins
+"""
+
+from openflexure_microscope.common.labthings_core.utilities import get_docstring
+from ..utilities import description_from_view
+
+from openflexure_microscope.common.flask_labthings.find import registered_plugins
+from openflexure_microscope.common.flask_labthings.resource import Resource
+
+from flask import jsonify, url_for
+
+import logging
+
+
+def plugins_representation(plugin_dict):
+    """
+    Generate a dictionary representation of all plugins, including Flask route URLs
+
+    Args:
+        plugin_dict (dict): Dictionary of plugin objects
+
+    Returns:
+        dict: Dictionary representation of all plugins
+    """
+    plugins = {}
+
+    for plugin_name, plugin in plugin_dict.items():
+        logging.debug(f"Representing plugin {plugin._name}")
+        d = {
+            "python_name": plugin._name_python_safe,
+            "plugin": str(plugin),
+            "links": {},
+            "gui": plugin.gui,
+            "description": get_docstring(plugin),
+        }
+
+        for view_id, view_data in plugin.views.items():
+            uri = url_for(f"PluginListResource", _external=True) + "/" + view_data["rule"][1:]
+            # Make links dictionary if it doesn't yet exist
+            view_d = {"href": uri, **description_from_view(view_data["view"])}
+
+            d["links"][view_id] = view_d
+
+        plugins[plugin_name] = d
+
+    return plugins
+
+
+class PluginListResource(Resource):
+    def get(self):
+        """
+        Return the current plugin forms
+
+        .. :quickref: Plugin; Get forms
+
+        Returns an array of present plugin forms (describing plugin user interfaces.)
+        Please note, this is *not* a list of all enabled plugins, only those with associated
+        user interface forms.
+
+        A complete list of enabled plugins can be found in the microscope state.
+
+        """
+        return jsonify(plugins_representation(registered_plugins()))

openflexure_microscope/common/flask_labthings/views/tasks.py

@@ -0,0 +1,59 @@
+from flask import abort
+
+from openflexure_microscope.common.flask_labthings.schema import Schema
+from openflexure_microscope.common.flask_labthings import fields
+from openflexure_microscope.common.labthings_core import tasks
+from openflexure_microscope.common.flask_labthings.resource import Resource
+from openflexure_microscope.common.flask_labthings.utilities import description_from_view
+
+
+class TaskList(Resource):
+    def get(self):
+        return task_list_schema.jsonify(tasks.tasks())
+
+
+class TaskResource(Resource):
+    def get(self, id):
+        try:
+            task = tasks.dict()[id]
+        except KeyError:
+            return abort(404)  # 404 Not Found
+
+        # Return task state
+        return task_schema.jsonify(task)
+
+    def delete(self, id):
+        try:
+            task = tasks.dict()[id]
+        except KeyError:
+            return abort(404)  # 404 Not Found
+
+        task.terminate()
+
+        return task_schema.jsonify(task)
+
+
+class TaskSchema(Schema):
+    _ID = fields.String(data_key="id")
+    target_string = fields.String(data_key="function")
+    _status = fields.String(data_key="status")
+    progress = fields.String()
+    data = fields.Raw()
+    _return_value = fields.Raw(data_key="return")
+    _start_time = fields.String(data_key="start_time")
+    _end_time = fields.String(data_key="end_time")
+
+    # TODO: Add HTTP methods
+    links = fields.Hyperlinks(
+        {
+            "self": {
+                "href": fields.AbsoluteUrlFor(TaskResource, id="<id>"),
+                "mimetype": "application/json",
+                **description_from_view(TaskResource)
+            }
+        }
+    )
+
+
+task_schema = TaskSchema()
+task_list_schema = TaskSchema(many=True)