U
    “üjbì ã                
   @   sr  d Z ddlZddlmZ ddlmZ ddlmZ ddl	m
Z
 ddlmZ ddlmZ dd	lmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlmZ d
dlm Z  d
dl!m"Z" ej#rddl$Z%d
dl&m'Z' ej(dej)dej*f dZ+dddddd d!d"d#œZ,e+e+d$œd%d&„Z-e.ej)d'ej/d(gdf d)œd*d+„Z0e.ej)d'ej1d(gdf d)œd,d-„Z2dFej3d0ej4e. ej4e. ej4ej5 e6e6ej4e. d1œd2d3„Z7ej*e6d4œd5d6„Z8ej9ej: ej9e. ej;e. d7œd8d9„Z<G d:d;„ d;ƒZ=G d<d(„ d(ƒZ>G d=d>„ d>e?ƒZ@G d?d@„ d@e"ƒZAG dAdB„ dBe"ƒZBG dCdD„ dDeCƒZDG dEd'„ d'e"ƒZEdS )Gz0Compiles nodes from the parser into Python code.é    N)Úcontextmanager)Úupdate_wrapper)ÚStringIO)Úchain)Ú	iskeyword)Úescape)ÚMarkupé   )Únodes)ÚTemplateAssertionError)ÚSymbols)ÚVAR_LOAD_ALIAS)ÚVAR_LOAD_PARAMETER)ÚVAR_LOAD_RESOLVE)ÚVAR_LOAD_UNDEFINED)ÚEvalContext)Ú	Optimizer)Ú_PassArg)Úconcat)ÚNodeVisitor)ÚEnvironmentÚF.)Úboundz==z!=ú>z>=ú<z<=Úinznot in)ÚeqÚneÚgtZgteqÚltZlteqr   Únotin)ÚfÚreturnc                    s2   dt jdtjtjdœ‡ fdd„}tt t|¡ˆ ƒS )NÚCodeGeneratorÚFrame)ÚselfÚnodeÚframeÚkwargsr"   c                    sF   | j d k	r6|jjs6| j  ||j¡}||kr6|  ||¡S ˆ | ||f|ŽS ©N)Ú	optimizerÚeval_ctxÚvolatileÚvisit)r%   r&   r'   r(   Znew_node©r!   © ú3/tmp/pip-unpacked-wheel-9y7w5xh9/jinja2/compiler.pyÚnew_func,   s
    zoptimizeconst.<locals>.new_func)r
   ÚExprÚtÚAnyr   Úcastr   )r!   r1   r/   r.   r0   Úoptimizeconst+   s       þr6   r#   r$   )Úopr"   c                    s"   t dtjtd dœ‡ fdd„ƒ}|S )Nr#   ©r%   r&   r'   r"   c                    s”   | j jrNˆ | j jkrN|  dˆ ›d¡ |  |j|¡ |  d¡ |  |j|¡ n8|  d¡ |  |j|¡ |  dˆ › d¡ |  |j|¡ |  d¡ d S )Nz environment.call_binop(context, ú, ú(ú ú))ÚenvironmentÚ	sandboxedZintercepted_binopsÚwriter-   ÚleftÚright©r%   r&   r'   ©r7   r/   r0   Úvisitor<   s    ÿ
þ

z_make_binop.<locals>.visitor)r6   r
   ÚBinExprr$   ©r7   rD   r/   rC   r0   Ú_make_binop;   s    rG   c                    s"   t dtjtd dœ‡ fdd„ƒ}|S )Nr#   r8   c                    s`   | j jr6ˆ | j jkr6|  dˆ ›d¡ |  |j|¡ n|  dˆ  ¡ |  |j|¡ |  d¡ d S )Nzenvironment.call_unop(context, r9   r:   r<   )r=   r>   Zintercepted_unopsr?   r-   r&   rB   rC   r/   r0   rD   T   s    ÿ
þz_make_unop.<locals>.visitor)r6   r
   Ú	UnaryExprr$   rF   r/   rC   r0   Ú
_make_unopQ   s    rI   FTr   )r&   r=   ÚnameÚfilenameÚstreamÚ
defer_initÚ	optimizedr"   c                 C   sH   t | tjƒstdƒ‚| ||||||¡}| | ¡ |dkrD|j ¡ S dS )z+Generate the python source for a node tree.z Can't compile non template nodesN)Ú
isinstancer
   ÚTemplateÚ	TypeErrorZcode_generator_classr-   rL   Úgetvalue)r&   r=   rJ   rK   rL   rM   rN   Ú	generatorr/   r/   r0   Úgeneratee   s    
     ÿ

