U Ldg@sXUddlZddlZddlZddlmZddlmZmZddlm Z ddlm Z ddl m Z mZddlmZddlmZdd lmZmZmZdd lmZmZeeed <zdd lmZdd lm Z Wn"e!k reZe"dZ YnXe #e$Z%GdddeZ&GdddeZ'Gddde j(Z)d"ddZ*d#ddZ+e,ee,dddZ-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@s eZdZdS)JinjaLoadErrorN__name__ __module__ __qualname__rrC/usr/lib/python3/dist-packages/cloudinit/handlers/jinja_template.pyr!src@s eZdZdS) NotJinjaErrorNrrrrrr%src@s(eZdZdgZedddZddZdS)JinjaTemplatePartHandlerz## template: jinja)pathscKsLtjj|tdd||_i|_|dgD]}|D]}||j|<q6q*dS)N)version sub_handlers)rHandler__init__r rrgetZ list_types)selfrZ_kwargsZhandlerctyperrrr!-s  z!JinjaTemplatePartHandler.__init__c Cs|tjkrdS|jd}t|||}|s.dSt|} |j| } | sZt d|| dS| j dkrz| ||||||n| j dkr| |||||dS)NZinstance_data_sensitivezPIgnoring jinja template for %s. Could not find supported sub-handler for type %sr) rZCONTENT_SIGNALSrZ get_runpathrender_jinja_payload_from_fileZtype_from_starts_withrr"LOGwarningZhandler_version handle_part) r#datar$filenamepayloadZ frequencyZheadersZjinja_json_filerendered_payloadZsubtypeZ sub_handlerrrrr)5sH      z$JinjaTemplatePartHandler.handle_partN)rrrprefixesr r!r)rrrrr)srFc Cst|ddkrtdi}d}tj|s8td|ztt|}WnPtk r}z2d}t |t t fr~|j t kr~d|}t||W5d}~XYnXt||||}|sdS|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. rZjinjazPayload 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)r, payload_fnZinstance_data_filedebug instance_datar-emsgrrrr&Rs<  r&c Cst||dgdd}|r*tdt|zt||}Wn>ttfk rv}ztd|t |WYdSd}~XYnXddt dt |D}|rtd |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'r9rr TypeErrorrr(strrefindallr join)r,r8r:r9Zinstance_jinja_varsr-r;warningsrrrr7s8 r7) 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)rGsubr)rK alias_namerrrget_jinja_variable_aliassrPr?/rc Csi}dd|D}t|D]\}}|r8d|||n|}||krLt|}t|trt|||||d||<td|r||D]\} } t | || <qn|||<|rt |} | rt |||| <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)-rM)r@)rAr0rrrrCsz/convert_jinja_instance_data..z {0}{1}{2})sepr=r>zv\d+$) sorteditemsformatrr3dictrDrGmatchcopydeepcopyrP) r*prefixrSr=r>resultkeyvalueZkey_pathZsubkeyZsubvaluerOrrrrDs.   rD)F)F)r?rQrF)/rYr/rGr6rtypingrrZ cloudinitrrZloggingZcloudinit.atomic_helperrrZcloudinit.helpersr Zcloudinit.settingsr Zcloudinit.templaterr r r Zcloudinit.utilrrr2__annotations__Zjinja2.exceptionsrrZ jinja2.lexerr ImportErrorcompileZ getLoggerrr'rrr rr&r7rFrPrDrrrrs@        * -