1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/moinformat/tree/pretty.py Fri Jun 30 00:40:39 2023 +0200
1.3 @@ -0,0 +1,65 @@
1.4 +#!/usr/bin/env python
1.5 +
1.6 +"""
1.7 +Prettyprinted document tree nodes.
1.8 +
1.9 +Copyright (C) 2017, 2018, 2019, 2023 Paul Boddie <paul@boddie.org.uk>
1.10 +
1.11 +This program is free software; you can redistribute it and/or modify it under
1.12 +the terms of the GNU General Public License as published by the Free Software
1.13 +Foundation; either version 3 of the License, or (at your option) any later
1.14 +version.
1.15 +
1.16 +This program is distributed in the hope that it will be useful, but WITHOUT
1.17 +ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
1.18 +FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
1.19 +details.
1.20 +
1.21 +You should have received a copy of the GNU General Public License along with
1.22 +this program. If not, see <http://www.gnu.org/licenses/>.
1.23 +"""
1.24 +
1.25 +from moinformat.tree.moin import Container
1.26 +
1.27 +class Node:
1.28 +
1.29 + "A simplified tree node representation."
1.30 +
1.31 + def __init__(self, name):
1.32 + self.name = name
1.33 + self.nodes = []
1.34 +
1.35 + def __repr__(self):
1.36 + return "Node(%r, %r)" % (self.name, self.nodes)
1.37 +
1.38 + def visit(self, visitor):
1.39 + visitor.node(self)
1.40 +
1.41 + def append(self, node):
1.42 + self.nodes.append(node)
1.43 +
1.44 + def test(self, other):
1.45 +
1.46 + """
1.47 + Test whether this node is considered equivalent to 'other', where
1.48 + 'other' is a moinparser.tree node.
1.49 +
1.50 + Return any failing tree nodes or None.
1.51 + """
1.52 +
1.53 + if other.__class__.__name__ != self.name:
1.54 + return self, other, "name"
1.55 +
1.56 + if isinstance(other, Container):
1.57 + for node, other_node in map(None, self.nodes, other.nodes):
1.58 + if node is None or other_node is None:
1.59 + return self, other, node is None and "simple" or "document"
1.60 + t = node.test(other_node)
1.61 + if t:
1.62 + return t
1.63 + elif self.nodes:
1.64 + return self, other, "empty"
1.65 +
1.66 + return None
1.67 +
1.68 +# vim: tabstop=4 expandtab shiftwidth=4