rT   ©Úvaluer"   c                 C   s†   | dks| t ks| tkrdS t| ƒttttttt	hkr:dS t| ƒt
ttthkr`tdd„ | D ƒƒS t| ƒtkr‚tdd„ |  ¡ D ƒƒS dS )z)Does the node have a safe representation?NTc                 s   s   | ]}t |ƒV  qd S r)   ©Úhas_safe_repr)Ú.0Úvr/   r/   r0   Ú	<genexpr>†   s     z has_safe_repr.<locals>.<genexpr>c                 s   s"   | ]\}}t |ƒot |ƒV  qd S r)   rW   ©rY   ÚkrZ   r/   r/   r0   r[   ‰   s     F)ÚNotImplementedÚEllipsisÚtypeÚboolÚintÚfloatÚcomplexÚrangeÚstrr   ÚtupleÚlistÚsetÚ	frozensetÚallÚdictÚitems©rV   r/   r/   r0   rX   }   s    rX   )r
   Únamesr"   c                 C   s<   t |ƒ}z| D ]}| |¡ qW n tk
r4   Y nX |jS )z”Check if the names passed are accessed undeclared.  The return value
    is a set of all the undeclared names from the sequence of names found.
    )ÚUndeclaredNameVisitorr-   ÚVisitorExitÚ
undeclared)r
   ro   rD   r&   r/   r/   r0   Úfind_undeclaredŽ   s    rs   c                   @   s*   e Zd Zejejejf ddœdd„ZdS )ÚMacroRefN©r&   r"   c                 C   s   || _ d| _d| _d| _d S ©NF)r&   Úaccesses_callerÚaccesses_kwargsÚaccesses_varargs©r%   r&   r/   r/   r0   Ú__init__ž   s    zMacroRef.__init__)	Ú__name__Ú
__module__Ú__qualname__r3   ÚUnionr
   ÚMacroÚ	CallBlockr{   r/   r/   r/   r0   rt      s   rt   c                   @   sd   e Zd ZdZdeejd  eje ddœdd„Zd dœdd„Z	de
d d
œdd„Zd dœdd„Ze	ZdS )r$   z&Holds compile time information for us.N)r+   ÚparentÚlevelr"   c                 C   s~   || _ || _|d kr4t|d| _d| _d | _d | _n(t|j|d| _|j| _|j| _|j| _d| _d| _d| _	d| _
d| _d S )N©rƒ   F)r+   r‚   r   ÚsymbolsÚrequire_output_checkÚbufferÚblockÚtoplevelÚ	rootlevelÚ
loop_frameÚblock_frameÚ
soft_frame)r%   r+   r‚   rƒ   r/   r/   r0   r{   ¨   s     zFrame.__init__©r"   c                 C   s*   t  | j¡}|j | j¡ | j ¡ |_|S )z!Create a copy of the current one.)ÚobjectÚ__new__Ú	__class__Ú__dict__Úupdater…   Úcopy©r%   Úrvr/   r/   r0   r”   Û   s    z
Frame.copyF)Úisolatedr"   c                 C   s&   |rt | j| jjd dS t | j| ƒS )zReturn an inner frame.r	   r„   )r$   r+   r…   rƒ   )r%   r—   r/   r/   r0   Úinnerâ   s    zFrame.innerc                 C   s   |   ¡ }d|_d|_|S )a&  Return a soft frame.  A soft frame may not be modified as
        standalone thing as it shares the resources with the frame it
        was created of, but it's not a rootlevel frame any longer.

        This is only used to implement if-statements and conditional
        expressions.
        FT)r”   rŠ   r   r•   r/   r/   r0   Úsoftè   s    z
Frame.soft)NN)F)r|   r}   r~   Ú__doc__r   r3   ÚOptionalrb   r{   r”   ra   r˜   r™   Ú__copy__r/   r/   r/   r0   r$   ¥   s     üû3c                   @   s   e Zd ZdZdS )rq   z?Exception used by the `UndeclaredNameVisitor` to signal a stop.N©r|   r}   r~   rš   r/   r/   r/   r0   rq   ø   s   rq   c                   @   sT   e Zd ZdZddœdd„Zejddœdd„Zejddœd	d
„Z	ej
ddœdd„ZdS )ÚDependencyFinderVisitorz.A visitor that collects filter and test calls.NrŽ   c                 C   s   t ƒ | _t ƒ | _d S r)   )ri   ÚfiltersÚtests©r%   r/   r/   r0   r{   ÿ   s    z DependencyFinderVisitor.__init__ru   c                 C   s   |   |¡ | j |j¡ d S r)   )Úgeneric_visitrŸ   ÚaddrJ   rz   r/   r/   r0   Úvisit_Filter  s    
z$DependencyFinderVisitor.visit_Filterc                 C   s   |   |¡ | j |j¡ d S r)   )r¢   r    r£   rJ   rz   r/   r/   r0   Ú
visit_Test  s    
z"DependencyFinderVisitor.visit_Testc                 C   s   dS )zStop visiting at blocks.Nr/   rz   r/   r/   r0   Úvisit_Block  s    z#DependencyFinderVisitor.visit_Block)r|   r}   r~   rš   r{   r
   ÚFilterr¤   ÚTestr¥   ÚBlockr¦   r/   r/   r/   r0   rž   ü   s
   rž   c                   @   sJ   e Zd ZdZeje ddœdd„Zej	ddœdd„Z
ejddœd	d
„ZdS )rp   z¡A visitor that checks if a name is accessed without being
    declared.  This is different from the frame visitor as it will
    not stop at closure frames.
    N)ro   r"   c                 C   s   t |ƒ| _t ƒ | _d S r)   )ri   ro   rr   )r%   ro   r/   r/   r0   r{     s    
zUndeclaredNameVisitor.__init__ru   c                 C   sJ   |j dkr8|j| jkr8| j |j¡ | j| jkrFtƒ ‚n| j |j¡ d S )NÚload)ÚctxrJ   ro   rr   r£   rq   Údiscardrz   r/   r/   r0   Ú
visit_Name  s
    z UndeclaredNameVisitor.visit_Namec                 C   s   dS )zStop visiting a blocks.Nr/   rz   r/   r/   r0   r¦   !  s    z!UndeclaredNameVisitor.visit_Block)r|   r}   r~   rš   r3   ÚIterablerf   r{   r
   ÚNamer­   r©   r¦   r/   r/   r/   r0   rp     s   rp   c                   @   s   e Zd ZdZdS )ÚCompilerExitz¼Raised if the compiler encountered a situation where it just
    doesn't make sense to further process the code.  Any block that
    raises such an exception is not further processed.
    Nr   r/   r/   r/   r0   r°   %  s   r°   c                	   @   sJ  e Zd ZU dëdeje eje ejej eeddœdd„Ze	edœd	d
„ƒZ
eeddœdd„Zedœdd„Zeddœdd„Zdìeeddœdd„Zddœdd„Zdíeddœdd„Zdîeejej ddœdd„Zeddœd d!„Zdïeeejej dd"œd#d$„Zejej edd%œd&d'„Zedd(œd)d*„Zdðeejej edd,œd-d.„Zdñejej edd/œd0d1„Zdòejejejej f eejej!eej"f  dd2œd3d4„Z#ejej dd5œd6d7„Z$eddœd8d9„Z%dóeedd:œd;d<„Z&dôeeed?œd@dA„Z'eedBœdCdD„Z(ejej)ej*f eej+ee,f dEœdFdG„Z-e,eddHœdIdJ„Z.ejedKœdLdM„Z/eedœdNdO„Z0ddœdPdQ„Z1eddœdRdS„Z2ddœdTdU„Z3eddVœdWdX„Z4eddVœdYdZ„Z5ddœd[d\„Z6edœd]d^„Z7edœd_d`„Z8eedœdadb„Z9eedVœdcdd„Z:ddœdedf„Z;eddœdgdh„Z<dõej=eje ddEœdidj„Z>ej?eddEœdkdl„Z@ejAeddEœdmdn„ZBejCeddEœdodp„ZDejejEejFf eddEœdqdr„ZGejEeddEœdsdt„ZHejFeddEœdudv„ZIejJeddEœdwdx„ZKejLeddEœdydz„ZMej)eddEœd{d|„ZNej*eddEœd}d~„ZOejPeddEœdd€„ZQejReddEœdd‚„ZSejTeddEœdƒd„„ZUG d…d†„ d†ejVƒZWeXej"ej"d‡œdˆd‰„ƒZYdZZejeW e[dŠ< eWdœd‹dŒ„Z\ejej" edœdŽd„Z]ej^eeWedœd‘d’„Z_ej^eeWddœd“d”„Z`ej^eeWddœd•d–„ZaejbeddEœd—d˜„ZcejdeddEœd™dš„ZeejfeddEœd›dœ„ZgejheddEœddž„ZiejjeddEœdŸd „ZkejleddEœd¡d¢„ZmejneddEœd£d¤„Zoej+eddEœd¥d¦„ZpejqeddEœd§d¨„ZrejseddEœd©dª„Zteud«ƒZveud¬ƒZweud­ƒZxeud®ƒZyeud¯ƒZzeud°ƒZ{eud±ƒZ|eud²ƒZ}eud³ƒZ~ed«ƒZ€ed¬ƒZed´ƒZ‚eƒej„eddEœdµd¶„ƒZ…eƒej†eddEœd·d¸„ƒZ‡ejˆeddEœd¹dº„Z‰eƒejŠeddEœd»d¼„ƒZ‹eƒejŒeddEœd½d¾„ƒZejŽeddEœd¿dÀ„Zeejejej f eeej‘d dÁœdÂdÃ„ƒZ’eƒejeddEœdÄdÅ„ƒZ“eƒej eddEœdÆdÇ„ƒZ”eƒej•eddEœdÈdÉ„ƒZ–eƒdöejeeddÊœdËdÌ„ƒZ—ej˜eddEœdÍdÎ„Z™ejšeddEœdÏdÐ„Z›ejœeddEœdÑdÒ„ZejžeddEœdÓdÔ„ZŸej eddEœdÕdÖ„Z¡ej¢eddEœd×dØ„Z£ej¤eddEœdÙdÚ„Z¥ej¦eddEœdÛdÜ„Z§ej¨eddEœdÝdÞ„Z©ejªeddEœdßdà„Z«ej¬eddEœdádâ„Z­ej®eddEœdãdä„Z¯ej°eddEœdådæ„Z±ej²eddEœdçdè„Z³ej´eddEœdédê„ZµdS )÷r#   NFTr   )r=   rJ   rK   rL   rM   rN   r"   c                 C   s²   |d krt ƒ }|| _|| _|| _|| _d| _|| _d | _|rFt|ƒ| _i | _	i | _
d| _d| _d| _i | _i | _g | _d | _d| _d| _d| _d| _d| _g | _g | _dg| _d S )NFr   r	   TÚcontext)r   r=   rJ   rK   rL   Zcreated_block_contextrM   r*   r   Úimport_aliasesÚblocksÚextends_so_farÚhas_known_extendsÚcode_linenor    rŸ   Ú
debug_infoÚ_write_debug_infoÚ
_new_linesÚ
_last_lineÚ_first_writeÚ_last_identifierÚ_indentationÚ_assign_stackÚ_param_def_blockÚ_context_reference_stack)r%   r=   rJ   rK   rL   rM   rN   r/   r/   r0   r{   -  s8    	
zCodeGenerator.__init__rŽ   c                 C   s
   | j d k	S r)   )r*   r¡   r/   r/   r0   rN   v  s    zCodeGenerator.optimizedzte.NoReturn)ÚmsgÚlinenor"   c                 C   s   t ||| j| jƒ‚dS )z*Fail with a :exc:`TemplateAssertionError`.N)r   rJ   rK   )r%   rÁ   rÂ   r/   r/   r0   Úfail|  s    zCodeGenerator.failc                 C   s   |  j d7  _ d| j › S )zGet a new unique identifier.r	   Zt_)r¼   r¡   r/   r/   r0   Útemporary_identifier€  s    z"CodeGenerator.temporary_identifier)r'   r"   c                 C   s    |   ¡ |_|  |j› d¡ dS )z7Enable buffering for the frame from that point onwards.z = []N)rÄ   r‡   Ú	writeline©r%   r'   r/   r/   r0   r‡   …  s    
zCodeGenerator.buffer)r'   Úforce_unescapedr"   c                 C   s¤   |sŒ|j jrl|  d¡ |  ¡  |  d|j› d¡ |  ¡  |  d¡ |  ¡  |  d|j› d¡ |  ¡  dS |j jrŒ|  d|j› d¡ dS |  d|j› d¡ dS )z(Return the buffer contents of the frame.zif context.eval_ctx.autoescape:zreturn Markup(concat(ú))úelse:zreturn concat(r<   N)r+   r,   rÅ   Úindentr‡   ÚoutdentÚ
autoescape)r%   r'   rÇ   r/   r/   r0   Úreturn_buffer_contentsŠ  s    

