3
aG�\� � @ s4 d Z ddlmZ G dd� de�ZG dd� de�ZdS )z�
jinja2.visitor
~~~~~~~~~~~~~~
This module implements a visitor for the nodes.
:copyright: (c) 2017 by the Jinja Team.
:license: BSD.
� )�Nodec @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � NodeVisitora Walks the abstract syntax tree and call visitor functions for every
node found. The visitor functions may return values which will be
forwarded by the `visit` method.
Per default the visitor functions for the nodes are ``'visit_'`` +
class name of the node. So a `TryFinally` node visit function would
be `visit_TryFinally`. This behavior can be changed by overriding
the `get_visitor` function. If no visitor function exists for a node
(return value `None`) the `generic_visit` visitor is used instead.
c C s d|j j }t| |d�S )z�Return the visitor function for this node or `None` if no visitor
exists for this node. In that case the generic visit function is
used instead.
Zvisit_N)� __class__�__name__�getattr)�self�node�method� r
� /usr/lib/python3.6/visitor.py� get_visitor s zNodeVisitor.get_visitorc O s4 | j |�}|dk r"||f|�|�S | j|f|�|�S )z
Visit a node.N)r �
generic_visit)r r �args�kwargs�fr
r
r �visit" s
zNodeVisitor.visitc O s( x"|j � D ]}| j|f|�|� q
W dS )z9Called if no explicit visitor function exists for a node.N)Ziter_child_nodesr )r r r r r
r
r r
) s zNodeVisitor.generic_visitN)r �
__module__� __qualname__�__doc__r r r
r
r
r
r r s
r c @ s e Zd ZdZdd� Zdd� ZdS )�NodeTransformera� Walks the abstract syntax tree and allows modifications of nodes.
The `NodeTransformer` will walk the AST and use the return value of the
visitor functions to replace or remove the old node. If the return
value of the visitor function is `None` the node will be removed
from the previous location otherwise it's replaced with the return
value. The return value may be the original node in which case no
replacement takes place.
c O s� x�|j � D ]�\}}t|t�r�g }xT|D ]L}t|t�rh| j|f|�|�}|d krRq&nt|t�sh|j|� q&|j|� q&W ||d d �< q
t|t�r
| j|f|�|�}|d kr�t||� q
t|||� q
W |S )N) Z iter_fields�
isinstance�listr r �extend�append�delattr�setattr) r r r r ZfieldZ old_valueZ
new_values�valueZnew_noder
r
r r
: s&
z NodeTransformer.generic_visitc O s& | j |f|�|�}t|t�s"|g}|S )z{As transformers may return lists in some places this method
can be used to enforce a list as return value.
)r r r )r r r r �rvr
r
r �
visit_listP s
zNodeTransformer.visit_listN)r r r r r
r r
r
r
r r / s r N)r Z jinja2.nodesr �objectr r r
r
r
r �<module>
s !