U a,@sdZddlZddlmZddlmZddlmZejrBddl m Z Gdd d eZ ej ej ed d d ZejejejejfZejejejejejfZej ejejed d dZdS)z_Functions that expose information about templates that might be interesting for introspection. N)nodes) CodeGenerator)Frame) EnvironmentcsLeZdZdZdddfdd Zedddd Zedd fd d ZZ S) TrackingCodeGeneratorz.We abuse the code generator for introspection.rN) environmentreturncst|ddt|_dS)Nz)super__init__setundeclared_identifiers)selfr __class__LC:\Users\vtejo\AppData\Local\Temp\pip-unpacked-wheel-8gbwd15f\jinja2\meta.pyr szTrackingCodeGenerator.__init__)xr cCsdS)z Don't write.Nr)rrrrrwriteszTrackingCodeGenerator.write)framer csJt||jjD],\}\}}|dkr||jjkr|j|qdS)z$Remember all undeclared identifiers.resolveN) r enter_framesymbolsloadsitemsrglobalsr add)rr_actionparamrrrrs z!TrackingCodeGenerator.enter_frame) __name__ __module__ __qualname____doc__r strrrr __classcell__rrrrrsr)astr cCst|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. )rrvisitr )r&Zcodegenrrrfind_undeclared_variables!s  r(ccs|tD]}|j}t|tjspt|tjtjfrh|jD],}t|tjr^t|j t rd|j Vq8dVq8q dVq t|j t r|j Vq t|tj rt|j t t fr|j D]}t|t r|Vqq dVq dS)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_typestemplate isinstancerZConstTupleListrvaluer$Includetuplelist)r&noder*Z template_namerrrfind_referenced_templates=s*           r3)r#typingtrcompilerrr TYPE_CHECKINGrrrTemplateSetr$r(ZExtendsZ FromImportZImportr/r)UnionZ_RefTypeIteratorOptionalr3rrrrs