z$CodeGenerator.return_buffer_contentsc                 C   s   |  j d7  _ dS )zIndent by one.r	   N©r½   r¡   r/   r/   r0   rÊ   ž  s    zCodeGenerator.indentr	   )Ústepr"   c                 C   s   |  j |8  _ dS )zOutdent by step.NrÎ   )r%   rÏ   r/   r/   r0   rË   ¢  s    zCodeGenerator.outdent)r'   r&   r"   c                 C   s0   |j dkr|  d|¡ n|  |j › d|¡ dS )z%Yield or write into the frame buffer.Núyield ú.append()r‡   rÅ   )r%   r'   r&   r/   r/   r0   Ústart_write¦  s    
zCodeGenerator.start_writec                 C   s   |j dk	r|  d¡ dS )z1End the writing process started by `start_write`.Nr<   )r‡   r?   rÆ   r/   r/   r0   Ú	end_write­  s    
zCodeGenerator.end_write)Úsr'   r&   r"   c                 C   s$   |   ||¡ |  |¡ |  |¡ dS )z4Simple shortcut for start_write + write + end_write.N)rÒ   r?   rÓ   )r%   rÔ   r'   r&   r/   r/   r0   Úsimple_write²  s    
zCodeGenerator.simple_write)r
   r'   r"   c                 C   s>   z$|   d¡ |D ]}|  ||¡ qW n tk
r8   Y nX dS )z•Visit a list of nodes as block in a frame.  If the current frame
        is no buffer a dummy ``if 0: yield None`` is written automatically.
        ÚpassN)rÅ   r-   r°   )r%   r
   r'   r&   r/   r/   r0   Ú
blockvisitº  s    
zCodeGenerator.blockvisit)Úxr"   c                 C   s€   | j rp| jsR| j d| j  ¡ |  j| j 7  _| jdk	rR| j | j| jf¡ d| _d| _| j d| j ¡ d| _ | j |¡ dS )z&Write a string into the output stream.Ú
NFz    r   )	r¹   r»   rL   r?   r¶   r¸   r·   Úappendr½   )r%   rØ   r/   r/   r0   r?   Å  s    
zCodeGenerator.writer   )rØ   r&   Úextrar"   c                 C   s   |   ||¡ |  |¡ dS )z!Combination of newline and write.N)Únewliner?   )r%   rØ   r&   rÛ   r/   r/   r0   rÅ   Ó  s    zCodeGenerator.writeline)r&   rÛ   r"   c                 C   s:   t | jd| ƒ| _|dk	r6|j| jkr6|j| _|j| _dS )z/Add one or more newlines before the next write.r	   N)Úmaxr¹   rÂ   rº   r¸   )r%   r&   rÛ   r/   r/   r0   rÜ   Ú  s    zCodeGenerator.newline)r&   r'   Úextra_kwargsr"   c           	      C   s²  t dd„ tdd„ |jD ƒ|pdƒD ƒƒ}|jD ]}|  d¡ |  ||¡ q.|s |jD ]}|  d¡ |  ||¡ qT|dk	r | ¡ D ]\}}|  d|› d|› ¡ q€|jr¾|  d¡ |  |j|¡ |rŠ|jdk	rÚ|  d	¡ n
|  d
¡ |jD ].}|  |j	›d¡ |  |j
|¡ |  d¡ qê|dk	rN| ¡ D ] \}}|  |›d|› d¡ q,|jdk	r~|  d¡ |  |j|¡ |  d¡ n
|  d¡ n$|jdk	r®|  d¡ |  |j|¡ dS )a+  Writes a function call to the stream for the current node.
        A leading comma is added automatically.  The extra keyword
        arguments may not include python keywords otherwise a syntax
        error could occur.  The extra keyword arguments should be given
        as python dict.
        c                 s   s   | ]}t t t|¡ƒV  qd S r)   )Úis_python_keywordr3   r5   rf   )rY   r]   r/   r/   r0   r[   ï  s   ÿz*CodeGenerator.signature.<locals>.<genexpr>c                 s   s   | ]}|j V  qd S r)   )Úkey©rY   rØ   r/   r/   r0   r[   ñ  s     r/   r9   Nú=z, *z
, **dict({z, **{ú: z}, **r<   Ú}z, **)Úanyr   r(   Úargsr?   r-   rm   Zdyn_argsZ
dyn_kwargsrà   rV   )	r%   r&   r'   rÞ   Zkwarg_workaroundÚargÚkwargrà   rV   r/   r/   r0   Ú	signatureá  sF    þ










zCodeGenerator.signature)r
   r"   c              
   C   s  t ƒ }|D ]}| |¡ q
| j|jdf| j|jdffD ]Ð\}}}t|ƒD ]¼}||kr`|  ¡ ||< |  d¡ |  ¡  |  || › d|› d|›d¡ |  ¡  |  d¡ |  ¡  |  d¡ |  d	|| › d
¡ |  ¡  |  d|dd… › d|›d¡ |  ¡  |  ¡  qHq6dS )aæ  Find all filter and test names used in the template and
        assign them to variables in the compiled namespace. Checking
        that the names are registered with the environment is done when
        compiling the Filter and Test nodes. If the node is in an If or
        CondExpr node, the check is done at runtime instead.

        .. versionchanged:: 3.0
            Filters and tests in If and CondExpr nodes are checked at
            runtime instead of compile time.
        rŸ   r    útry:z = environment.ú[ú]zexcept KeyError:z@internalcodeúdef z
(*unused):zraise TemplateRuntimeError("No Néÿÿÿÿú named z	 found."))	rž   r-   rŸ   r    ÚsortedrÄ   rÅ   rÊ   rË   )r%   r
   rD   r&   Zid_mapro   Ú
dependencyrJ   r/   r/   r0   Úpull_dependencies  s2    ý
 

ÿzCodeGenerator.pull_dependenciesc              	   C   sª   g }|j j ¡ D ]z\}\}}|tkr&q|tkrP|  |› d|  ¡ › d|›d¡ q|tkrn|  |› d|› ¡ q|tkr‚| 	|¡ qt
dƒ‚q|r¦|  d |¡› d¡ d S )Nú = r:   r<   zunknown load instructionú
 = missing)r…   Úloadsrm   r   r   rÅ   Úget_resolve_funcr   r   rÚ   ÚNotImplementedErrorÚjoin)r%   r'   ÚundefsÚtargetÚactionÚparamr/   r/   r0   Úenter_frameE  s    "
zCodeGenerator.enter_frame)r'   Úwith_python_scoper"   c                 C   s>   |s:g }|j jD ]}| |¡ q|r:|  d |¡› d¡ d S )Nró   rô   )r…   rõ   rÚ   rÅ   rø   )r%   r'   rþ   rù   rú   r/   r/   r0   Úleave_frameU  s    zCodeGenerator.leave_frameúasync Ú )Úasync_valueÚ
sync_valuer"   c                 C   s   | j jr|S |S r)   )r=   Úis_async)r%   r  r  r/   r/   r0   Úchoose_async]  s    zCodeGenerator.choose_async)rJ   r"   c                 C   s   |   ¡ › d|› S )Nrí   )r  )r%   rJ   r/   r/   r0   Úfunc`  s    zCodeGenerator.func)r&   r'   r"   c                 C   sš  |  ¡ }|j |¡ t|ƒ}d}tƒ }g }t|jƒD ]@\}}|jdkrJ|}|jdkr`| |j¡ | 	|j 
|j¡¡ q4t|jdƒ}	d|	krè|dk	rÐz|j|t|jƒ   W qâ tk
rÌ   |  d|j¡ Y qâX n| 	|j d¡¡ d|_d|	krd|kr| 	|j d¡¡ d|_d|	kr@d|kr@| 	|j d¡¡ d|_d	|_|j |¡ |  |  d
¡› dd |¡› d|¡ |  ¡  |  |¡ |  |¡ |  |¡ t|jƒD ]°\}}|j 
|j¡}
|  d|
› d¡ |  ¡  z|j|t|jƒ  }W n6 tk
r$   |  |
› d|j›d|j›d¡ Y nX |  |
› d¡ |  ||¡ |  |
¡ |   ¡  q¦|  !¡  |  "|j|¡ | j#|dd | j$|dd |   ¡  ||fS )z/Dump the function def of a macro or call block.NÚcaller)r(   Úvarargs)r  r(   r  zhWhen defining macros or call blocks the special "caller" argument must be omitted or be given a default.Tr(   r  FÚmacror:   r9   ú):úif ú is missing:z = undefined("parameter z was not provided", name=r<   ró   )rÇ   ©rþ   )%r˜   r…   Úanalyze_nodert   ri   Ú	enumerateræ   rJ   r£   rÚ   Úrefrs   ÚbodyÚdefaultsÚlenÚ
IndexErrorrÃ   rÂ   Údeclare_parameterrw   rx   ry   r†   rÅ   r  rø   rÊ   r‡   rý   Úpush_parameter_definitionsr-   Úmark_parameter_storedrË   Úpop_parameter_definitionsr×   rÍ   rÿ   )r%   r&   r'   Ú	macro_refZexplicit_callerZskip_special_paramsræ   Úidxrç   rr   r  Údefaultr/   r/   r0   Ú
macro_bodyc  sr    

