cdZddlmZmZmZmZddlmZmZm Z dZ dede fdZ Gdd e ZGd d e ZGd d e Zy)z/A generic visitor abstract implementation. )AnyCallableOptionalUnion)Node HTMLWriter TextWriterzrestructuredtext en_returncy)N)r s 8/usr/lib/python3/dist-packages/logilab/common/visitor.py no_filterrs c BeZdZddededeeddfdZdeefdZeZ y) FilteredIteratorNnode list_func filter_funcr cD|dfg|_|t}||||_yNr)_nextr_list)selfrrrs r__init__zFilteredIterator.__init__$s(Qi[  #Kt[1 rcX |jjdS#t$rYywxYwr)rpop Exceptionrs r__next__zFilteredIterator.__next__*s, ::>>!$ $  s  ))N) __name__ __module__ __qualname__rrrrrr!nextrrrrr#s?2T2h2Xc]2^b2 (4. Drrc2eZdZddZdZdZdZdZdZy) VisitorNc ||_||_yr" _iter_classfilter)riterator_classrs rrzVisitor.__init__5s)! rcn|j|g|i||j|j|S)z launch the visit on a given node call 'open_visit' before the beginning of the visit, with extra args given when all nodes have been visited, call the 'close_visit' method ) open_visit close_visit_visit)rrargskargss rvisitz Visitor.visit9s6 -t-u- D 122rc|j|}t|}|r|j|}t|}|rSr") _get_iteratorr&accept)rriteratornresults rr1zVisitor._visitDs@%%d+ NXXd^FXA rc:|j||jSr"r*)rrs rr6zVisitor._get_iteratorLsdkk22rcy)z= method called at the beginning of the visit Nr)rr2r3s rr/zVisitor.open_visitOsrc|S)z7 method called at the end of the visit r)rr:s rr0zVisitor.close_visitTs  rr") r#r$r%rr4r1r6r/r0rrrr(r(4s " 33 rr(c JeZdZdZdefdZdeeefde de de e fdZ dZ y ) VisitedMixInz? Visited interface allow node visitors to use the node r c |jjddS#t$r'|jjj cYSwxYw)z return the visit name for the mixed class. When calling 'accept', the method <'visit_' + name returned by this method> will be called on the visitor -r )TYPEreplacer __class__r#lowerr s rget_visit_namezVisitedMixIn.get_visit_nameasH  399$$S#. . 3>>**002 2 3s-A Avisitorr2kwargscTt|d|jz}||g|i|S)Nzvisit_%sgetattrrFrrGr2rHfuncs rr7zVisitedMixIn.acceptns3w T-@-@-B BCD*4*6**rcTt|d|jz}||g|i|S)Nzleave_%srJrLs rleavezVisitedMixIn.leavets1w T-@-@-B BCD*4*6**rN) r#r$r%__doc__strrFrrr rrr7rOrrrr?r?\sK 3 3+Z34+=@+LO+ #+ +rr?N)rPtypingrrrrlogilab.common.typesrrr __docformat__introbjectrr(r?rrrrWsU$ 21==%   # v "$f$P+6+r