www.eklavya.in - Cubjrnet7

upload mass deface mass delete console info server
name : query.cpython-39.pyc
a

'�Dgk,�@s�dZddlZddlZddlZddlZddlmZddlmZm	Z	ddl
mZddlm
Z
mZmZddlmZddlmZd	Ze�e�Zdd
d�Zdd
�Zed�dd�Zeeeed�dd�Zdd�Zdd�Z edkr�e �dS)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)�
atomic_helper�util)�read_cfg_paths)�convert_jinja_instance_data�get_jinja_variable_alias�render_jinja_payload)�REDACT_SENSITIVE_VALUE)�JinjaSyntaxParsingException�querycCs�|stjttd�}|jdddddd�|jdd	td
t��d���d�|jd
ddddd�|jddtdd�|jddtdd�|jdtddd�|jddddddd�|jdd 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.
    )�prog�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)�typerz-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�nargsrz-az--all�dump_allz Dump all available instance-data)rr�destrz-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 }}")rrr)�argparse�ArgumentParser�NAME�__doc__�add_argument�strr�get_runpath��parser�r#�7/usr/lib/python3.9/site-packages/cloudinit/cmd/query.py�
get_parser&st	�
��	�
�	�	���r%cCsBtj|dd�}z|�d�WSty<tj|ddd�YS0dS)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.
    T)�quietzutf-8F)r&�decodeN)rZload_binary_filer'�UnicodeDecodeErrorZdecomp_gzip)Zud_file_pathZbdatar#r#r$�
load_userdata~s
r))�returnc
Cs�t��}t�}|r|}nF|�d�}|dkrZ|�d�}tj�|�rF|}q^t�d||�|}n|}|rh|}ntj�|j	d�}|r�|}	ntj�|j	d�}	|�d�}
zt
�|�}WnNtt
fy�}z2|jtkr�t�d|�nt�d	|��WYd
}~n
d
}~00t
�|�}zt
�t
�|
��}
Wntt
f�y4d
}
Yn0|dk�rrdt|f|d<dt|	f|d
<dt|
f|d<n t|�|d<t|	�|d
<|
|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.
    rrZinstance_data_sensitivez4Missing root-readable %s. Using redacted %s instead.z
user-data.txtzvendor-data.txt�combined_cloud_configz$No read permission on '%s'. Try sudozMissing instance-data file: %sNz<%s> file:%sZuserdataZ
vendordata)�os�getuidrr �path�exists�LOGZwarning�joinZ
instance_linkrZload_text_file�IOError�OSError�errnor�errorZ	load_jsonr	r))r�	user_data�vendor_data�uid�pathsZinstance_data_fnZredacted_data_fnZsensitive_data_fnZuser_data_fnZvendor_data_fnZcombined_cloud_config_fnZ
instance_json�er+r#r#r$�_read_instance_data�sl

�


�

�
�
�r;��jinja_vars_without_aliases�jinja_vars_with_aliasesr�	list_keysc
Cs�d}|}|�d�D]�}z||}WnLtyn}z4|rFdj||d�}n
d�|�}t|�|�WYd}~n
d}~00||vr�||}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>rr?Zwalked_key_path�responseZ
key_path_partr:�msg�keyr#r#r$�(_find_instance_data_leaf_by_varname_path�s,�
 

rHc
Cs�t|j|j|j|jg�s0t�d�t���dSzt	|j
|j|j�}Wnt
tfy^YdS0|jr�dj|jd�}zt|d||jr�dndd�}Wn6ty�}zt�d	t|��WYd
}~dSd
}~00|r�t|�dSdSt|�}|j�rJt|dd�}zt|||j|jd
�}Wn6ttf�yH}zt�|�WYd
}~dSd
}~00|j�r�t|t��spt�d|j�dSd�t|����}t|t��s�t�|�}t|�dS)z3Handle calls to 'cloud-init query' as a subcommand.zDExpected one of the options: --all, --format, --list-keys or varname�z## template: jinja
{fmt})�fmtzquery command lineTF)�payloadZ
payload_fnr�debugz#Failed to render templated data. %sNr)Zinclude_key_aliasesr<z+--list-keys provided but '%s' is not a dict�
)�anyr?rrrr0r5r%�
print_helpr;rr6r7r2r3rrLr
r�printrrHrCrD�
isinstance�dictr1�sorted�keysrZ
json_dumps)�name�argsrrKZrendered_payloadr:rEr>r#r#r$�handle_argssr�
��
���

�
rWcCst�}t�tt|����dS)z,Tool to query specific instance-data values.N)r%�sys�exitrWr�
parse_argsr!r#r#r$�mainKsr[�__main__)N)!rrZloggingr,rXr4rZ	cloudinitrrZcloudinit.cmd.develrZ!cloudinit.handlers.jinja_templaterrrZcloudinit.sourcesr	Zcloudinit.templaterr
rZ	getLogger�__name__r0r%r)rRr;r�boolrHrWr[r#r#r#r$�<module>s0


XQ�+C
Cubjrnet7 Shell