ü$


ÿ

zCodeGenerator.macro_body)r  r'   r"   c                 C   sr   d  dd„ |jjD ƒ¡}t|jddƒ}t|jjƒdkr>|d7 }|  d|›d	|› d
|j›d|j›d|j›d¡ dS )z<Dump the macro definition for the def created by macro_body.r9   c                 s   s   | ]}t |jƒV  qd S r)   )ÚreprrJ   rá   r/   r/   r0   r[   ¸  s     z*CodeGenerator.macro_def.<locals>.<genexpr>rJ   Nr	   ú,zMacro(environment, macro, z, (z), z, context.eval_ctx.autoescape))	rø   r&   ræ   Úgetattrr  r?   rx   ry   rw   )r%   r  r'   Z	arg_tuplerJ   r/   r/   r0   Ú	macro_def¶  s    (ÿzCodeGenerator.macro_defru   c                 C   s*   d|j › }| jdk	r&|› d| j›}|S )z.Return a human readable position for the node.zline Nú in )rÂ   rJ   )r%   r&   r–   r/   r/   r0   ÚpositionÂ  s    
zCodeGenerator.positionc                 C   s*   d  dd„ |j ¡  ¡ D ƒ¡}d|› dS )Nr9   c                 s   s    | ]\}}|›d |› V  qdS ©rã   Nr/   )rY   rJ   rú   r/   r/   r0   r[   Ê  s   ÿz3CodeGenerator.dump_local_context.<locals>.<genexpr>Ú{rä   )rø   r…   Zdump_storesrm   )r%   r'   Zitems_kvr/   r/   r0   Údump_local_contextÉ  s    
þz CodeGenerator.dump_local_contextc                 C   s6   |   d¡ |   d¡ |   d¡ |   d¡ |   d¡ dS )z·Writes a common preamble that is used by root and block functions.
        Primarily this sets up common local helpers and enforces a generator
        through a dead branch.
        z$resolve = context.resolve_or_missingz!undefined = environment.undefinedzconcat = environment.concatzcond_expr_undefined = Undefinedzif 0: yield NoneN©rÅ   r¡   r/   r/   r0   Úwrite_commonsÐ  s
    



zCodeGenerator.write_commonsc                 C   s   | j  |j ¡ ¡ dS )aQ  Pushes all parameter targets from the given frame into a local
        stack that permits tracking of yet to be assigned parameters.  In
        particular this enables the optimization from `visit_Name` to skip
        undefined expressions for parameters in macros as macros can reference
        otherwise unbound parameters.
        N)r¿   rÚ   r…   Zdump_param_targetsrÆ   r/   r/   r0   r  Ý  s    z(CodeGenerator.push_parameter_definitionsc                 C   s   | j  ¡  dS )z+Pops the current parameter definitions set.N)r¿   Úpopr¡   r/   r/   r0   r  æ  s    z'CodeGenerator.pop_parameter_definitions)rú   r"   c                 C   s   | j r| j d  |¡ dS )z€Marks a parameter in the current parameter definitions as stored.
        This will skip the enforced undefined checks.
        rî   N)r¿   r¬   ©r%   rú   r/   r/   r0   r  ê  s    z#CodeGenerator.mark_parameter_storedc                 C   s   | j  |¡ d S r)   )rÀ   rÚ   r)  r/   r/   r0   Úpush_context_referenceñ  s    z$CodeGenerator.push_context_referencec                 C   s   | j  ¡  d S r)   )rÀ   r(  r¡   r/   r/   r0   Úpop_context_referenceô  s    z#CodeGenerator.pop_context_referencec                 C   s
   | j d S )Nrî   ©rÀ   r¡   r/   r/   r0   Úget_context_ref÷  s    zCodeGenerator.get_context_refc                 C   s    | j d }|dkrdS |› dS )Nrî   r±   Úresolvez.resolver,  r)  r/   r/   r0   rö   ú  s    
zCodeGenerator.get_resolve_funcc                 C   s   |   ¡ › d|  |¡› dS )Nz	.derived(r<   )r-  r%  rÆ   r/   r/   r0   Úderive_context   s    zCodeGenerator.derive_contextc                 C   s   | j s
dS || j d kS )z4Checks if a given target is an undeclared parameter.Frî   )r¿   r)  r/   r/   r0   Úparameter_is_undeclared  s    z%CodeGenerator.parameter_is_undeclaredc                 C   s   | j  tƒ ¡ dS )z+Pushes a new layer for assignment tracking.N)r¾   rÚ   ri   r¡   r/   r/   r0   Úpush_assign_tracking	  s    z"CodeGenerator.push_assign_trackingc                 C   sˆ  | j  ¡ }|js|js|jr |s$dS dd„ |D ƒ}t|ƒdkr®tt|ƒƒ}|j 	|¡}|jrv|  
d|›d|› ¡ dS |jr–|  
d|›d|› ¡ dS |  
d|›d|› ¡ nx|jrÀ|  
d	¡ n|jrÒ|  
d
¡ n
|  
d¡ t|ƒD ]6\}}|rú|  d¡ |j 	|¡}|  |›d|› ¡ qä|  d¡ |js„|js„|r„t|ƒdkrb|  
d|d ›d¡ n"d tt|ƒ¡}|  
d|› d¡ dS )zoPops the topmost level for assignment tracking and updates the
        context variables if necessary.
        Nc                 S   s    g | ]}|d d… dkr|‘qS )Nr	   Ú_r/   rá   r/   r/   r0   Ú
<listcomp>  s      z5CodeGenerator.pop_assign_tracking.<locals>.<listcomp>r	   z_loop_vars[ú] = z_block_vars[úcontext.vars[z_loop_vars.update({z_block_vars.update({úcontext.vars.update({r9   rã   ú})úcontext.exported_vars.add(r   r<   zcontext.exported_vars.update((rÈ   )r¾   r(  rŒ   r‹   r‰   r  ÚnextÚiterr…   r  rÅ   r  r?   rø   Úmapr  )r%   r'   ÚvarsZpublic_namesrJ   r  r  Ú	names_strr/   r/   r0   Úpop_assign_tracking  sJ    
ÿþýü


z!CodeGenerator.pop_assign_trackingc              	   C   s  |d kst dƒ‚t| j| jƒ}ddlm}m} | jjrDt|| ƒ}nt|ƒ}|  	dd 
|¡ ¡ | jrjdnd}| tj¡d k	}| tj¡D ]4}	|	j| jkr²|  d|	j›d	|	j¡ |	| j|	j< qŠ| tj¡D ]v}
|
j| jkrÌ|
j}|  ¡  | j|< }d
|kr,| d
d¡\}}|  	d|› d|› d|› ¡ qÌ|  	d|› d|› ¡ qÌ|  	d| j›¡ | j	|  d¡› d|› ddd |  ¡  |  ¡  t|ƒ}dt|jdƒkr¼|j  d¡}|  	|› d¡ |j !|¡ d |_"|_#|oà| j$ |_%|rô|  	d¡ |  &|¡ |  '|j¡ |  (|j|¡ | j)|dd |  *¡  |r¢| j$sN|  ¡  |  	d¡ |  ¡  | jjsl|  	d¡ n$|  	d¡ |  ¡  |  	d¡ |  *¡  |  *d| j$  ¡ | j +¡ D ]\}}	|  	|  d| ¡› d|› d|	d¡ |  ¡  |  ¡  t|ƒ}d|_,t|	jdƒ}d|kr*|j  d¡}|  	|› d¡ d |kr\|j  d ¡}|  	|› d!|›d"|› d#¡ |j !|	¡ ||_-|  	d$¡ |  &|¡ |  '|	j¡ |  (|	j|¡ | j)|dd |  *¡  q¬d 
d%d&„ | jD ƒ¡}| j	d'|› d(dd d) 
d*d&„ | j.D ƒ¡}|  	d+|›¡ d S ),Nzno root frame allowedr	   )ÚexportedÚasync_exportedzfrom jinja2.runtime import r9   r  z, environment=environmentzblock z defined twiceÚ.zfrom z import z as zimport zname = Úrootz(context, missing=missingr
  )rÛ   r%   r¡   z = TemplateReference(context)Tzparent_template = Noner  úif parent_template is not None:z4yield from parent_template.root_render_func(context)z=async for event in parent_template.root_render_func(context):zyield eventZblock_)r%   ÚsuperrD  z = context.super(z, block_r<   z_block_vars = {}c                 s   s   | ]}|›d |› V  qdS )z: block_Nr/   rá   r/   r/   r0   r[   «  s     z/CodeGenerator.visit_Template.<locals>.<genexpr>z
blocks = {rä   ú&c                 s   s    | ]\}}|› d |› V  qdS )râ   Nr/   r\   r/   r/   r0   r[   ­  s     zdebug_info = )/ÚAssertionErrorr   r=   rJ   Zruntimer?  r@  r  rð   rÅ   rø   rM   Úfindr
   ÚExtendsÚfind_allr©   r³   rÃ   rÂ   ÚImportedNameÚ
