Source code for anytree.node.symlinknodemixin

# -*- coding: utf-8 -*-
from .nodemixin import NodeMixin


[docs]class SymlinkNodeMixin(NodeMixin): """ The :any:`SymlinkNodeMixin` class extends any Python class to a symbolic link to a tree node. The class **MUST** have a `target` attribute refering to another tree node. The :any:`SymlinkNodeMixin` class has its own parent and its own child nodes. All other attribute accesses are just forwarded to the target node. A minimal implementation looks like (see :any:`SymlinkNode` for a full implemenation): >>> from anytree import SymlinkNodeMixin, Node, RenderTree >>> class SymlinkNode(SymlinkNodeMixin): ... def __init__(self, target, parent=None, children=None): ... self.target = target ... self.parent = parent ... if children: ... self.children = children ... def __repr__(self): ... return "SymlinkNode(%r)" % (self.target) >>> root = Node("root") >>> s1 = Node("sub1", parent=root) >>> l = SymlinkNode(s1, parent=root) >>> l0 = Node("l0", parent=l) >>> print(RenderTree(root)) Node('/root') ├── Node('/root/sub1') └── SymlinkNode(Node('/root/sub1')) └── Node('/root/sub1/l0') Any modifications on the target node are also available on the linked node and vice-versa: >>> s1.foo = 4 >>> s1.foo 4 >>> l.foo 4 >>> l.foo = 9 >>> s1.foo 9 >>> l.foo 9 """ def __getattr__(self, name): if name in ("_NodeMixin__parent", "_NodeMixin__children"): return super(SymlinkNodeMixin, self).__getattr__(name) if name == "__setstate__": raise AttributeError(name) return getattr(self.target, name) def __setattr__(self, name, value): if name in ("_NodeMixin__parent", "_NodeMixin__children", "parent", "children", "target"): super(SymlinkNodeMixin, self).__setattr__(name, value) else: setattr(self.target, name, value)