o w7eg@s\UddlZddlZddlZddlmZddlmZmZddlm Z ddlm Z ddl m Z mZddlmZddlmZdd lmZmZmZdd lmZmZeeed <zdd lmZdd lm Z Wne!yqeZe"dZ Ynwe #e$Z%GdddeZ&GdddeZ'Gddde j(Z) d#ddZ*d#ddZ+de,dee,fddZ- d$d!d"Z.dS)%N)EACCES)OptionalType)handlers)log)b64d json_dumps)Paths) PER_ALWAYS)MISSING_JINJA_PREFIXdetect_template render_string) load_file load_jsonJUndefinedError)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}t|||}|sdSt|} |j| } | s-t d|| dS| j dkr>| ||||||dS| j dkrN| |||||dSdS)Ninstance_data_sensitivezPIgnoring jinja template for %s. Could not find supported sub-handler for type %sr) rCONTENT_SIGNALSr get_runpathrender_jinja_payload_from_filetype_from_starts_withr!r$LOGwarninghandler_version handle_part) r&datar)filenamepayload frequencyheadersjinja_json_filerendered_payloadsubtype sub_handlerrrrr35s4        z$JinjaTemplatePartHandler.handle_partN)rrrprefixesr r#r3rrrrr)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)aRRender 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. 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)r6 payload_fninstance_data_filedebug instance_datar:emsgrrrr.Rs:    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$r0rJrr TypeErrorrr1strrefindallr join)r6rHrKrJinstance_jinja_varsr:rLwarningsrrrrGs8 rG 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_aliassrbrP/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_)rQ)rRr@rrrrTsz/convert_jinja_instance_data..z {0}{1}{2})seprNrOzv\d+$) sorteditemsformatrrCdictrUrXmatchcopydeepcopyrb) r4prefixrerNrOresultkeyvaluekey_pathsubkeysubvaluerarrrrUs2  rU)F)rPrcrF)/rkr?rXrFrtypingrr cloudinitrrloggingcloudinit.atomic_helperrrcloudinit.helpersr cloudinit.settingsr cloudinit.templaterr r r cloudinit.utilrrrB__annotations__jinja2.exceptionsrr jinja2.lexerr ImportErrorcompile getLoggerrr0rrr"rr.rGrWrbrUrrrrs<          * -