File: //lib/python3/dist-packages/cloudinit/handlers/__pycache__/jinja_template.cpython-310.pyc
o
�AdD � @ sT 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
< zd d lm Z d d l m Z W n e ym eZ e�!d
�Z Y nw e �"e#�Z$G dd� de�Z%G dd� de�Z&G dd� de j'�Z( d"dd�Z)d"dd�Z*de+dee+ fdd �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 @ � e Zd ZdS )�JinjaLoadErrorN��__name__�
__module__� __qualname__� r r �C/usr/lib/python3/dist-packages/cloudinit/handlers/jinja_template.pyr � r c @ r )�
NotJinjaErrorNr r r r r r $ r r c @ s( e Zd ZdgZdefdd�Zdd� ZdS )�JinjaTemplatePartHandlerz## template: jinja�pathsc K sL t jj| tdd� || _i | _|�dg �D ]}|�� D ]}|| j|< qqd S )N� )�version� sub_handlers)r �Handler�__init__r r r! �get�
list_types)�selfr �_kwargs�handler�ctyper r r r# , s ��z!JinjaTemplatePartHandler.__init__c C s� |t jv rd S | j�d�}t|||�}|sd S t �|�} | j�| �}
|
s-t� d|| � d S |
j
dkr>|
� ||||||� d S |
j
dkrN|
� |||||� d S d S )N�instance_data_sensitivezPIgnoring jinja template for %s. Could not find supported sub-handler for type %sr � ) r �CONTENT_SIGNALSr � get_runpath� render_jinja_payload_from_file�type_from_starts_withr! r$ �LOG�warning�handler_version� handle_part) r& �datar) �filename�payload� frequency�headers�jinja_json_file�rendered_payload�subtype� sub_handlerr r r r3 4 s4
�
�
�
��z$JinjaTemplatePartHandler.handle_partN)r r r �prefixesr r# r3 r r r r r ( s r Fc
C s� t | �d dkr td��i }d}tj�|�s td| ��ztt|��}W n# tyG } zd}t |t
t f�r>|j t
kr>d| }t|�|�d}~ww t| |||�}|sSdS |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 �jinjaz 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)r6 �
payload_fn�instance_data_file�debug�
instance_datar: �e�msgr r r r. Q s: ��
��
��
�r. c
C s� t ||�dg �dd�}|rt�dt|�� zt| |�}W n ttfy: } zt�d|t |�� W Y d }~d S d }~ww dd� t
� dt |�D �}|rTt�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$ r0 rJ r
r � TypeErrorr r1 �str�re�findallr �join)r6 rH rK rJ �instance_jinja_varsr: rL �warningsr r r rG ~ s8
�������rG � 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)rX �subr )r] �
alias_namer r r �get_jinja_variable_alias� s rb rP �/r c C s� i }dd� |D �}t | �� �D ]U\}}|r d�|||�n|}||v r&t|�}t|t�rOt|||||d�||< t�d|�rN|| �� D ] \} }
t �
|
�|| < qBn|||< |rdt |�} | rdt �
|| �|| < 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 )�-r_ )rQ )rR r@ r r r rT � s z/convert_jinja_instance_data.<locals>.<listcomp>z {0}{1}{2})�seprN rO zv\d+$) �sorted�items�formatr rC �dictrU rX �match�copy�deepcopyrb ) r4 �prefixre rN rO �result�key�value�key_path�subkey�subvaluera r r r rU � s2
� ��rU )F)rP rc r F).rk r? rX rF r �typingr r � cloudinitr r �logging�cloudinit.helpersr �cloudinit.settingsr �cloudinit.templaterr r
r �cloudinit.utilr r
r r rB �__annotations__�jinja2.exceptionsr r � jinja2.lexerr � ImportError�compile� getLoggerr r0 r r r" r r. rG rW rb rU r r r r �<module> s:
�
*
�
- �