Your IP : 216.73.216.52
U
�Ad�( � @ s� d Z ddlZddlZddlZddlmZ ddlmZmZ ddl m
Z
m Z ddl m
Z
mZmZ ddlmZ dZe�e�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� �d �� �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$ sp �� �
� � � �� r# c C sD t j| dd�}z |�d�W S tk
r> t j|ddd� Y 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 �� }t� }| r| }nF|�d�}|dkrZ|�d�}t j�|�rF|}q^t�d||� |}n|}|rh|}nt j�|j d�}|r�|} nt j�|j d�} zt
� |�}
W nL t t
fk
r� } z*| jtkr�t�d|� n t�d|� � W 5 d } ~ X Y nX t
�|
�} |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� sT
�
�
� r8 ��jinja_vars_without_aliases�jinja_vars_with_aliasesr � list_keysc
C s� d}| }|� d�D ]�}z || }W nJ tk
rl } z,|rHdj||d�}n
d�|�}t|�|�W 5 d}~X Y nX ||kr�|| }n"|D ] } t| �|kr�|| } q�q�|r�|d7 }||7 }q|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 )
r: r; r r< Zwalked_key_path�responseZ
key_path_partr7 �msg�keyr! r! r" �(_find_instance_data_leaf_by_varname_path� s. �
rE c
C s� t t|jrtjntj� t|j|j|j |j
g�sHt� d� t � �
� dS zt|j|j|j�}W n ttfk
rx Y dS X |j r�dj |j d�}t|d||jr�dndd�}|r�t|� d S dS t|�}|j�r*t|dd
�}zt|||j|jd �}W n8 ttfk
�r( } zt� |� W Y �dS d }~X Y nX |j�rbt|t��sPt� d
|j� dS d� t |� � ��}t|t ��sxt �!|�}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_aliasesr9 Nz+--list-keys provided but '%s' is not a dict�
)"r r. rH r �DEBUGZWARNING�anyr< r r r r3 r# Z
print_helpr8 r r4 r5 r0 r1 r �printr rE r@ rA �
isinstance�dictr/ �sorted�keysr r Z
json_dumps)�name�argsr rG Zrendered_payloadrB r; r7 r! r! r" � handle_args� sn �
� � ��
�
rS c C s t � } t�tt| �� �� dS )z,Tool to query specific instance-data values.N)r# �sys�exitrS r �
parse_argsr r! r! r" �main4 s rW �__main__)N) r r r* rT r2 r Z cloudinitr r Zcloudinit.cmd.develr r Z!cloudinit.handlers.jinja_templater r r Zcloudinit.sourcesr
r Z getLoggerr. r# r( rN r8 r �boolrE rS rW �__name__r! r! r! r" �<module> s,
XB� +=