3
\�me� � @ s
U d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm Z d dl
m Z e y$d dl
mZ d dl
mZ dZeZ W n eefk
r� d ZeZ Y nX eje�Zejd
ej�Zejd �Z d Z G d
d� de �Z dd� Z dd� Z dd� Z!d dd�Z"dd� Z#d dd�Z$dS ) � N)�Any)�
type_utils)�util)�
write_file)�DebugUndefined)�TemplateTFz##\s*template:(.*)z)\$\{([A-Za-z0-9_.]+)\}|\$([A-Za-z0-9_.]+)zCI_MISSING_JINJA_VAR/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_name)�self� r � /usr/lib/python3.6/templater.py�__str__2 s z UndefinedJinjaVariable.__str__c C s( t |�jtd�}tdj| j|d���d S )N� zhUndefined jinja variable: "{this}-{other}". Jinja tried subtraction. Perhaps you meant "{this}_{other}"?)�this�other)�str�replacer � TypeError�formatr
)r r r r r
�__sub__5 s z UndefinedJinjaVariable.__sub__N)�__name__�
__module__� __qualname__�__doc__r r r r r r
r / s r c s � fdd�}t j|| �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� | j d�}|d kr | j d�}|d kr,td��tj|jd��}� }xBt|�dkr�|j� }t|t�sxt d|t
j |�|f ��|| }qBW |j� }t|t�s�t d||t
j |�f ��t || �S )N� � 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�dictr �tuZobj_namer )�match�name�pathZselected_params�key)�paramsr r
�replacerG s(
z basic_render.<locals>.replacer)�
BASIC_MATCHER�sub)�contentr, r- r )r, r
� basic_render? s r1 c C s� dd� }| j d�d
kr(| jdd�\}}n| }d}tj|�}|sHdt| fS |jd�j� j� }|d krntd| ��|dkr�t r�t
j d � dt|fS |dkr�t r�d||fS dt|fS d S ) Nc S s0 | j d�rdnd}t| tddgd�jf |�| S )N�
r Tz
jinja2.ext.do)Z undefinedZ trim_blocks�
extensions)�endswith� JTemplater Zrender)r0 r, �addr r r
� jinja_renderf s z%detect_template.<locals>.jinja_renderr2 r r �basic�jinjaz.Unknown template rendering type '%s' requestedzcJinja not available as the selected renderer for desired template, reverting to the basic renderer.���)r9 r8 ) �findr" � TYPE_MATCHERr( r1 r �lower�strip�
ValueError�JINJA_AVAILABLE�LOGZwarning)�textr7 Zident�restZ
type_match�
template_typer r r
�detect_templatee s(
rE c C s4 |si }t tj| ��\}}}tjd| |� |||�S )Nz+Rendering content of '%s' using renderer %s)rE r Z load_filerA �debug)�fnr, rD �rendererr0 r r r
�render_from_file� s
rI � c C s t | |�}tj|||d� d S )N)�mode)rI r r )rG Zoutfnr, rK �contentsr r r
�render_to_file� s
rM c C s |si }t | �\}}} || |�S )z
Render string)rE )r0 r, Z_template_typerH r r r
�
render_string� s rN c C s� t |d��}|j� }W d Q R X | |d�}t||�j� d }|r\tj|dd�}|s\td| ��|dkrrtjj |� nt
||dd � d S )
N�r)�variant�prefixr2 T)�defaultz.Cannot render template file %s - invalid yaml.�-�w)Zomode) �open�readrN �rstripr Z load_yamlr �sys�stdout�writer ) rP �template�outputZis_yamlrQ ZfhrL Z
tpl_params�outr r r
�render_template� s
r^ )rJ )N)%r Zlogging�rerX Ztypingr Z cloudinitr r' r Zcloudinit.atomic_helperr Z
JUndefinedZjinja2r Z_DebugUndefinedr r5 r@ � ImportError�AttributeError�objectZ getLoggerr rA �compile�Ir< r. r r r1 rE rI rM rN r^ r r r r
�<module> s8
&)