Updates

src/fom_rest_node.py

@@ -1 +1,187 @@
 """Interface for FOM Rest Node"""
+
+import json
+import traceback
+from argparse import ArgumentParser, Namespace
+from contextlib import asynccontextmanager
+
+from fastapi import FastAPI
+from fastapi.responses import JSONResponse
+from fom_driver import FomDriver
+from wei.core.data_classes import (
+    ModuleStatus,
+    StepFileResponse,
+    StepResponse,
+    StepStatus,
+)
+
+
+def parse_args() -> Namespace:
+    """Parses the command line arguments for the fom REST node"""
+
+    parser = ArgumentParser()
+    parser.add_argument("--host", type=str, default="0.0.0.0", help="Host IP")
+    parser.add_argument("--port", type=str, default="3019", help="Port")
+    parser.add_argument(
+        "--fom_url", type=str, default="http://127.0.0.1:8080", help="URL for fom"
+    )
+
+    return parser.parse_args()
+
+
+global state, module_resources
+
+
+@asynccontextmanager
+async def lifespan(app: FastAPI):
+    """Initial run function for the fom app, initializes the state
+    Parameters
+    ----------
+    app : FastApi
+       The REST API app being initialized
+
+    Returns
+    -------
+    None"""
+    global fom, state, module_resources
+
+    try:
+        # Do any instrument configuration here
+        args = parse_args()
+        fom = FomDriver(url=args.fom_url)
+        state = ModuleStatus.IDLE
+        module_resources = []
+    except Exception as err:
+        print(err)
+        state = ModuleStatus.ERROR
+
+    # Yield control to the application
+    yield
+
+    # Do any cleanup here
+    pass
+
+
+app = FastAPI(
+    lifespan=lifespan,
+)
+
+
+@app.get("/state")
+def get_state():
+    """Returns the current state of the module"""
+    global state
+    return JSONResponse(content={"State": state})
+
+
+@app.get("/about")
+async def about():
+    """Returns a description of the actions and resources the module supports"""
+    global state
+    return JSONResponse(
+        content={
+            "name": "fom",
+            "model": "Unknown",
+            "version": "0.0.1",
+            "actions": {
+                "open_gate": "config : %s",
+                "close_gate": "config : %s",
+                "run_fom": "config : %s",
+            },
+            "repo": "https://github.com/AD-SDL/fom_module.git",
+        }
+    )
+
+
+@app.get("/resources")
+async def resources():
+    """Returns the current resources available to the module"""
+    global state, module_resources
+    resource_info = ""
+    if not (module_resources == ""):
+        with open(module_resources) as f:
+            resource_info = f.read()
+    return JSONResponse(content={"Resources": resource_info})
+
+
+@app.post("/action")
+def do_action(
+    action_handle: str,  # The action to be performed
+    action_vars: str,  # Any arguments necessary to run that action
+) -> StepResponse:
+    """Execute action"""
+
+    global fom, state
+    if state == ModuleStatus.BUSY:
+        return StepResponse(
+            action_response=StepStatus.FAILED,
+            action_msg="",
+            action_log="Fom is busy",
+        )
+    state = ModuleStatus.BUSY
+    action_args = json.loads(action_vars)  # Noqa
+    # speed = action_args.get("speed", None)
+    try:
+        if action_handle == "load_protocol":
+            result = fom.load_protocol()
+
+            state = ModuleStatus.IDLE
+            return StepResponse(
+                action_response=StepStatus.SUCCEEDED,
+                action_msg=result["action_msg"],
+                action_log=result["action_log"],
+            )
+
+        elif action_handle == "run_protocol":
+            result = fom.run_protocol()
+
+            state = ModuleStatus.IDLE
+            return StepResponse(
+                action_response=StepStatus.SUCCEEDED,
+                action_msg=result["action_msg"],
+                action_log=result["action_log"],
+            )
+
+        elif action_handle == "get_output_file":
+            result = fom.get_output_file()
+
+            file_name = result["action_msg"]
+
+            state = ModuleStatus.IDLE
+            return StepFileResponse(
+                action_response=StepStatus.SUCCEEDED,
+                path=file_name,
+                action_log=result["action_log"],
+            )
+
+        else:
+            # Handle Unsupported actions
+            state = ModuleStatus.IDLE
+            return StepResponse(
+                action_response=StepStatus.FAILED,
+                action_msg="",
+                action_log="Unsupported action",
+            )
+    except Exception as e:
+        print(str(e))
+        print(traceback.format_exc())
+        state = ModuleStatus.IDLE
+        return StepResponse(
+            action_response=StepStatus.FAILED,
+            action_msg="",
+            action_log=str(e),
+        )
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    # Add any additional arguments here
+    args = parse_args()
+
+    uvicorn.run(
+        "fom_rest_node:app",
+        host=args.host,
+        port=int(args.port),
+        reload=True,
+    )