← Back
Editing: 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 � - �
Save File
Cancel