| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758 |
- # -*- coding: utf-8 -*-
- from .nodemixin import NodeMixin
- class SymlinkNodeMixin(NodeMixin):
- u"""
- 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)
- else:
- 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:
- return setattr(self.target, name, value)
|
