Current File : //snap/core18/current/usr/lib/python3/dist-packages/cloudinit/cmd/__pycache__/query.cpython-36.pyc
3
�Ad�( � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
mZmZ ddlmZ dZeje�Zdd d
�Zd d � Zed
�dd�Zeeeed�dd�Zdd� Z dd� Z e dkr�e � dS )aT Query 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)�log�util)�addLogHandlerCLI�read_cfg_paths)�convert_jinja_instance_data�get_jinja_variable_alias�render_jinja_payload)�REDACT_SENSITIVE_VALUEZqueryc C s� | st jttd�} | jdddddd� | jdd td
t� jd �� �d � | jd
ddddd� | jddtdd � | jddtdd � | jdtddd� | jddddd d d � | jd d td!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--debug�
store_trueFz+Add verbose messages during template render)�action�default�helpz-iz--instance-dataz,Path to instance-data.json file. Default is �
instance_data)�typer z-lz --list-keyszBList query keys available at the provided instance-data <varname>.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.txt�varname�?z�A 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:". )r �nargsr z-az--all�dump_allz Dump all available instance-data)r
r �destr z-fz--format�formatz�Optionally specify a custom output format string. Any instance-data variable can be specified between double-curly braces. For example -f "{{ v2.cloud_name }}")r r r )�argparse�ArgumentParser�NAME�__doc__� add_argument�strr � get_runpath)�parser� r �5/usr/lib/python3/dist-packages/cloudinit/cmd/query.py�
get_parser$ s` r" c C s> t j| dd�}y
|jd�S tk
r8 t j|ddd�S X dS )z�Attempt 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.
F)�decodezutf-8T)�quietr# N)r � load_filer# �UnicodeDecodeErrorZ decomp_gzip)Z ud_file_pathZbdatar r r! �
load_userdata| s
r' )�returnc C s@ t j� }t� }| r| }nF|jd�}|dkrZ|jd�}t jj|�rF|}q^tjd||� |}n|}|rh|}nt jj|j d�}|r�|} nt jj|j d�} yt
j |�}
W nL t t
fk
r� } z,| jtkr�tjd|� n tjd|� � W Y d d } ~ X nX t
j|
�} |dk�r$d
t|f | d < d
t| f | d < nt|�| d < t| �| d < | S )
a� Return a dict of merged instance-data, vendordata and userdata.
The dict will contain supplemental userdata and vendordata keys sourced
from default user-data and vendor-data files.
Non-root users will have redacted INSTANCE_JSON_FILE content and redacted
vendordata and userdata values.
:raise: IOError/OSError on absence of instance-data.json file or invalid
access perms.
r r Zinstance_data_sensitivez4Missing root-readable %s. Using redacted %s instead.z
user-data.txtzvendor-data.txtz$No read permission on '%s'. Try sudoz Missing instance-data file: %sNz <%s> file:%sZuserdataZ
vendordata)�os�getuidr r �path�exists�LOGZwarning�joinZ
instance_linkr r% �IOError�OSError�errnor �errorZ load_jsonr
r' ) r � user_data� vendor_dataZuid�pathsZinstance_data_fnZredacted_data_fnZsensitive_data_fnZ user_data_fnZvendor_data_fnZ
instance_json�er r r! �_read_instance_data� sN
r7 )�jinja_vars_without_aliases�jinja_vars_with_aliasesr � list_keysc
C s� d}| }x�|j d�D ]�}y || }W nJ tk
rn } z.|rJdj||d�}n
dj|�}t|�|�W Y dd}~X nX ||kr�|| }n$x"|D ]} t| �|kr�|| }P q�W |r�|d7 }||7 }qW |S )a� Return the value of the dot-delimited varname path in instance-data
Split a dot-delimited jinja variable name path into components, walk the
path components into the instance_data and look up a matching jinja
variable name or cloud-init's underscore-delimited key aliases.
:raises: ValueError when varname represents an invalid key name or path or
if list-keys is provided by varname isn't a dict object.
� �.z*instance-data '{key_path}' has no '{leaf}')ZleafZkey_pathz Undefined instance-data key '{}'N)�split�KeyErrorr �
ValueErrorr )
r8 r9 r r: Zwalked_key_path�responseZ
key_path_partr6 �msg�keyr r r! �(_find_instance_data_leaf_by_varname_path� s*
rC c C sz t t|jrtjntj� t|j|j|j |j
g�sHtj d� t � j
� dS yt|j|j|j�}W n ttfk
rv dS X |j r�dj |j d�}t|d||jr�dndd�}|r�t|� d S dS t|�}|j�r t|dd
�}yt|||j|jd �}W n0 ttfk
�r } ztj |� dS d }~X nX |j�rXt|t��sFtj d
|j� dS dj t |j � ��}t|t ��snt j!|�}t|� d S )z3Handle calls to 'cloud-init query' as a subcommand.zDExpected one of the options: --all, --format, --list-keys or varname� z## template: jinja
{fmt})Zfmtzquery commandlineTF)�payloadZ
payload_fnr �debugr )Zinclude_key_aliases)r8 r9 r r: Nz+--list-keys provided but '%s' is not a dict�
)"r r- rF r �DEBUGZWARNING�anyr: r r r r2 r" Z
print_helpr7 r r3 r4 r/ r0 r �printr rC r> r? �
isinstance�dictr. �sorted�keysr r Z
json_dumps)�name�argsr rE Zrendered_payloadr@ r9 r6 r r r! � handle_args� sZ
rQ c C s t � } tjtt| j� �� dS )z,Tool to query specific instance-data values.N)r" �sys�exitrQ r �
parse_args)r r r r! �main4 s rU �__main__)N) r r r) rR r1 r Z cloudinitr r Zcloudinit.cmd.develr r Z!cloudinit.handlers.jinja_templater r r Zcloudinit.sourcesr
r Z getLoggerr- r" r' rL r7 r �boolrC rQ rU �__name__r r r r! �<module> s*
XB'=
Mini Shell