From a1e87c4dae794b32ed8ec1e5aa4d524c12369f4d Mon Sep 17 00:00:00 2001 From: anthonyrocom <57441926+anthonyrocom@users.noreply.github.com> Date: Wed, 6 Nov 2019 12:49:01 +0100 Subject: [PATCH] Extended to_json() to support sort key Simple change for to_json() to support the sort key. Documentation needs to be updated for this change. --- treelib/tree.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/treelib/tree.py b/treelib/tree.py index 62d3549..cac0262 100644 --- a/treelib/tree.py +++ b/treelib/tree.py @@ -858,9 +858,9 @@ def to_dict(self, nid=None, key=None, sort=True, reverse=False, with_data=False) {ntag: {"data": self[nid].data}} return tree_dict - def to_json(self, with_data=False, sort=True, reverse=False): + def to_json(self, with_data=False, sort=True, reverse=False, key=None): """To format the tree in JSON format.""" - return json.dumps(self.to_dict(with_data=with_data, sort=sort, reverse=reverse)) + return json.dumps(self.to_dict(with_data=with_data, sort=sort, reverse=reverse, key=key)) def to_graphviz(self, filename=None, shape='circle', graph='digraph'): """Exports the tree in the dot format of the graphviz software"""