WIP

Author: evvaaaa

src/fastcs/controller.py

@@ -44,7 +44,6 @@ def get_sub_controllers(self) -> dict[str, BaseController]:
 
     def get_attributes(self) -> dict[str, Attribute]:
         """For getting any attributes which aren't defined on the class itself."""
-
         return {}
 
 

src/fastcs/mapping.py

@@ -1,7 +1,8 @@
 from collections.abc import Iterator
 from dataclasses import dataclass
+from typing import get_type_hints
 
-from .attributes import Attribute
+from .attributes import Attribute, AttrR, AttrW, AttrRW
 from .controller import BaseController, Controller
 from .cs_methods import Command, Put, Scan
 from .wrappers import WrappedMethod
@@ -52,11 +53,18 @@ def _get_single_mapping(controller: BaseController) -> SingleMapping:
                 command_methods[attr_name] = command_method
 
     attributes: dict[str, Attribute] = {}
-    for name in dir(type(controller)):
-        if isinstance((attr := getattr(controller, name, None)), Attribute):
+    for name in set(get_type_hints(type(controller))) | set(dir(controller)):
+        if (
+            isinstance(
+                (attr := getattr(controller, name, None)), AttrRW | AttrR | AttrW
+            )
+            and attr.enabled
+        ):
             attributes[name] = attr
 
-    object_defined_attributes = controller.get_attributes()
+    object_defined_attributes = {
+        name: attr for name, attr in controller.get_attributes().items() if attr.enabled
+    }
 
     if conflicting_keys := {
         key

tests/backends/epics/test_ioc.py

@@ -1,4 +1,4 @@
-from typing import Any
+from typing import Any, get_type_hints
 
 import pytest
 from pytest_mock import MockerFixture
@@ -401,6 +401,7 @@ def test_long_pv_names_discarded(mocker: MockerFixture):
     long_rw_name = "attr_rw_with_a_reallyreally_long_name_that_is_too_long_for_RBV"
     assert long_name_controller.attr_rw_short_name.enabled
     assert getattr(long_name_controller, long_attr_name).enabled
+
     EpicsIOC(DEVICE, long_name_mapping)
     assert long_name_controller.attr_rw_short_name.enabled
     assert not getattr(long_name_controller, long_attr_name).enabled

tests/backends/tango/test_dsr.py

@@ -14,7 +14,7 @@ def tango_context(self, assertable_controller):
             yield proxy
 
     def test_list_attributes(self, tango_context):
-        assert list(tango_context.get_attribute_list()) == [
+        assert set(tango_context.get_attribute_list()) == {
             "BigEnum",
             "ReadBool",
             "ReadInt",
@@ -26,7 +26,7 @@ def test_list_attributes(self, tango_context):
             "SubController02_ReadInt",
             "State",
             "Status",
-        ]
+        }
 
     def test_list_commands(self, tango_context):
         assert list(tango_context.get_command_list()) == [

tests/conftest.py

@@ -57,15 +57,6 @@ class TestSubController(SubController):
 
 
 class TestController(Controller):
-    def __init__(self) -> None:
-        super().__init__()
-
-        self._sub_controllers: list[TestSubController] = []
-        for index in range(1, 3):
-            controller = TestSubController()
-            self._sub_controllers.append(controller)
-            self.register_sub_controller(f"SubController{index:02d}", controller)
-
     read_int: AttrR = AttrR(Int(), handler=TestUpdater())
     read_write_int: AttrRW = AttrRW(Int(), handler=TestHandler())
     read_write_float: AttrRW = AttrRW(Float())
@@ -77,6 +68,15 @@ def __init__(self) -> None:
         allowed_values=[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17],
     )
 
+    def __init__(self) -> None:
+        super().__init__()
+
+        self._sub_controllers: list[TestSubController] = []
+        for index in range(1, 3):
+            controller = TestSubController()
+            self._sub_controllers.append(controller)
+            self.register_sub_controller(f"SubController{index:02d}", controller)
+
     initialised = False
     connected = False
     count = 0

tests/test_controller.py

@@ -1,6 +1,8 @@
 import pytest
 
+from fastcs.attributes import Attribute, AttrR
 from fastcs.controller import Controller, SubController
+from fastcs.datatypes import Int
 from fastcs.mapping import _get_single_mapping, _walk_mappings
 
 
@@ -29,3 +31,25 @@ def test_controller_nesting():
         ValueError, match=r"SubController is already registered under .*"
     ):
         controller.register_sub_controller("c", sub_controller)
+
+
+def test_attribute_parsing():
+    runtime_attribute = AttrR(Int())
+
+    class SomeController(Controller):
+        annotated_attr: AttrR[int]
+        annotated_attr_not_defined_in_init: AttrR[int]
+
+        def get_attributes(self) -> dict[str, Attribute]:
+            return {"get_attributes_attr": runtime_attribute}
+
+        def __init__(self):
+            self.annotated_attr = AttrR(Int())
+            super().__init__()
+
+    controller = SomeController()
+    mapping = next(_walk_mappings(controller))
+    assert mapping.attributes == {
+        "get_attributes_attr": runtime_attribute,
+        "annotated_attr": controller.annotated_attr,
+    }