Merge pull request #244 from c0fec0de/kylewhite21-devel

Kylewhite21 devel

anytree/node/symlinknodemixin.py

@@ -48,6 +48,8 @@ class SymlinkNodeMixin(NodeMixin):
     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):

tests/test_pickle.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+import pickle
+
+from anytree import AnyNode, LoopError, Node, NodeMixin, PostOrderIter, PreOrderIter, RenderTree, SymlinkNode, TreeError
+
+
+def test_pickle(tmp_path):
+    """Pickling Compatibilty."""
+    root = Node(name="root")
+    a = Node(name="a", parent=root)
+    b = Node(name="b", parent=a)
+    c = SymlinkNode(target=b, parent=a)
+
+    lines = str(RenderTree(root)).splitlines()
+    assert lines == [
+        "Node('/root')",
+        "└── Node('/root/a')",
+        "    ├── Node('/root/a/b')",
+        "    └── SymlinkNode(Node('/root/a/b'))",
+    ]
+
+    filepath = tmp_path / "test.pkl"
+    with open(filepath, "wb") as file:
+        pickle.dump(root, file)
+
+    with open(filepath, "rb") as file:
+        loaded = pickle.load(file)
+
+    assert str(RenderTree(root)).splitlines() == lines