File: //lib/python3/dist-packages/cloudinit/handlers/__pycache__/jinja_template.cpython-38.pyc
U
Ӈg� � @ 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 dlmZmZmZmZ d d lmZmZ ee ed
< z d d l m 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)�b64d�
json_dumps)�Paths)�
PER_ALWAYS)�MISSING_JINJA_PREFIX�JinjaSyntaxParsingException�detect_template�
render_string)� load_json�load_text_file�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�}zt|||�}W n: tk
rd } z t�d|t| �� W Y �d S d } ~ X Y nX |snd S t � |�}
| j
� |
�} | s�t�d||
� d S | j dkr�| �
||||||� n | j dkr�| �
|||||� d S )NZinstance_data_sensitivez=Ignoring jinja template for %s. Failed to render template. %szPIgnoring jinja template for %s. Could not find supported sub-handler for type %sr � )r ZCONTENT_SIGNALSr Z get_runpath� render_jinja_payload_from_filer �LOG�warning�strZtype_from_starts_withr r" Zhandler_version� handle_part) r# �datar$ �filename�payloadZ frequencyZheadersZjinja_json_file�rendered_payload�eZsubtypeZ sub_handlerr r r r* 6 sX
��
�
�
�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 )aK Render a jinja template sourcing variables from jinja_vars_path.
@param payload: String of jinja template content. Should begin with
## template: jinja\n.
@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. r/ �msgr r r r& ] 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' r; r r
� TypeErrorr r( r) �re�findallr
�join)r- r: r< r; Zinstance_jinja_varsr. r/ �warningsr r r r9 � s8
� � ���r9 )� 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 )rA )rB r2 r r r rD � s z/convert_jinja_instance_data.<locals>.<listcomp>z {0}{1}{2})�sepr> r? zv\d+$) �sorted�items�formatr r5 �dictrE rG �match�copy�deepcopyrP ) r+ �prefixrS r> r? �result�key�valueZkey_pathZsubkeyZsubvaluerO r r r rE � s.
�
rE )F)F)r@ rQ r F)/rY Zloggingr1 rG r8 r �typingr r Z cloudinitr Zcloudinit.atomic_helperr r Zcloudinit.helpersr Zcloudinit.settingsr Zcloudinit.templaterr
r r r
Zcloudinit.utilr r r4 �__annotations__Zjinja2.exceptionsr r Z jinja2.lexerr � ImportError�compileZ getLoggerr r' r r r r r& r9 r) rP rE r r r r �<module> s@
4 �
-
