Added custom LofarDeviceProxy to enhance support for high-dimensional arrays in attribtues

lofar_station_client/devices.py

+""" Enhanced interfaces towards the station's Tango devices. """
+
 import ast
 from functools import lru_cache
 
@@ -11,13 +13,15 @@ class LofarDeviceProxy(DeviceProxy):
     """A LOFAR-specific tango.DeviceProxy that provides
     a richer experience."""
 
-    # cache attribute configurations, as they are not expected to change,
-    # but carry a cost to retrieve from the server.
     @lru_cache()
     def get_attribute_config(self, name):
+        """Get cached attribute configurations, as they are not expected to change."""
+
         return super().get_attribute_config(name)
 
     def read_attribute(self, name, extract_as=ExtractAs.Numpy):
+        """Read an attribute from the server."""
+
         attr = super().read_attribute(name, extract_as)
 
         # "format" property describes actual dimensions as a tuple (x, y, z, ...),
@@ -30,11 +34,14 @@ class LofarDeviceProxy(DeviceProxy):
         return attr
 
     def write_attribute(self, name, value):
+        """Write an attribute to the server."""
+
         config = self.get_attribute_config(name)
 
         # 2D arrays also represent arrays of higher dimensionality. reshape them
         # to fit their original Tango shape before writing.
         if config.max_dim_y > 0:
-            attr.value = attr.value.reshape((config.max_dim_y, config.max_dim_x))
+            value = numpy.array(value)
+            value = value.reshape((config.max_dim_y, config.max_dim_x))
 
         return super().write_attribute(name, value)