Your IP : 216.73.217.13
U
�AdD � @ sP U d dl Z d dlZd dlZd dlmZ d dlmZmZ d dlm Z d dlm
Z d dl m
Z
d dlmZ d dlmZmZmZ d d lmZmZmZmZ ee ed
< z d d lm Z d d l m Z W n" e k
r� eZ e�!d
�Z Y nX e �"e#�Z$G dd� de�Z%G dd� de�Z&G dd� de j'�Z(d!dd�Z)d"dd�Z*e+ee+ d�dd�Z,d#d d �Z-dS )$� N)�EACCES)�Optional�Type)�handlers)�log)�Paths)�
PER_ALWAYS)�MISSING_JINJA_PREFIX�detect_template�
render_string)�b64d�
json_dumps� load_file� load_json�JUndefinedError)�UndefinedError)� operator_rez[-.]c @ s e Zd ZdS )�JinjaLoadErrorN��__name__�
__module__� __qualname__� r r �C/usr/lib/python3/dist-packages/cloudinit/handlers/jinja_template.pyr s r c @ s e Zd ZdS )�
NotJinjaErrorNr r r r r r $ s r c @ s( e Zd ZdgZed�dd�Zdd� ZdS )�JinjaTemplatePartHandlerz## template: jinja)�pathsc K sL t jj| tdd� || _i | _|�dg �D ] }|�� D ]}|| j|< q6q*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 | j�d�}t|||�}|s.d S t �|�} | j�| �}
|
sZt� d|| � d S |
j
dkrz|
� ||||||� n |
j
dkr�|
� |||||� 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_payloadZsubtypeZ sub_handlerr r r r) 4 sH
�
�
�
�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}tj�|�s8td| ��ztt|��}W nP tk
r� } z2d}t |t
t f�r~|j t
kr~d| }t|�|�W 5 d}~X Y 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& Q s< ��
�� �r& c
C s� t ||�dg �dd�}|r*t�dt|�� zt| |�}W n> ttfk
rv } z t�d|t |�� W Y �d S d }~X Y nX dd� t
� dt |�D �}|r�t�d |d
�
|�� |S ) Nzbase64-encoded-keysT)� decode_paths�include_key_aliasesz Converted jinja variables
%sz"Ignoring jinja template for %s: %sc S s g | ]}d |� 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' r9 r
r � TypeErrorr r( �str�re�findallr �join)r, r8 r: r9 Zinstance_jinja_varsr- r; �warningsr r r r7 ~ s8
� � ���r7 )� orig_name�returnc C s t �td| �}|| 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)rG �subr )rK �
alias_namer r r �get_jinja_variable_alias� s rP r? �/r c C s� i }dd� |D �}t | �� �D ]�\}}|r8d�|||�n|}||krLt|�}t|t�r�t|||||d�||< t�d|�r�|| �� D ]\} }
t �
|
�|| < q�n|||< |r t |�} | r t �
|| �|| < q |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 | ]}|� d d��qS )�-rM )r@ )rA r0 r r r rC � s z/convert_jinja_instance_data.<locals>.<listcomp>z {0}{1}{2})�sepr= r> zv\d+$) �sorted�items�formatr r3 �dictrD rG �match�copy�deepcopyrP ) r* �prefixrS r= r> �result�key�valueZkey_pathZsubkeyZsubvaluerO r r r rD � s.
�
rD )F)F)r? rQ r F).rY r/ rG r6 r �typingr r Z cloudinitr r ZloggingZcloudinit.helpersr Zcloudinit.settingsr Zcloudinit.templaterr r
r Zcloudinit.utilr r
r r r2 �__annotations__Zjinja2.exceptionsr r Z jinja2.lexerr � ImportError�compileZ getLoggerr r' r r r r r& r7 rF rP rD r r r r �<module> s>
* �
-
