o x[h@s\UddlZddlZddlZddlZddlmZddlmZmZddl m Z ddl m Z m Z ddlmZddlmZddlmZmZmZmZdd lmZmZeeed <zdd lmZdd lm Z Wne!yqeZe"d Z Ynwe#e$Z%GdddeZ&GdddeZ'Gddde j(Z) d"ddZ*d"ddZ+de,dee,fddZ- d#d d!Z.dS)$N)EACCES)OptionalType)handlers)b64d json_dumps)Paths) PER_ALWAYS)MISSING_JINJA_PREFIXJinjaSyntaxParsingExceptiondetect_template render_string) load_jsonload_text_fileJUndefinedError)UndefinedError) operator_rez[-.]c@ eZdZdS)JinjaLoadErrorN__name__ __module__ __qualname__rrC/usr/lib/python3/dist-packages/cloudinit/handlers/jinja_template.pyr"rc@r) NotJinjaErrorNrrrrrr&rrc@s(eZdZdgZdefddZddZdS)JinjaTemplatePartHandlerz## template: jinjapathscKsLtjj|tdd||_i|_|dgD]}|D]}||j|<qqdS)N)version sub_handlers)rHandler__init__r rr!get list_types)selfr_kwargshandlerctyperrrr#.s  z!JinjaTemplatePartHandler.__init__c Cs|tjvrdS|jd}zt|||}Wnty1} ztd|t| WYd} ~ dSd} ~ ww|s6dSt |} |j | } | sLtd|| dS| j dkr]| ||||||dS| j dkrm| |||||dSdS)Ninstance_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)rCONTENT_SIGNALSr get_runpathrender_jinja_payload_from_filer LOGwarningstrtype_from_starts_withr!r$handler_version handle_part) r&datar)filenamepayload frequencyheadersjinja_json_filerendered_payloadesubtype sub_handlerrrrr46sH         z$JinjaTemplatePartHandler.handle_partN)rrrprefixesrr#r4rrrrr*s rFc Cst|ddkr tdi}d}tj|std|ztt|}Wn#tyG}zd}t |t t fr>|j t kr>d|}t||d}~wwt||||}|sSdS|S)aKRender 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. rjinjazPayload 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 rospathexistsrrr Exception isinstanceIOErrorOSErrorerrnorrender_jinja_payload)r7 payload_fninstance_data_filedebug instance_datar;r<msgrrrr.]s:    r.c Cst||dgdd}|rtdt|zt||}Wnttfy:}ztd|t |WYd}~dSd}~wwddt dt |D}|rTtd |d ||S) Nzbase64-encoded-keysT) decode_pathsinclude_key_aliaseszConverted jinja variables %sz"Ignoring jinja template for %s: %scSsg|] }d|tdqS)z'%s')replacer ).0varrrr sz(render_jinja_payload..z%s[^\s]+z:Could not render jinja template variables in file '%s': %sz, )convert_jinja_instance_datar$r/rLrr TypeErrorrr0r1refindallr join)r7rJrMrLinstance_jinja_varsr;r<warningsrrrrIs8 rI orig_namereturncCsttd|}||kr |SdS)aReturn 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)rXsubr)r] alias_namerrrget_jinja_variable_aliassrbrQ/rc Csi}dd|D}t|D]U\}}|rd|||n|}||vr&t|}t|trOt|||||d||<td|rN||D] \} } t | || <qBn|||<|rdt |} | rdt |||| <q|S)zProcess instance-data.json dict for use in jinja templates. Replace hyphens with underscores for jinja templates and decode any base64_encoded_keys. cSsg|]}|ddqS)-r_)rR)rSrBrrrrUsz/convert_jinja_instance_data..z {0}{1}{2})seprOrPzv\d+$) sorteditemsformatrrEdictrVrXmatchcopydeepcopyrb) r5prefixrerOrPresultkeyvaluekey_pathsubkeysubvaluerarrrrVs2  rV)F)rQrcrF)/rkloggingrArXrHrtypingrr cloudinitrcloudinit.atomic_helperrrcloudinit.helpersrcloudinit.settingsr cloudinit.templaterr r r r cloudinit.utilrrrD__annotations__jinja2.exceptionsrr jinja2.lexerr ImportErrorcompile getLoggerrr/rrr"rr.rIr1rbrVrrrrs<         4 -