"T`@sdZddlZddlmZddlZddlZddlmZmZddl m Z m Z ddl m Z ddlmZmZmZddl mZd Ze jeZdd d Zd d ZddZddZedkredS)aTQuery standardized instance metadata provided to machine, returning a JSON structure. Some instance-data values may be binary on some platforms, such as userdata and vendordata. Attempt to decompress and decode UTF-8 any binary values. Any binary values in the instance metadata will be base64-encoded and prefixed with "ci-b64:" in the output. userdata and, where applicable, vendordata may be provided to the machine gzip-compressed (and therefore as binary data). query will attempt to decompress these to a string before emitting the JSON output; if this fails, they are treated as binary. N)EACCES)convert_jinja_instance_datarender_jinja_payload)addLogHandlerCLIread_cfg_paths)log)INSTANCE_JSON_FILEINSTANCE_JSON_SENSITIVE_FILEREDACT_SENSITIVE_VALUE)utilZqueryc Cs'|stjdtdt}|jddddddd d |jd d d td dt|jddddddd d|jddd td d|jddd td d|jdd tddd d|jddddddddd d |jd!d"d tdd#d d$|S)%a#Build or extend an arg parser for query utility. @param parser: Optional existing ArgumentParser instance representing the query subcommand which will be extended to support the args of this utility. @returns: ArgumentParser with proper argument configuration. progZ descriptionz-dz--debugaction store_truedefaultFhelpz+Add verbose messages during template renderz-iz--instance-datatypez>Path to instance-data.json file. Default is /run/cloud-init/%sz-lz --list-keyszBList query keys available at the provided instance-data .z-uz --user-datazHPath to user-data file. Default is /var/lib/cloud/instance/user-data.txtz-vz --vendor-datazLPath to vendor-data file. Default is /var/lib/cloud/instance/vendor-data.txtvarnamenargs?zA dot-delimited specific variable to query from instance-data. For example: v1.local_hostname. If the value is not JSON serializable, it will be base64-encoded and will contain the prefix "ci-b64:". z-az--alldestdump_allz Dump all available instance-dataz-fz--formatformatzOptionally specify a custom output format string. Any instance-data variable can be specified between double-curly braces. For example -f "{{ v2.cloud_name }}")argparseArgumentParserNAME__doc__ add_argumentstrr)parserr5/usr/lib/python3/dist-packages/cloudinit/cmd/query.py get_parser!s:  r!c CsXtj|dd}y|jdSWn+tk rStj|ddddSYnXdS)zAttempt to return a string of user-data from ud_file_path Attempt to decode or decompress if needed. If unable to decode the content, raw bytes will be returned. @returns: String of uncompressed userdata if possible, otherwise bytes. decodeFzutf-8quietTN)r load_filer"UnicodeDecodeErrorZ decomp_gzip)Z ud_file_pathZbdatarrr load_userdataQs  r&cCsd}tt|jrtjntjt|j|j|j |j gsjtj dt j dStj}t|j|j|jgst}|jr|j}nytjj|jt}|dkr%tjj|jt}tjj|r |}q+tjd|||}n|}|jr@|j}ntjj|jd}|jrm|j}ntjj|jd}ytj|} Wnatt fk r} z;| j!t"krtj d|ntj d |dSWYdd} ~ XnXtj#| } |dkrBd t$|f| d file:%sZuserdataZ vendordataz## template: jinja {fmt}ZfmtpayloadZ payload_fnzquery commandline instance_datadebugTF.zUndefined instance-data key %sz+--list-keys provided but '%s' is not a dict )1rLOGr*rDEBUGZWARNINGanyZ list_keysrrrerrorr!Z print_helposgetuidallr)Z user_dataZ vendor_datarpathjoinZrun_dirrr existsZwarningZ instance_linkr r$IOErrorOSErrorerrnorZ load_jsonr r&rprintrsplitKeyError isinstancedictsortedkeysrZ json_dumps)nameargspathsZuidZinstance_data_fnZredacted_data_fnZsensitive_data_fnZ user_data_fnZvendor_data_fnZ instance_jsoner)r(Zrendered_payloadZresponsevarrrr handle_args`s"$                         rFcCs)t}tjtt|jdS)z,Tool to query specific instance-data values.N)r!sysexitrFr parse_args)rrrr mains rJ__main__)rrr9rr1rGZ!cloudinit.handlers.jinja_templaterrZcloudinit.cmd.develrrZ cloudinitrZcloudinit.sourcesrr r r rZ getLoggerr-r!r&rFrJ__name__rrrr s"   0  Z