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 deZ$G dd deZ%G dd de
j&Z'd ddZ(d ddZ)e*ee* dddZ+dd f dfd d Z,dS )! N)EACCES)OptionalType)handlers)b64d
json_dumps)Paths)
PER_ALWAYS)MISSING_JINJA_PREFIXdetect_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 ZdgZedddZdd 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_kwargsZhandlerctyper 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 LOGwarningZhandler_version handle_part) r datar! filenamepayloadZ frequencyZheadersZjinja_json_filerendered_payloadsubtypeZ 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 dkrtdi }d}tjj|s8td| ytt|}W nP tk
r } z4d}t |t
t fr~|j t
kr~d| }t||W Y dd}~X nX t| |||}|sdS |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 ospathexistsr r r
Exception
isinstanceIOErrorOSErrorerrnor render_jinja_payload)r)
payload_fnZinstance_data_filedebug
instance_datar* emsgr 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 }|rtjd |d
j
| |S ) Nzbase64-encoded-keysT) decode_pathsinclude_key_aliasesz Converted jinja variables
%sz"Ignoring jinja template for %s: %sc S s g | ]}d |j td qS )z'%s' )replacer
).0varr r r
s z(render_jinja_payload..z%s[^\s]+z:Could not render jinja template variables in file '%s': %sz, )convert_jinja_instance_datar r$ r7 r r TypeErrorJUndefinedErrorr% strrefindallr
join)r) r6 r8 r7 Zinstance_jinja_varsr* r9 warningsr r r r5 s*
r5 ) orig_namereturnc 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 }xt | j D ]\}}|r:dj|||n|}||krNt|}t|trt|||||d||< tjd|rx0|| j D ]\} }
t j
|
|| < qW n|||< |r t |} | r t j
|| || < q W |S )zProcess 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 dqS )-rL )r> )r? r. r r r rA s z/convert_jinja_instance_data..z {0}{1}{2})sepr; r< zv\d+$) sorteditemsformatr r1 dictrB rF matchcopydeepcopyrO ) r' prefixrR r; r< resultkeyvalueZkey_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 ImportErrorcompileZ getLoggerr r$ r r r r r# r5 rE rO rB r r r r s6
*
,