3
\�meQ � @ sX U d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZ d dl m
Z
d dl m Z m
Z
d dlmZ d dlmZ d dlmZmZmZ d d lmZmZ ee y d d
lm Z d d l m Z W n" e k
r� eZej d �Z Y nX ej!e"�Z#G d
d� de�Z$G dd� de�Z%G dd� de
j&�Z'd dd�Z(d dd�Z)e*ee* d�dd�Z+dd f dfd d �Z,dS )!� N)�EACCES)�Optional�Type)�handlers)�b64d�
json_dumps)�Paths)�
PER_ALWAYS)�MISSING_JINJA_PREFIX�detect_template�
render_string)� load_file� load_json)�UndefinedError)� operator_rez[-.]c @ s e Zd ZdS )�JinjaLoadErrorN)�__name__�
__module__� __qualname__� r r �$/usr/lib/python3.6/jinja_template.pyr ! s r c @ s e Zd ZdS )�
NotJinjaErrorN)r r r r r r r r % s r c @ s( e Zd ZdgZed�dd�Zdd� ZdS )�JinjaTemplatePartHandlerz## template: jinja)�pathsc K sT t jj| tdd� || _i | _x0|jdg �D ] }x|j� D ]}|| j|< q:W q,W d S )N� )�version� sub_handlers)r �Handler�__init__r r r �getZ
list_types)�selfr Z_kwargsZhandler�ctyper r r r - s z!JinjaTemplatePartHandler.__init__c C s� |t jkrd S | jjd�}t|||�}|s.d S t j|�} | jj| �}
|
sZtj d|| � d S |
j
dkrz|
j ||||||� n |
j
dkr�|
j |||||� d S )NZinstance_data_sensitivezPIgnoring jinja template for %s. Could not find supported sub-handler for type %sr � ) r ZCONTENT_SIGNALSr Z get_runpath� render_jinja_payload_from_fileZtype_from_starts_withr r �LOG�warningZhandler_version� handle_part) r �datar! �filename�payloadZ frequencyZheadersZjinja_json_file�rendered_payload�subtypeZ sub_handlerr r r r& 5 s*
z$JinjaTemplatePartHandler.handle_partN)r r r �prefixesr r r& r r r r r ) s r Fc C s� t | �d dkrtd��i }d}tjj|�s8td| ��ytt|��}W nP tk
r� } z4d}t |t
t f�r~|j t
kr~d| }t|�|�W Y dd}~X nX t| |||�}|s�dS |S )aR Render a jinja template payload sourcing variables from jinja_vars_path.
@param payload: String of jinja template content. Should begin with
## template: jinja
.
@param payload_fn: String representing the filename from which the payload
was read used in error reporting. Generally in part-handling this is
'part-##'.
@param instance_data_file: A path to a json file containing variables that
will be used as jinja template variables.
@return: A string of jinja-rendered content with the jinja header removed.
Returns None on error.
r Zjinjaz Payload is not a jinja templateNzFCannot render jinja template vars. Instance data not yet present at %sz"Loading Jinja instance data failedzGCannot render jinja template vars. No read permission on '%s'. Try sudo)r r �os�path�existsr r r
� Exception�
isinstance�IOError�OSError�errnor �render_jinja_payload)r) �
payload_fnZinstance_data_file�debug�
instance_datar* �e�msgr r r r# R s,
r# c C s� t ||jdg �dd�}|r*tjdt|�� yt| |�}W n6 ttfk
rn } ztjd|t |�� d S d }~X nX dd� t
j dt |�D �}|r�tjd |d
j
|�� |S ) Nzbase64-encoded-keysT)� decode_paths�include_key_aliasesz Converted jinja variables
%sz"Ignoring jinja template for %s: %sc S s g | ]}d |j td� �qS )z'%s'� )�replacer
)�.0�varr r r �
<listcomp>� s z(render_jinja_payload.<locals>.<listcomp>z%s[^\s]+z:Could not render jinja template variables in file '%s': %sz, )�convert_jinja_instance_datar r$ r7 r r � TypeError�JUndefinedErrorr% �str�re�findallr
�join)r) r6 r8 r7 Zinstance_jinja_varsr* r9 �warningsr r r r5 s*
r5 )� orig_name�returnc C s t jtd| �}|| kr|S dS )a� Return a jinja variable alias, replacing any operators with underscores.
Provide underscore-delimited key aliases to simplify dot-notation
attribute references for keys which contain operators "." or "-".
This provides for simpler short-hand jinja attribute notation
allowing one to avoid quoting keys which contain operators.
{{ ds.v1_0.config.user_network_config }} instead of
{{ ds['v1.0'].config["user.network-config"] }}.
:param orig_name: String representing a jinja variable name to scrub/alias.
:return: A string with any jinja operators replaced if needed. Otherwise,
none if no alias required.
�_N)rF �subr )rJ �
alias_namer r r �get_jinja_variable_alias� s rO r= �/c C s� i }dd� |D �}x�t | j� �D ]�\}}|r:dj|||�n|}||krNt|�}t|t�r�t|||||d�||< tjd|�r�x0|| j� D ]\} }
t j
|
�|| < q�W n|||< |r t |�} | r t j
|| �|| < q W |S )z�Process instance-data.json dict for use in jinja templates.
Replace hyphens with underscores for jinja templates and decode any
base64_encoded_keys.
c S s g | ]}|j d d��qS )�-rL )r> )r? r. r r r rA � s z/convert_jinja_instance_data.<locals>.<listcomp>z {0}{1}{2})�sepr; r< zv\d+$) �sorted�items�formatr r1 �dictrB rF �match�copy�deepcopyrO ) r' �prefixrR r; r< �result�key�valueZkey_pathZsubkeyZsubvaluerN r r r rB � s,
rB )F)F)-rX Zloggingr- rF r4 r Ztypingr r Z cloudinitr Zcloudinit.atomic_helperr r Zcloudinit.helpersr Zcloudinit.settingsr Zcloudinit.templaterr
r r Zcloudinit.utilr
r r0 rD Zjinja2.exceptionsr Z jinja2.lexerr � ImportError�compileZ getLoggerr r$ r r r r r# r5 rE rO rB r r r r �<module> s6
*
,