importnamer²   rÄ   Úrsplitr  rÊ   r'  r$   rs   r  r…   r  r  r‰   rŠ   rµ   r†   rý   rò   r×   rÿ   rË   rm   rŒ   rˆ   r·   )r%   r&   r'   r+   r?  r@  Zexported_namesZenvenvZhave_extendsrˆ   Úimport_ÚimpÚaliasÚmoduleÚobjr  rJ   rŒ   rr   Zblocks_kv_strZdebug_kv_strr/   r/   r0   Úvisit_Template:  sª    
 ÿ



ÿ
ý



zCodeGenerator.visit_Templatec                 C   s  d}|j r8| jrdS | jdkr8|  d¡ |  ¡  |d7 }|jrJ|  |¡}n|  ¡ }|jr”|  d|j	›d|¡ |  ¡  |  d|j	›d|¡ |  
¡  | jjsÄ|jdkrÄ|  d	|j	›d
|› d|¡ n@|  |  ¡ › d|j	›d
|› d|¡ |  ¡  |  d|¡ |  
¡  |  
|¡ dS )z.Call a block and register it for the template.r   Núif parent_template is None:r	   zif len(context.blocks[z]) <= 1:z+raise TemplateRuntimeError("Required block z not found")zyield from context.blocks[z][0](r<   zfor event in context.blocks[r
  Úevent)r‰   rµ   r´   rÅ   rÊ   Úscopedr/  r-  ÚrequiredrJ   rË   r=   r  r‡   r  rÕ   )r%   r&   r'   rƒ   r±   r/   r/   r0   r¦   °  s@    

þ ÿýzCodeGenerator.visit_Blockc                 C   sÆ   |j s|  d|j¡ | jdkrV| js6|  d¡ |  ¡  |  d¡ | jrNtƒ ‚n|  ¡  |  d|¡ |  	|j
|¡ |  d| j›d¡ |  d¡ |  ¡  |  d	¡ |  ¡  |jr´d
| _|  jd7  _dS )zCalls the extender.z,cannot use extend from a non top-level scoper   rC  z5raise TemplateRuntimeError("extended multiple times")z+parent_template = environment.get_template(r9   r<   z9for name, parent_block in parent_template.blocks.items():z8context.blocks.setdefault(name, []).append(parent_block)Tr	   N)r‰   rÃ   rÂ   r´   rµ   rÅ   rÊ   r°   rË   r-   Útemplater?   rJ   rŠ   rB   r/   r/   r0   Úvisit_ExtendsÛ  s(    




zCodeGenerator.visit_Extendsc                 C   sp  |j r|  d¡ |  ¡  d}t|jtjƒrVt|jjtƒr>d}qnt|jjt	t
fƒrnd}nt|jtjtjfƒrnd}|  d|› d|¡ |  |j|¡ |  d| j›d¡ |j rè|  ¡  |  d	¡ |  ¡  |  d
¡ |  ¡  |  d¡ |  ¡  d}|jr|  |  ¡ › d|  |¡› d¡ n$| jjr,|  d¡ n|  d¡ d}|s\|  ¡  |  d|¡ |  ¡  |j rl|  ¡  dS )zHandles includes.rê   Zget_or_select_templateZget_templateZselect_templateztemplate = environment.r:   r9   r<   zexcept TemplateNotFound:rÖ   rÉ   FzUfor event in template.root_render_func(template.new_context(context.get_all(), True, z)):zGfor event in (await template._get_default_module_async())._body_stream:z6yield from template._get_default_module()._body_streamTrT  N)Zignore_missingrÅ   rÊ   rO   rW  r
   ÚConstrV   rf   rg   rh   ÚTupleÚListr-   r?   rJ   rË   Úwith_contextr  r%  r=   r  rÕ   )r%   r&   r'   Ú	func_nameZskip_event_yieldr/   r/   r0   Úvisit_Include  sN    



ÿ
ÿ
zCodeGenerator.visit_Includec                 C   sˆ   |   |  d¡› d¡ |  |j|¡ |   d| j›d¡ |jrld|  d¡› }|   |› d|  |¡› d¡ n|   d	|  d¡› d
¡ d S )Núawait zenvironment.get_template(r9   z).Zmake_moduleZ_asyncz(context.get_all(), True, r<   Z_get_default_modulez	(context))r?   r  r-   rW  rJ   r\  r%  )r%   r&   r'   Úf_namer/   r/   r0   Ú_import_common9  s    ÿzCodeGenerator._import_commonc                 C   sl   |   |j |j¡› d|¡ |jr6|  d|j›d¡ |  ||¡ |jrh|j d¡sh|   d|j›d¡ dS )zVisit regular imports.ró   r5  r4  r2  úcontext.exported_vars.discard(r<   N)rÅ   r…   r  rú   r‰   r?   ra  Ú
startswithrB   r/   r/   r0   Úvisit_ImportH  s    zCodeGenerator.visit_Importc           
   	      sª  |   |¡ |  d¡ |  |ˆ ¡ g }g }|jD ]Â}t|tƒrF|\}}n|}|  ˆ j |¡› d|›d¡ |  dˆ j |¡› d¡ |  	¡  d|  
|¡› d|›}|  ˆ j |¡› d|›d	|›d
¡ |  ¡  ˆ jr.| |¡ | d¡s.| |¡ q.|rXt|ƒdkr.|d }|  d|›dˆ j |¡› ¡ n*d ‡ fdd„|D ƒ¡}|  d|› d¡ |r¦t|ƒdkr„|  d|d ›d
¡ n"d tt|ƒ¡}	|  d|	› d¡ dS )zVisit named imports.zincluded_template = z = getattr(included_template, z
, missing)r  r  z9the template {included_template.__name__!r} (imported on z%) does not export the requested name z = undefined(fz, name=r<   r2  r	   r   r5  r4  r9   c                 3   s$   | ]}|›d ˆ j  |¡› V  qdS r#  )r…   r  )rY   rJ   ©r'   r/   r0   r[   x  s    z1CodeGenerator.visit_FromImport.<locals>.<genexpr>r6  r7  rb  z)context.exported_vars.difference_update((rÈ   N)rÜ   r?   ra  ro   rO   rg   rÅ   r…   r  rÊ   r"  rË   r‰   rÚ   rc  r  rø   r;  r  )
r%   r&   r'   Z	var_namesZdiscarded_namesrJ   rO  ÚmessageZnames_kvr=  r/   re  r0   Úvisit_FromImportS  sN    




ÿÿÿ

 ÿ
