Dynamic FastCS drivers#

Demo Simulation#

The demo simulation used in the previous tutorial has a command API? to list all of its commands. This allows you to introspect the device and create the API dynamically, instead of defining all the attributes statically. The response will look like this

{
    "Device ID": {"command": "ID", "type": "str", "access_mode": "r"},
    "Power": {"command": "P", "type": "float", "access_mode": "rw"},
    "Ramp Rate": {"command": "R", "type": "float", "access_mode": "rw"},
    "Ramps": [
        {
            "Start": {"command": "S01", "type": "int", "access_mode": "rw"},
            "End": {"command": "E01", "type": "int", "access_mode": "rw"},
            "Enabled": {"command": "N01", "type": "int", "access_mode": "rw"},
            "Target": {"command": "T01", "type": "float", "access_mode": "rw"},
            "Actual": {"command": "A01", "type": "float", "access_mode": "rw"},
        },
        ...,
    ],
}

This contains all the metadata about the parameters in the API needed to create the Attributes from the previous tutorial. For a real device, this might also include fields such as the units of numerical parameters, limits that a parameter can be set to, or a description for the parameter.

FastCS Initialisation#

Specific Controller classes can optionally implement an async initialise method to perform any start up logic. The intention here is that the __init__ method should be minimal and the initialise method performs any long running calls, such as querying an API, allowing FastCS to run these concurrently to reduce start times.

Take the driver implementation from the previous tutorial and remove the statically defined Attributes and creation of sub controllers in __init__. Then implement an initialise method to create these dynamically instead.

Create a pydantic model to validate the response from the device

import json
from dataclasses import dataclass
from typing import Any, Literal
from fastcs.transport.epics.ca.options import EpicsCAOptions
from fastcs.transport.epics.options import EpicsIOCOptions


class TemperatureControllerParameter(BaseModel):
    model_config = ConfigDict(extra="forbid")

    command: str
    type: Literal["bool", "int", "float", "str"]
    access_mode: Literal["r", "rw"]

    @property
    def fastcs_datatype(self) -> DataType:
        match self.type:
            case "bool":
                return Bool()
            case "int":
                return Int()
            case "float":
                return Float()

Create a function to parse the dictionary and validate the entries against the model



def create_attributes(parameters: dict[str, Any]) -> dict[str, Attribute]:
    attributes: dict[str, Attribute] = {}
    for name, parameter in parameters.items():
        name = name.replace(" ", "_").lower()

        try:
            parameter = TemperatureControllerParameter.model_validate(parameter)
        except ValidationError as e:
            print(f"Failed to validate parameter '{parameter}'\n{e}")
            continue

        handler = TemperatureControllerHandler(parameter.command)
        match parameter.access_mode:
            case "r":
                attributes[name] = AttrR(parameter.fastcs_datatype, handler=handler)
            case "rw":
                attributes[name] = AttrRW(parameter.fastcs_datatype, handler=handler)

Update the controllers to not define attributes statically and implement initialise methods to create these attributes dynamically.

        )
        super().__init__(f"Ramp {index}")

        self.connection = connection
        self.attributes.update(create_attributes(parameters))

    async def connect(self):
        await self.connection.connect(self._ip_settings)

    async def initialise(self):
        await self.connect()

        api = json.loads((await self.connection.send_query("API?\r\n")).strip("\r\n"))

        ramps_api = api.pop("Ramps")
        self.attributes.update(create_attributes(api))

        for idx, ramp_parameters in enumerate(ramps_api):
            ramp_controller = TemperatureRampController(idx + 1, self.connection)
            self.register_sub_controller(f"Ramp{idx + 1:02d}", ramp_controller)

The suffix field should also be removed from TemperatureController and TemperatureRampController and then not used in TemperatureControllerHandler because the command field on TemperatureControllerParameter includes this.

TODO: Add enabled back in to TemperatureRampController and recreate disable_all to demonstrate validation of introspected Attributes.

The full code is as follows