CasperSecurity
o
쑛h� � @ s& U d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dlm
Z d dlm Z d dl
mZ eed< zd d lmZ d d
lmZ d ZeZW n eefyZ d ZeZY nw e�e�Zd
Z G dd� de�Z G dd� de�Z e � d�dd� �Z dd� Z!dd� Z"d dd�Z#d d � Z$d!d d �Z%dS )"� N)�Any)�TemplateSyntaxError)� performance)�
type_utils)�util)�
write_file�
JUndefined)�DebugUndefined)�TemplateTFzCI_MISSING_JINJA_VAR/c
sT e Zd Zdeddf� fdd� Zdd� Ze dd ed
ed edefd d
��ZdZ � Z
S )�JinjaSyntaxParsingException�error�returnNc s* t � �|jpd|j|j|j� |j| _d S )Nzunknown syntax error)�super�__init__�message�lineno�name�filename�source)�selfr �� __class__� �5/usr/lib/python3/dist-packages/cloudinit/templater.pyr / s � z$JinjaSyntaxParsingException.__init__c C s( | j | j| j| j�� | jd �� d�S )�;Avoid jinja2.TemplateSyntaxError multi-line __str__ format.� �� syntax_error� line_number� line_content)�format_error_messager r r �
splitlines�strip�r r r r �__str__; s
�z#JinjaSyntaxParsingException.__str__� r r r c C s$ |rd|� �nd}t jj| ||d�S )r z: r% r )r �message_template�formatr r r r r C s �z0JinjaSyntaxParsingException.format_error_messagezfUnable to parse Jinja template due to syntax error: {syntax_error} on line {line_number}{line_content})r% ) �__name__�
__module__� __qualname__r r r$ � staticmethod�strr r& �
__classcell__r r r r r . s( �� ���� �
�r c @ s e Zd ZdZdd� Zdd� ZdS )�UndefinedJinjaVariablez>Class used to represent any undefined jinja template variable.c C s dt | jf S )Nz%s%s)�MISSING_JINJA_PREFIX�_undefined_namer# r r r r$ \ s z UndefinedJinjaVariable.__str__c C s$ t |��td�}tdj| j|d���)Nr% zhUndefined jinja variable: "{this}-{other}". Jinja tried subtraction. Perhaps you meant "{this}_{other}"?)�this�other)r, �replacer/ � TypeErrorr' r0 )r r2 r r r �__sub___ s ��z UndefinedJinjaVariable.__sub__N)r( r) r* �__doc__r$ r5 r r r r r. Y s r. zRendering basic templatec s � fdd�}t �d|| �S )z�This does simple replacement of bash variable like templates.
It identifies patterns like ${a} or $a and can also identify patterns like
${a.b} or $a.b which will look for a key 'b' in the dictionary rooted
by key 'a'.
c s� | � d�}|d u r| � d�}|d u rtd��t�|�d��}� }t|�dkrE|�� }t|t�s;t d|t
� |�|f ��|| }t|�dks&|�� }t|t�sZt d||t
� |�f ��t || �S )N� r z,Match encountered but no valid group present�.zRCan not traverse into non-dictionary '%s' of type %s while looking for subkey '%s'z<Can not extract key '%s' from non-dictionary '%s' of type %s)
�group� RuntimeError� collections�deque�split�len�popleft�
isinstance�dictr4 �tu�obj_namer, )�matchr �path�selected_params�key��paramsr r �replacerr s2
�� �
�� z basic_render.<locals>.replacerz)\$\{([A-Za-z0-9_.]+)\}|\$([A-Za-z0-9_.]+))�re�sub)�contentrI rJ r rH r � basic_renderi s �rN c C s� dd� }| � d�dkr| �dd�\}}n| }d}t�d|tj�}|s'dt| fS |�d��� �� }|d vr:t d
| ��|d krJt
sJt � d � dt|fS |d krUt
rUd ||fS dt|fS )
Nc
S s� | � d�rdnd}z+t�d�� t| tddgd�jd i |��| W d � W S 1 s-w Y W d S tyM } z
| jd7 _t|d�|�d }~w t y[ } z||�d }~ww )
N�
r% zRendering jinja2 templateTz
jinja2.ext.do)� undefined� trim_blocks�
extensionsr7 )r r )
�endswithr �Timed� JTemplater. �renderr r r � Exception)rM rI �add�template_syntax_error�
unknown_errorr r r � jinja_render� s: �����(�
�����z%detect_template.<locals>.jinja_renderrO ���r7 r% z##\s*template:(.*)�basic)�jinjar] z.Unknown template rendering type '%s' requestedr^ zcJinja not available as the selected renderer for desired template, reverting to the basic renderer.)
�findr= rK rD �IrN r9 �lowerr" �
ValueError�JINJA_AVAILABLE�LOG�warning)�textr[ �ident�rest�
type_match�
template_typer r r �detect_template� s.
�� �
rk c C s4 |si }t t�| ��\}}}t�d| |� |||�S )Nz+Rendering content of '%s' using renderer %s)rk r �load_text_filerd �debug)�fnrI rj �rendererrM r r r �render_from_file� s
rp � c C s t | |�}tj|||d� d S )N)�mode)rp r r )rn �outfnrI rr �contentsr r r �render_to_file� s
ru c C s |si }t | �\}}} || |�S )z
Render string)rk )rM rI �_template_typero r r r �
render_string� s
rw c C sr t �|�}| |d�}t||��� d }|r$t j|dd�}|s$td| ��|dkr0tj�|� d S t ||dd� d S ) N)�variant�prefixrO T)�defaultz.Cannot render template file %s - invalid yaml.�-�w)�omode)
r rl rw �rstrip� load_yamlr: �sys�stdout�writer )rx �template�output�is_yamlry rt �
tpl_params�outr r r �render_template� s
�r� )rq )N)&r; �loggingrK r� �typingr �jinja2r � cloudinitr r rB r �cloudinit.atomic_helperr �__annotations__r �_DebugUndefinedr
rU rc r � ImportError�AttributeError�object� getLoggerr( rd r/ r r. �timedrN rk rp ru rw r� r r r r �<module> s>
�
+
(2