ÿzCodeGenerator.visit_FromImportc                 C   s€  |  ¡ }d|_|  ¡ }|  ¡ }|jpRdt|jdddƒkpRtdd„ | tj¡D ƒƒ}d }|rh|j	 
d¡}|j	j|dd	 |jrŽ|j	j|d
d	 |jr†|  ¡ }|j	j|dd	 |  |  |¡› d|j¡ |  ¡  |  |¡ |  |  dd¡¡ |  |j|¡ |  d¡ |  |  dd¡¡ |  d¡ |  ¡  |  d|j¡ |  |j|¡ |  d¡ |  ¡  |  d¡ |  |j|¡ |  d¡ | j|dd |jrÀ|  |  d¡› d|¡ |  ¡  |  |¡ |j|_|rÖ|  |› d¡ | tj¡D ],}	|	jdkrâ|	jdkrâ|  d|	j¡ qâ|jr0|  ¡ }
|  |
› d¡ |  |  dd¡|¡ |  |j|¡ |rx|  d|› d|  d¡› d¡ n
|  d¡ |jrš|  |› d¡ |jr®|  d ¡ nB| jj rÈ|sÈ|  d!¡ |  |j!|¡ | jj rð|sð|  d"¡ |jr|  d"¡ |jr|  d#¡ n|  |r$d$nd¡ |  ¡  |  |¡ |  d%¡ |  "|j#|¡ |jrl|  |
› d&¡ |  ¡  | j||joˆ|j d |jrÚ|  d|
› d¡ |  ¡  |  |¡ |  "|j|¡ |  |¡ |  ¡  |jr`|  $|¡ |  ¡  |  %||¡ |  |  d'¡› d(¡ | jj r*|  d!¡ |  |j!|¡ | jj rL|  d"¡ |  d)¡ |  &|¡ | j'r|| j'd*  (|j	j)¡ d S )+NTÚloop)r  )Úonly)rh  c                 s   s   | ]}|j V  qd S r)   )rU  )rY   rˆ   r/   r/   r0   r[   ’  s     z*CodeGenerator.visit_For.<locals>.<genexpr>r  )Z
for_branchÚelseÚtestz(fiter):z
async for zfor r!  zauto_aiter(fiter)Zfiterú:r  rÐ   é   r  z%(reciter, loop_render_func, depth=0):rô   Ústorez8Can't assign to special loop variable in for-loop targetz = 1r9   ZAsynczLoopContext(r:   Zreciterzauto_aiter(r<   z&, undefined, loop_render_func, depth):z, undefined):z_loop_vars = {}z = 0r_  zloop(z, loop)rî   )*r˜   r‹   Ú	recursivers   Ziter_child_nodesrå   rI  r
   r©   r…   r  r  Úelse_rk  rÄ   rÅ   r  rÊ   rý   r  r-   rú   r?   rË   rÿ   r‡   r¯   r«   rJ   rÃ   rÂ   r=   r  r:  r×   r  rÍ   rÒ   rÓ   r¾   Údifference_updateZstores)r%   r&   r'   r‹   Z
test_frameZ
else_frameZextended_loopZloop_refZloop_filter_funcrJ   Ziteration_indicatorr/   r/   r0   Ú	visit_For…  sÖ    ÿÿü





 ÿ
þ 





 ÿ








zCodeGenerator.visit_Forc                 C   sÊ   |  ¡ }|  d|¡ |  |j|¡ |  d¡ |  ¡  |  |j|¡ |  ¡  |j	D ]F}|  d|¡ |  |j|¡ |  d¡ |  ¡  |  |j|¡ |  ¡  qP|j
rÆ|  d¡ |  ¡  |  |j
|¡ |  ¡  d S )Nr  rl  zelif rÉ   )r™   rÅ   r-   rk  r?   rÊ   r×   r  rË   Úelif_rp  )r%   r&   r'   Zif_framers  r/   r/   r0   Úvisit_If  s&    




zCodeGenerator.visit_Ifc                 C   s|   |   ||¡\}}|  ¡  |jrR|j d¡s>|  d|j›d¡ |  d|j›d¡ |  |j |j¡› d¡ |  	||¡ d S )Nr2  r8  r<   r5  r4  ró   )
r  rÜ   r‰   rJ   rc  r?   rÅ   r…   r  r   )r%   r&   r'   Zmacro_framer  r/   r/   r0   Úvisit_Macro%  s    zCodeGenerator.visit_Macroc                 C   sR   |   ||¡\}}|  d¡ |  ||¡ |  ||¡ | j|j|dd |  |¡ d S )Nz	caller = T)Úforward_caller)r  rÅ   r   rÒ   Ú
visit_CallÚcallrÓ   )r%   r&   r'   Z
call_framer  r/   r/   r0   Úvisit_CallBlock/  s    
zCodeGenerator.visit_CallBlockc                 C   sh   |  ¡ }|j |¡ |  |¡ |  |¡ |  |j|¡ |  ||¡ |  |j	|¡ |  
|¡ |  |¡ d S r)   )r˜   r…   r  rý   r‡   r×   r  rÒ   r¤   ÚfilterrÓ   rÿ   )r%   r&   r'   Zfilter_framer/   r/   r0   Úvisit_FilterBlock7  s    


zCodeGenerator.visit_FilterBlockc                 C   s|   |  ¡ }|j |¡ |  |¡ t|j|jƒD ]2\}}|  ¡  |  ||¡ |  	d¡ |  ||¡ q,|  
|j|¡ |  |¡ d S ©Nró   )r˜   r…   r  rý   ÚzipÚtargetsÚvaluesrÜ   r-   r?   r×   r  rÿ   )r%   r&   r'   Z
with_framerú   Úexprr/   r/   r0   Ú
visit_WithB  s    

zCodeGenerator.visit_Withc                 C   s   |   |¡ |  |j|¡ d S r)   )rÜ   r-   r&   rB   r/   r/   r0   Úvisit_ExprStmtN  s    
zCodeGenerator.visit_ExprStmtc                   @   s4   e Zd ZU ejejdef  ed< eje ed< dS )zCodeGenerator._FinalizeInfo.ÚconstÚsrcN)r|   r}   r~   r3   r›   ÚCallablerf   Ú__annotations__r/   r/   r/   r0   Ú_FinalizeInfoR  s   
r‡  rU   c                 C   s   t | ƒS )z¼The default finalize function if the environment isn't
        configured with one. Or, if the environment has one, this is
        called on that function's output for constants.
        )rf   rn   r/   r/   r0   Ú_default_finalizeV  s    zCodeGenerator._default_finalizeÚ	_finalizec                    sÂ   ˆj dk	rˆj S ˆj }‰ d}ˆjjr®d}ˆjj‰tjdtjdtjdi t ˆ¡¡}d}|dkr|t	j
