Attribute parsing with Controller.attributes

Author: evvaaaa

src/fastcs/controller.py

@@ -3,6 +3,7 @@
 from collections.abc import Iterator
 from copy import copy
 from dataclasses import dataclass
+from typing import get_type_hints
 
 from .attributes import Attribute
 from .cs_methods import Command, Put, Scan
@@ -19,7 +20,12 @@ class SingleMapping:
 
 
 class BaseController:
+    #! Attributes passed from the device at runtime.
+    attributes: dict[str, Attribute]
+
     def __init__(self, path: list[str] | None = None) -> None:
+        if not hasattr(self, "attributes"):
+            self.attributes = {}
         self._path: list[str] = path or []
         self.__sub_controller_tree: dict[str, BaseController] = {}
 
@@ -37,12 +43,26 @@ def set_path(self, path: list[str]):
         self._path = path
 
     def _bind_attrs(self) -> None:
-        for attr_name in dir(self):
-            attr = getattr(self, attr_name)
+        # Using a dictionary instead of a set to maintain order.
+        class_dir = {key: None for key in dir(type(self))}
+        class_type_hints = get_type_hints(type(self))
+
+        for attr_name in {**class_dir, **class_type_hints}:
+            attr = getattr(self, attr_name, None)
             if isinstance(attr, Attribute):
+                if (
+                    attr_name in self.attributes
+                    and self.attributes[attr_name] is not attr
+                ):
+                    raise ValueError(
+                        f"`{type(self).__name__}` has conflicting attribute "
+                        f"`{attr_name}` already present in the attributes dict."
+                    )
                 new_attribute = copy(attr)
                 setattr(self, attr_name, new_attribute)
 
+                self.attributes[attr_name] = new_attribute
+
     def register_sub_controller(self, name: str, sub_controller: SubController):
         if name in self.__sub_controller_tree.keys():
             raise ValueError(
@@ -69,7 +89,6 @@ def _get_single_mapping(controller: BaseController) -> SingleMapping:
     scan_methods: dict[str, Scan] = {}
     put_methods: dict[str, Put] = {}
     command_methods: dict[str, Command] = {}
-    attributes: dict[str, Attribute] = {}
     for attr_name in dir(controller):
         attr = getattr(controller, attr_name)
         match attr:
@@ -79,11 +98,14 @@ def _get_single_mapping(controller: BaseController) -> SingleMapping:
                 scan_methods[attr_name] = scan_method
             case WrappedMethod(fastcs_method=Command(enabled=True) as command_method):
                 command_methods[attr_name] = command_method
-            case Attribute(enabled=True):
-                attributes[attr_name] = attr
 
+    enabled_attributes = {
+        name: attribute
+        for name, attribute in controller.attributes.items()
+        if attribute.enabled
+    }
     return SingleMapping(
-        controller, scan_methods, put_methods, command_methods, attributes
+        controller, scan_methods, put_methods, command_methods, enabled_attributes
     )
 
 

tests/test_controller.py

@@ -1,11 +1,13 @@
 import pytest
 
+from fastcs.attributes import AttrR
 from fastcs.controller import (
     Controller,
     SubController,
     _get_single_mapping,
     _walk_mappings,
 )
+from fastcs.datatypes import Int
 
 
 def test_controller_nesting():
@@ -33,3 +35,59 @@ def test_controller_nesting():
         ValueError, match=r"SubController is already registered under .*"
     ):
         controller.register_sub_controller("c", sub_controller)
+
+
+class SomeSubController(SubController):
+    def __init__(self):
+        super().__init__()
+
+    sub_attribute = AttrR(Int())
+
+    root_attribute = AttrR(Int())
+
+
+class SomeController(Controller):
+    annotated_attr: AttrR
+    annotated_attr_not_defined_in_init: AttrR[int]
+    equal_attr = AttrR(Int())
+    annotated_and_equal_attr: AttrR[int] = AttrR(Int())
+
+    def __init__(self, sub_controller: SubController):
+        self.attributes = {}
+
+        self.annotated_attr = AttrR(Int())
+        self.attr_on_object = AttrR(Int())
+
+        self.attributes["_attributes_attr"] = AttrR(Int())
+        self.attributes["_attributes_attr_equal"] = self.equal_attr
+
+        super().__init__()
+        self.register_sub_controller("sub_controller", sub_controller)
+
+
+def test_attribute_parsing():
+    sub_controller = SomeSubController()
+    controller = SomeController(sub_controller)
+
+    mapping_walk = _walk_mappings(controller)
+
+    controller_mapping = next(mapping_walk)
+    assert set(controller_mapping.attributes.keys()) == {
+        "_attributes_attr",
+        "annotated_attr",
+        "_attributes_attr_equal",
+        "annotated_and_equal_attr",
+        "equal_attr",
+        "sub_controller",
+    }
+
+    assert SomeController.equal_attr is not controller.equal_attr
+    assert (
+        SomeController.annotated_and_equal_attr
+        is not controller.annotated_and_equal_attr
+    )
+
+    sub_controller_mapping = next(mapping_walk)
+    assert sub_controller_mapping.attributes == {
+        "sub_attribute": sub_controller.sub_attribute,
+    }