Add option to exclude attributes from __pyaml_repr__.

Author: TeresiaOlsson

pyaml/common/element.py

@@ -8,11 +8,19 @@
     from ..common.element_holder import ElementHolder
 
 
-def __pyaml_repr__(obj):
+def __pyaml_repr__(obj, exclude: list[str] | None = None):
     """
-    Returns a string representation of a pyaml object
+    Returns a string representation of a pyaml object.
+
+    Parameters
+    ----------
+    exclude : list[str] | None
+        Attribute/property names to exclude from the output.
     """
 
+    if exclude is None:
+        exclude = []
+
     cls_name = obj.__class__.__name__
 
     # Keep the old behavior when _cfg exists
@@ -31,19 +39,19 @@ def __pyaml_repr__(obj):
 
     # Instance attributes
     for k, v in obj.__dict__.items():
-        # Exclude private attributes
-        if not k.startswith("_"):
+        # Exclude private attributes and excluded
+        if not k.startswith("_") and k not in exclude:
             attrs[k] = v
 
     # Properties
     for name, attr in vars(type(obj)).items():
-        if isinstance(attr, property):
+        if isinstance(attr, property) and name not in exclude:
             try:
                 attrs[name] = getattr(obj, name)
             except Exception as e:
                 attrs[name] = f"<error: {e}>"
 
-    if isinstance(obj, Element) and "name" not in attrs:
+    if isinstance(obj, Element) and "name" not in attrs and "name" not in exclude:
         try:
             attrs["name"] = obj.get_name()
         except Exception as e:

pyaml/rf/rf_plant.py

@@ -3,7 +3,7 @@
 
 from .. import PyAMLException
 from ..common import abstract
-from ..common.element import Element
+from ..common.element import Element, __pyaml_repr__
 from ..validation import DynamicValidation, register_schema
 from .rf_transmitter import RFTransmitter
 
@@ -90,3 +90,6 @@ def set_and_wait(self, value: float):
 
     def unit(self) -> str:
         return self.__trans[0].phase_device_access.unit()
+
+    def __repr__(self):
+        return __pyaml_repr__(self, exclude=["frequency", "voltage"])

pyaml/rf/rf_transmitter.py

@@ -3,7 +3,7 @@
 
 from .. import PyAMLException
 from ..common import abstract
-from ..common.element import Element
+from ..common.element import Element, __pyaml_repr__
 from ..validation import DynamicValidation, register_schema
 
 # Define the main class name for this module
@@ -104,3 +104,6 @@ def attach(
         obj.__phase = phase
         obj._peer = peer
         return obj
+
+    def __repr__(self):
+        return __pyaml_repr__(self, exclude=["voltage", "phase"])