t	j
dœ‡ ‡fdd„}n2|› |› d	}|dkr®t	j
t	j
dœ‡ ‡‡fd
d„}ˆ ||¡ˆ_ ˆj S )a‰  Build the finalize function to be used on constants and at
        runtime. Cached so it's only created once for all output nodes.

        Returns a ``namedtuple`` with the following attributes:

        ``const``
            A function to finalize constant data at compile time.

        ``src``
            Source code to output around nodes to be evaluated at
            runtime.
        Nzenvironment.finalize(r±   úcontext.eval_ctxr=   rU   c                    s   ˆ ˆ| ƒƒS r)   r/   rn   )r  Úenv_finalizer/   r0   Úfinalize‚  s    z.CodeGenerator._make_finalize.<locals>.finalizer9   c                    s   ˆ ˆˆj | ƒƒS r)   )r=   rn   ©r  r‹  r%   r/   r0   rŒ  Š  s    )r‰  rˆ  r=   rŒ  r   r±   Úeval_contextÚgetÚfrom_objr3   r4   r‡  )r%   rŒ  r„  Úpass_argr/   r  r0   Ú_make_finalize`  s0    

   ýûzCodeGenerator._make_finalize)Úgroupr"   c                 C   s   t t|ƒƒS )z™Given a group of constant values converted from ``Output``
        child nodes, produce a string to write to the template module
        source.
        )r  r   )r%   r“  r/   r/   r0   Ú_output_const_repr  s    z CodeGenerator._output_const_repr)r&   r'   rŒ  r"   c                 C   s:   |  |j¡}|jjrt|ƒ}t|tjƒr0t|ƒS | |¡S )aC  Try to optimize a child of an ``Output`` node by trying to
        convert it to constant, finalized data at compile time.

        If :exc:`Impossible` is raised, the node is not constant and
        will be evaluated at runtime. Any other exception will also be
        evaluated at runtime for easier debugging.
        )	Úas_constr+   rÌ   r   rO   r
   ÚTemplateDatarf   rƒ  )r%   r&   r'   rŒ  rƒ  r/   r/   r0   Ú_output_child_to_const—  s    
z$CodeGenerator._output_child_to_constc                 C   sL   |j jr|  d¡ n|j jr(|  d¡ n
|  d¡ |jdk	rH|  |j¡ dS )zXOutput extra source code before visiting a child of an
        ``Output`` node.
        z1(escape if context.eval_ctx.autoescape else str)(zescape(zstr(N)r+   r,   r?   rÌ   r„  ©r%   r&   r'   rŒ  r/   r/   r0   Ú_output_child_pre¬  s    

zCodeGenerator._output_child_prec                 C   s"   |   d¡ |jdk	r|   d¡ dS )zWOutput extra source code after visiting a child of an
        ``Output`` node.
        r<   N)r?   r„  r˜  r/   r/   r0   Ú_output_child_post¼  s    

z CodeGenerator._output_child_postc           	   
   C   sø  |j r"| jrd S |  d¡ |  ¡  |  ¡ }g }|jD ]ˆ}z,|jsTt|tjƒsTt 	¡ ‚|  
|||¡}W n( tj	tfk
rŒ   | |¡ Y q4Y nX |r°t|d tƒr°|d  |¡ q4| |g¡ q4|jd k	rt|ƒdkrê|  |j› d¡ n|  |j› d¡ |  ¡  |D ]ª}t|tƒrN|  |¡}|jd kr>|  d| ¡ n|  |d ¡ nb|jd krh|  d|¡ n
|  |¡ |  |||¡ |  ||¡ |  |||¡ |jd k	r|  d¡ q|jd k	rä|  ¡  |  t|ƒdkrÞdnd	¡ |j rô|  ¡  d S )
NrS  rî   r	   rÑ   z	.extend((rÐ   r  r<   rÈ   )r†   rµ   rÅ   rÊ   r’  r
   rƒ  rO   r–  Ú
Impossibler—  Ú	ExceptionrÚ   rh   r‡   r  r”  rÜ   r™  r-   rš  r?   rË   )	r%   r&   r'   rŒ  r  Úchildrƒ  ÚitemÚvalr/   r/   r0   Úvisit_OutputÇ  sZ    

ý
ú



zCodeGenerator.visit_Outputc                 C   sF   |   ¡  |  |¡ |  |j|¡ |  d¡ |  |j|¡ |  |¡ d S r|  )r1  rÜ   r-   rú   r?   r&   r>  rB   r/   r/   r0   Úvisit_Assign  s    

zCodeGenerator.visit_Assignc                 C   s¶   |   ¡  | ¡ }d|_|j |¡ |  |¡ |  |¡ |  |j|¡ |  	|¡ |  
|j|¡ |  d¡ |jd k	r€|  |j|¡ n|  d|j› d¡ |  d¡ |  |¡ |  |¡ d S )NFz9 = (Markup if context.eval_ctx.autoescape else identity)(úconcat(r<   )r1  r˜   r†   r…   r  rý   r‡   r×   r  rÜ   r-   rú   r?   rz  r¤   r>  rÿ   )r%   r&   r'   rŒ   r/   r/   r0   Úvisit_AssignBlock  s     






zCodeGenerator.visit_AssignBlockc              	   C   s¨   |j dkr4|js|js|jr4| jr4| jd  |j¡ |j |j¡}|j dkrš|j 	|¡}|d k	rv|d t
krv|  |¡rš|  d|j›d|› d|› d¡ d S |  |¡ d S )	Nrn  rî   rª   r   z(undefined(name=z) if z is missing else r<   )r«   r‰   r‹   rŒ   r¾   r£   rJ   r…   r  Z	find_loadr   r0  r?   )r%   r&   r'   r  rª   r/   r/   r0   r­   6  s.    
ÿÿÿ
ÿ
þýÿzCodeGenerator.visit_Namec                 C   sV   |j  |j¡}|  d|› d¡ |  ¡  |  d¡ |  ¡  |  |› d|j›d¡ d S )Nzif not isinstance(z, Namespace):zMraise TemplateRuntimeError("cannot assign attribute on non-namespace object")rë   rì   )r…   r  rJ   rÅ   rÊ   rË   Úattr)r%   r&   r'   r  r/   r/   r0   Úvisit_NSRefO  s    ÿzCodeGenerator.visit_NSRefc                 C   s8   |  |j¡}t|tƒr&|  t|ƒ¡ n|  t|ƒ¡ d S r)   )r•  r+   rO   rc   r?   rf   r  )r%   r&   r'   rŸ  r/   r/   r0   Úvisit_Const]  s    
zCodeGenerator.visit_Constc                 C   sJ   z|   t| |j¡ƒ¡ W n* tjk
rD   |   d|j›d¡ Y nX d S ©Nz6(Markup if context.eval_ctx.autoescape else identity)(r<   )r?   r  r•  r+   r
   r›  ÚdatarB   r/   r/   r0   Úvisit_TemplateDatad  s    ÿz CodeGenerator.visit_TemplateDatac                 C   sV   |   d¡ d}t|jƒD ]"\}}|r.|   d¡ |  ||¡ q|   |dkrLdnd¡ d S )Nr:   rî   r9   r   z,)r<   ©r?   r  rm   r-   ©r%   r&   r'   r  rž  r/   r/   r0   Úvisit_Tuplel  s    

zCodeGenerator.visit_Tuplec                 C   sF   |   d¡ t|jƒD ]"\}}|r*|   d¡ |  ||¡ q|   d¡ d S )Nrë   r9   rì   rª  r«  r/   r/   r0   Ú
visit_Listu  s    

zCodeGenerator.visit_Listc                 C   s`   |   d¡ t|jƒD ]<\}}|r*|   d¡ |  |j|¡ |   d¡ |  |j|¡ q|   d¡ d S )Nr$  r9   rã   rä   )r?   r  rm   r-   rà   rV   r«  r/   r/   r0   Ú
visit_Dict}  s    


zCodeGenerator.visit_Dictú+ú-Ú*ú/z//z**ú%ÚandÚorznot c                 C   s`   |j jrd}n|j jrd}nd}|  |› d¡ |jD ]}|  ||¡ |  d¡ q6|  d¡ d S )Nz8(markup_join if context.eval_ctx.volatile else str_join)Zmarkup_joinZstr_joinz((r9   rÈ   )r+   r,   rÌ   r?   r
   r-   )r%   r&   r'   r]  rç   r/   r/   r0   Úvisit_Concat”  s    
zCodeGenerator.visit_Concatc                 C   s>   |   d¡ |  |j|¡ |jD ]}|  ||¡ q|   d¡ d S )Nr:   r<   )r?   r-   r€  Úops)r%   r&   r'   r7   r/   r/   r0   Úvisit_Compare¢  s
    

