Method __setattr__

binarytree/__init__.py:177–230

Modified version of ``__setattr__`` with extra sanity checking. Class attributes **left**, **right** and **value** are validated. :param attr: Name of the class attribute. :type attr: str :param obj: Object to set. :type obj: object :raise binarytree

(self, attr: str, obj: Any)

        return "\n" + "\n".join((line.rstrip() for line in lines))

    def __setattr__(self, attr: str, obj: Any) -> None:
        """Modified version of ``__setattr__`` with extra sanity checking.

        Class attributes **left**, **right** and **value** are validated.

        :param attr: Name of the class attribute.
        :type attr: str
        :param obj: Object to set.
        :type obj: object
        :raise binarytree.exceptions.NodeTypeError: If left or right child is
            not an instance of :class:`binarytree.Node`.
        :raise binarytree.exceptions.NodeValueError: If node value is invalid.

        **Example**:

        .. doctest::

            >>> from binarytree import Node
            >>>
            >>> node = Node(1)
            >>> node.left = []  # doctest: +IGNORE_EXCEPTION_DETAIL
            Traceback (most recent call last):
             ...
            binarytree.exceptions.NodeTypeError: Left child must be a Node instance

        .. doctest::

            >>> from binarytree import Node
            >>>
            >>> node = Node(1)
            >>> node.val = [] # doctest: +IGNORE_EXCEPTION_DETAIL
            Traceback (most recent call last):
             ...
            binarytree.exceptions.NodeValueError: node value must be a float/int/str
        """
        if attr == _ATTR_LEFT:
            if obj is not None and not isinstance(obj, Node):
                raise NodeTypeError("left child must be a Node instance")

        elif attr == _ATTR_RIGHT:
            if obj is not None and not isinstance(obj, Node):
                raise NodeTypeError("right child must be a Node instance")

        elif attr == _ATTR_VALUE:
            if not isinstance(obj, _NODE_VAL_TYPES):
                raise NodeValueError("node value must be a float/int/str")
            object.__setattr__(self, _ATTR_VAL, obj)

        elif attr == _ATTR_VAL:
            if not isinstance(obj, _NODE_VAL_TYPES):
                raise NodeValueError("node value must be a float/int/str")
            object.__setattr__(self, _ATTR_VALUE, obj)

        object.__setattr__(self, attr, obj)

    def __iter__(self) -> Iterator["Node"]:
        """Iterate through the nodes in the binary tree in level-order_.