�
��+b, � �� � d Z ddlZddlmZ ddlmZ ddlmZ ej rddl m
Z
G d� d e� Z d
ej d ej e fd �Zej ej" ej$ ej& fZej* ej ej" ej$ ej& f Zd
ej d ej. ej0 e fd
�Zy)z_Functions that expose information about templates that might be
interesting for introspection.
� N� )�nodes)�
CodeGenerator)�Frame)� Environmentc �H � � e Zd ZdZd � fd� Zdeddfd�Zdeddf� fd� Z� xZ S )
�TrackingCodeGeneratorz.We abuse the code generator for introspection.�returnNc �F �� t �| � |dd� t � | _ y )Nz<introspection>)�super�__init__�set�undeclared_identifiers)�self� environment� __class__s ��-/usr/lib/python3/dist-packages/jinja2/meta.pyr
z TrackingCodeGenerator.__init__ s �� �
����&7�9J�K�25�%��#� �xc � � y)z Don't write.N� )r r s r �writezTrackingCodeGenerator.write s � r �framec � �� t �| � |� |j j j � D ]B \ }\ }}|dk( s�|| j
j vs�(| j j |� �D y)z$Remember all undeclared identifiers.�resolveN) r � enter_frame�symbols�loads�itemsr �globalsr �add)r r �_�action�paramr s �r r z!TrackingCodeGenerator.enter_frame sj �� �
���E�"�"'�-�-�"5�"5�";�";�"=� 7� �A������"�u�D�4D�4D�4L�4L�'L��+�+�/�/��6� 7r )r r r
N)
�__name__�
__module__� __qualname__�__doc__r
�strr r r �
__classcell__)r s @r r r s6 �� �8�8��s� �t� �7�� 7�4� 7� 7r r �astr
c �f � t | j � }|j | � |j S )a Returns a set of all variables in the AST that will be looked up from
the context at runtime. Because at compile time it's not known which
variables will be used depending on the path the execution takes at
runtime, all variables are returned.
>>> from jinja2 import Environment, meta
>>> env = Environment()
>>> ast = env.parse('{% set foo = 42 %}{{ bar + foo }}')
>>> meta.find_undeclared_variables(ast) == {'bar'}
True
.. admonition:: Implementation
Internally the code generator is used for finding undeclared variables.
This is good to know because the code generator might raise a
:exc:`TemplateAssertionError` during compilation and as a matter of
fact this function can currently raise that exception as well.
)r r �visitr )r+ �codegens r �find_undeclared_variablesr/ ! s* � �& $�C�O�O�4�G� �M�M�#�� � )� )�)r c # � K � | j t � D �]A }|j }t |t j
� s�t |t j t j f� rZ|j D ]J }t |t j
� r*t |j t � s�8|j �� �Gd�� �L nd�� ��t |j t � r|j �� ��t |t j � rHt |j t t f� r(|j D ] }t |t � s�|�� � ��>d�� ��D y�w)ab Finds all the referenced templates from the AST. This will return an
iterator over all the hardcoded template extensions, inclusions and
imports. If dynamic inheritance or inclusion is used, `None` will be
yielded.
>>> from jinja2 import Environment, meta
>>> env = Environment()
>>> ast = env.parse('{% extends "layout.html" %}{% include helper %}')
>>> list(meta.find_referenced_templates(ast))
['layout.html', None]
This function is useful for dependency tracking. For example if you want
to rebuild parts of the website after a layout template has changed.
N)�find_all�
_ref_types�template�
isinstancer �Const�Tuple�Listr �valuer) �Include�tuple�list)r+ �noder3 �
template_names r �find_referenced_templatesr> = s � �� �" � � �Z�(� !�� #�}�}���(�E�K�K�0��(�U�[�[�%�*�*�$=�>�%-�^�^� #�M� "�-����=�%�m�&9�&9�3�?�"/�"5�"5� 5� #�
�#� �
� � �h�n�n�c� *��.�.� � ��e�m�m�
,�� �N�N�U�D�M�2
� "*� � �
(�
� �m�S�1�'�'�
(�
�J�C!�s �B-E �0B E �E )r( �typing�t� r �compilerr r �
TYPE_CHECKINGr r r �Template�Setr) r/ �Extends�
FromImport�Importr9 r2 �Union�_RefType�Iterator�Optionalr> r r r �<module>rM s� ��� � � #� ��?�?�(�7�M� 7�&*�5�>�>� *�a�e�e�C�j� *�0 �m�m�U� -� -�u�|�|�U�]�]�
K�
� �7�7�5�=�=�%�"2�"2�E�L�L�%�-�-�O� P��2�5�>�>� 2�a�j�j����C� �6Q� 2r |