3

U'íc,ã@sºdZddlZddlmZddlmZddlmZejrBddl	m
Z
Gdd	„d	eƒZejej
ed
œdd„ZejejejejfZejejejejejfZejejejed
œd
d„ZdS)z_Functions that expose information about templates that might be
interesting for introspection.
éNé)Únodes)Ú
CodeGenerator)ÚFrame)ÚEnvironmentcsLeZdZdZdddœ‡fdd„Zeddœdd	„Zedd
œ‡fdd„Z‡Z	S)
ÚTrackingCodeGeneratorz.We abuse the code generator for introspection.rN)ÚenvironmentÚreturncstƒj|ddƒtƒ|_dS)Nz<introspection>)ÚsuperÚ__init__ÚsetÚundeclared_identifiers)Úselfr)Ú	__class__©ú-/tmp/pip-build-gk9425m9/Jinja2/jinja2/meta.pyrszTrackingCodeGenerator.__init__)Úxr	cCsdS)zDon't write.Nr)rrrrrÚwriteszTrackingCodeGenerator.write)Úframer	csNtƒj|ƒx<|jjjƒD],\}\}}|dkr||jjkr|jj|ƒqWdS)z$Remember all undeclared identifiers.ÚresolveN)	r
Úenter_frameÚsymbolsÚloadsÚitemsrÚglobalsr
Úadd)rrÚ_ÚactionÚparam)rrrrsz!TrackingCodeGenerator.enter_frame)
Ú__name__Ú
__module__Ú__qualname__Ú__doc__rÚstrrrrÚ
__classcell__rr)rrrsr)Úastr	cCst|jƒ}|j|ƒ|jS)aReturns 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.
    )rrÚvisitr
)r%ZcodegenrrrÚfind_undeclared_variables!s

r'ccsØxÒ|jtƒD]Ä}|j}t|tjƒsvt|tjtjfƒrnx>|jD],}t|tjƒrbt|j	t
ƒrh|j	Vq<dVq<WqdVqt|j	t
ƒrŒ|j	Vqt|tjƒrÊt|j	tt
fƒrÊx&|j	D]}t|t
ƒr°|Vq°WqdVqWdS)abFinds 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)Zfind_allÚ
_ref_typesÚtemplateÚ
isinstancerZConstÚTupleÚListrÚvaluer#ÚIncludeÚtupleÚlist)r%Únoder)Z
template_namerrrÚfind_referenced_templates=s&


r2)r"ÚtypingÚtÚrÚcompilerrrZ
TYPE_CHECKINGrrrÚTemplateÚSetr#r'ZExtendsZ
FromImportÚImportr.r(ÚUnionZ_RefTypeÚIteratorÚOptionalr2rrrrÚ<module>s