zCodeGenerator.visit_Comparec                 C   s*   |   dt|j › d¡ |  |j|¡ d S )Nr;   )r?   Ú	operatorsr7   r-   r€  rB   r/   r/   r0   Úvisit_Operandª  s    zCodeGenerator.visit_Operandc                 C   sT   | j jr|  d¡ |  d¡ |  |j|¡ |  d|j›d¡ | j jrP|  d¡ d S )Nú(await auto_await(zenvironment.getattr(r9   r<   rÈ   )r=   r  r?   r-   r&   r¤  rB   r/   r/   r0   Úvisit_Getattr®  s    

zCodeGenerator.visit_Getattrc                 C   s¢   t |jtjƒr@|  |j|¡ |  d¡ |  |j|¡ |  d¡ n^| jjrR|  d¡ |  d¡ |  |j|¡ |  d¡ |  |j|¡ |  d¡ | jjrž|  d¡ d S )Nrë   rì   r»  zenvironment.getitem(r9   r<   rÈ   )	rO   rç   r
   ÚSlicer-   r&   r?   r=   r  rB   r/   r/   r0   Úvisit_Getitemº  s    




zCodeGenerator.visit_Getitemc                 C   s`   |j d k	r|  |j |¡ |  d¡ |jd k	r:|  |j|¡ |jd k	r\|  d¡ |  |j|¡ d S )Nrl  )Ústartr-   r?   ÚstoprÏ   rB   r/   r/   r0   Úvisit_SliceÏ  s    




zCodeGenerator.visit_Slice)r&   r'   Ú	is_filterr"   c                 c   s  | j jr|  d¡ |r@|  | j|j › d¡ | j j |j¡}n(|  | j|j › d¡ | j j |j¡}|d kr |js |r~dnd}|  d|› d|j›d|j	¡ t
jdt
jd	t
j d
i t
 |¡¡}|d k	rÚ|  |› d¡ d V  |  ||¡ |  d¡ | j jr
|  d¡ d S )Nr»  r:   rz  rk  zNo rï   rA  r±   rŠ  r=   r9   r<   rÈ   )r=   r  r?   rŸ   rJ   r  r    r   rÃ   rÂ   r   r±   rŽ  r  ré   )r%   r&   r'   rÂ  r  Ú	type_namer‘  r/   r/   r0   Ú_filter_test_commonÙ  s4    
   ýû

z!CodeGenerator._filter_test_commonc              	   C   s   |   ||d¡x |jd k	r*|  |j|¡ nX|jjrP|  d|j› d|j› d¡ n2|jjrn|  d|j› d¡ n|  d|j› d¡ W 5 Q R X d S )NTz(Markup(concat(z.)) if context.eval_ctx.autoescape else concat(rÈ   zMarkup(concat(r¢  r<   )rÄ  r&   r-   r+   r,   r?   r‡   rÌ   rB   r/   r/   r0   r¤     s    
ÿzCodeGenerator.visit_Filterc              	   C   s,   |   ||d¡ |  |j|¡ W 5 Q R X d S rv   )rÄ  r-   r&   rB   r/   r/   r0   r¥     s    zCodeGenerator.visit_Testc                    sl   ˆ   ¡ ‰ d dœ‡ ‡‡fdd„}ˆ d¡ ˆ ˆjˆ ¡ ˆ d¡ ˆ ˆjˆ ¡ ˆ d¡ |ƒ  ˆ d¡ d S )NrŽ   c                      s8   ˆj d k	rˆ ˆj ˆ ¡ d S ˆ dˆ ˆ¡› d¡ d S )Nz1cond_expr_undefined("the inline if-expression on z6 evaluated to false and no else section was defined."))Zexpr2r-   r?   r"  r/   ©r'   r&   r%   r/   r0   Úwrite_expr2  s    
ÿz1CodeGenerator.visit_CondExpr.<locals>.write_expr2r:   z if z else r<   )r™   r?   r-   Úexpr1rk  )r%   r&   r'   rÆ  r/   rÅ  r0   Úvisit_CondExpr  s    


zCodeGenerator.visit_CondExpr)r&   r'   rv  r"   c                 C   sÈ   | j jr|  d¡ | j jr&|  d¡ n
|  d¡ |  |j|¡ |rJddind }|jr\ddini }|jrnddini }|r†|j|f|Ž n|sŽ|ršt	|f|Ž}|  
|||¡ |  d¡ | j jrÄ|  d¡ d S )	Nr»  zenvironment.call(context, zcontext.call(r  Z
_loop_varsZ_block_varsr<   rÈ   )r=   r  r?   r>   r-   r&   r‹   rŒ   r“   rl   ré   )r%   r&   r'   rv  rÞ   Zloop_kwargsZblock_kwargsr/   r/   r0   rw  0  s"    


zCodeGenerator.visit_Callc                 C   s"   |   |jd ¡ |  |j|¡ d S )Nrâ   )r?   rà   r-   rV   rB   r/   r/   r0   Úvisit_KeywordG  s    zCodeGenerator.visit_Keywordc                 C   s&   |   d¡ |  |j|¡ |   d¡ d S )NzMarkup(r<   ©r?   r-   r€  rB   r/   r/   r0   Úvisit_MarkSafeM  s    
zCodeGenerator.visit_MarkSafec                 C   s&   |   d¡ |  |j|¡ |   d¡ d S r§  rÊ  rB   r/   r/   r0   Úvisit_MarkSafeIfAutoescapeR  s    
z(CodeGenerator.visit_MarkSafeIfAutoescapec                 C   s   |   d|j ¡ d S )Nzenvironment.©r?   rJ   rB   r/   r/   r0   Úvisit_EnvironmentAttributeY  s    z(CodeGenerator.visit_EnvironmentAttributec                 C   s   |   d|j›d|j› ¡ d S )Nzenvironment.extensions[z].)r?   Ú
identifierrJ   rB   r/   r/   r0   Úvisit_ExtensionAttribute^  s    z&CodeGenerator.visit_ExtensionAttributec                 C   s   |   | j|j ¡ d S r)   )r?   r²   rK  rB   r/   r/   r0   Úvisit_ImportedNamec  s    z CodeGenerator.visit_ImportedNamec                 C   s   |   |j¡ d S r)   rÍ  rB   r/   r/   r0   Úvisit_InternalNamef  s    z CodeGenerator.visit_InternalNamec                 C   s   |   d¡ d S )Nr±   )r?   rB   r/   r/   r0   Úvisit_ContextReferencei  s    z$CodeGenerator.visit_ContextReferencec                 C   s   |   |  |¡¡ d S r)   )r?   r/  rB   r/   r/   r0   Úvisit_DerivedContextReferencen  s    z+CodeGenerator.visit_DerivedContextReferencec                 C   s   |   d|¡ d S )NÚcontinuer&  rB   r/   r/   r0   Úvisit_Continues  s    zCodeGenerator.visit_Continuec                 C   s   |   d|¡ d S )NÚbreakr&  rB   r/   r/   r0   Úvisit_Breakv  s    zCodeGenerator.visit_Breakc                 C   s:   |  ¡ }|j |¡ |  |¡ |  |j|¡ |  |¡ d S r)   )r˜   r…   r  rý   r×   r  rÿ   )r%   r&   r'   Úscope_framer/   r/   r0   Úvisit_Scopey  s
    
zCodeGenerator.visit_Scopec                 C   s   |   ¡ }|  |› d|  |¡› ¡ |  |› d¡ |  |j|¡ |  |¡ |jdd}|j |¡ |  	|¡ |  
|j|¡ |  |¡ |  ¡  d S )Nró   z.vars = T)r—   )rÄ   rÅ   r/  r-   r±   r*  r˜   r…   r  rý   r×   r  rÿ   r+  )r%   r&   r'   r«   rÙ  r/   r/   r0   Úvisit_OverlayScope€  s    


z CodeGenerator.visit_OverlayScopec              	   C   st   |j D ]h}|  d|j› d¡ |  |j|¡ z|j |j¡}W n tjk
r\   d|j_	Y qX t
|j|j|ƒ qd S )Nzcontext.eval_ctx.ró   T)ÚoptionsrÅ   rà   r-   rV   r•  r+   r
   r›  r,   Úsetattr)r%   r&   r'   ÚkeywordrŸ  r/   r/   r0   Úvisit_EvalContextModifierŽ  s    
z'CodeGenerator.visit_EvalContextModifierc                 C   sh   |   ¡ }|j ¡ }|  |› d¡ |  ||¡ |jD ]}|  ||¡ q4|j |¡ |  d|› d¡ d S )Nz = context.eval_ctx.save()zcontext.eval_ctx.revert(r<   )rÄ   r+   ÚsaverÅ   rß  r  r-   Úrevert)r%   r&   r'   Zold_ctx_nameZ	saved_ctxr  r/   r/   r0   Úvisit_ScopedEvalContextModifier›  s    

z-CodeGenerator.visit_ScopedEvalContextModifier)NFT)F)r	   )N)N)Nr   )Nr   )N)F)r   r  )N)F)¶r|   r}   r~   r3   r›   rf   ÚTextIOra   r{   ÚpropertyrN   rb   rÃ   rÄ   r$   r‡   rÍ   rÊ   rË   r
   ÚNoderÒ   rÓ   rÕ   r®   r×   r?   rÅ   rÜ   r   ZCallr§   r¨   ÚMappingr4   ré   rò   rý   rÿ   r  r  r€   r   rZ  rt   r  r   r"  r%  r'  r  r  r  r*  r+  r-  rö   r/  r0  r1  r>  rP   rR  r©   r¦   rH  rX  ÚIncluder^  ZImportZ
FromImportra  rd  rg  ZForrr  ZIfrt  ru  ry  ZFilterBlockr{  ZWithr  ZExprStmtr‚  Ú
NamedTupler‡  Ústaticmethodrˆ  r‰  r†  r’  r”  r2   r—  r™  rš  ZOutputr   ZAssignr¡  ZAssignBlockr£  r¯   r­   ZNSRefr¥  rY  r¦  r–  r©  r¬  r[  r­  ÚDictr®  rG   Z	visit_AddZ	visit_SubZ	visit_MulZ	visit_DivZvisit_FloorDivZ	visit_PowZ	visit_ModZ	visit_AndZvisit_OrrI   Z	visit_PosZ	visit_NegZ	visit_Notr6   ZConcatr¶  ZComparer¸  ZOperandrº  ZGetattrr¼  ZGetitemr¾  r½  rÁ  r   ÚIteratorrÄ  r¤   r¥   ZCondExprrÈ  rw  ÚKeywordrÉ  ZMarkSaferË  ZMarkSafeIfAutoescaperÌ  ZEnvironmentAttributerÎ  ZExtensionAttributerÐ  rJ  rÑ  ZInternalNamerÒ  ZContextReferencerÓ  ZDerivedContextReferencerÔ  ÚContinuerÖ  ZBreakrØ  ZScoperÚ  ZOverlayScoperÛ  ZEvalContextModifierrß  ZScopedEvalContextModifierrâ  r/   r/   r/   r0   r#   ,  s   
   ù
øI ÿ þ ÿ  
þ   ÿ 
 þ üû9+ þS	. ÿ þv++4 þ2 
0  þ  þ  þO	

  þ) ÿ  þ þ þ þ þ þ þ þ)NFT)Frš   Útypingr3   Ú
contextlibr   Ú	functoolsr   Úior   Ú	itertoolsr   rÞ  r   rß   Z
markupsafer   r   r  r
   Ú
exceptionsr   Z
idtrackingr   r   r   r   r   r   r*   r   Úutilsr   r   rD   r   ÚTYPE_CHECKINGZtyping_extensionsÚter=   r   ÚTypeVarr…  r4   r   r¹  r6   rf   rE   rG   rH   rI   rP   r›   rã  ra   rT   rX   r®   rå  ÚSetrs   rt   r$   ÚRuntimeErrorrq   rž   rp   rœ  r°   r#   r/   r/   r/   r0   Ú<module>   s~   ø"þ   ù
ø
 þS