44from lxml import etree
55from ..treebuilders .etree import tag_regexp
66
7- from . import _base
7+ from . import base
88
99from .. import ihatexml
1010
@@ -122,7 +122,7 @@ def __len__(self):
122122 return len (self .obj )
123123
124124
125- class TreeWalker (_base .NonRecursiveTreeWalker ):
125+ class TreeWalker (base .NonRecursiveTreeWalker ):
126126 def __init__ (self , tree ):
127127 # pylint:disable=redefined-variable-type
128128 if isinstance (tree , list ):
@@ -131,29 +131,29 @@ def __init__(self, tree):
131131 else :
132132 self .fragmentChildren = set ()
133133 tree = Root (tree )
134- _base .NonRecursiveTreeWalker .__init__ (self , tree )
134+ base .NonRecursiveTreeWalker .__init__ (self , tree )
135135 self .filter = ihatexml .InfosetFilter ()
136136
137137 def getNodeDetails (self , node ):
138138 if isinstance (node , tuple ): # Text node
139139 node , key = node
140140 assert key in ("text" , "tail" ), "Text nodes are text or tail, found %s" % key
141- return _base .TEXT , ensure_str (getattr (node , key ))
141+ return base .TEXT , ensure_str (getattr (node , key ))
142142
143143 elif isinstance (node , Root ):
144- return (_base .DOCUMENT ,)
144+ return (base .DOCUMENT ,)
145145
146146 elif isinstance (node , Doctype ):
147- return _base .DOCTYPE , node .name , node .public_id , node .system_id
147+ return base .DOCTYPE , node .name , node .public_id , node .system_id
148148
149149 elif isinstance (node , FragmentWrapper ) and not hasattr (node , "tag" ):
150- return _base .TEXT , ensure_str (node .obj )
150+ return base .TEXT , ensure_str (node .obj )
151151
152152 elif node .tag == etree .Comment :
153- return _base .COMMENT , ensure_str (node .text )
153+ return base .COMMENT , ensure_str (node .text )
154154
155155 elif node .tag == etree .Entity :
156- return _base .ENTITY , ensure_str (node .text )[1 :- 1 ] # strip &;
156+ return base .ENTITY , ensure_str (node .text )[1 :- 1 ] # strip &;
157157
158158 else :
159159 # This is assumed to be an ordinary element
@@ -172,7 +172,7 @@ def getNodeDetails(self, node):
172172 attrs [(match .group (1 ), match .group (2 ))] = value
173173 else :
174174 attrs [(None , name )] = value
175- return (_base .ELEMENT , namespace , self .filter .fromXmlName (tag ),
175+ return (base .ELEMENT , namespace , self .filter .fromXmlName (tag ),
176176 attrs , len (node ) > 0 or node .text )
177177
178178 def getFirstChild (self , node ):
0 commit comments