Google
Edit File: query.cpython-39.opt-1.pyc
a '�Dgk, � @ s� d Z ddlZddlZddlZddlZddlmZ ddlmZm Z ddl m Z ddl m Z mZmZ ddlmZ ddlmZ d Ze�e�Zdd d �Zd d � Zed�dd�Zeee e d�dd�Z dd� Z dd� Z edkr�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)� atomic_helper�util)�read_cfg_paths)�convert_jinja_instance_data�get_jinja_variable_alias�render_jinja_payload)�REDACT_SENSITIVE_VALUE)�JinjaSyntaxParsingException�queryc 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. )�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)�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# �7/usr/lib/python3.9/site-packages/cloudinit/cmd/query.py� get_parser& st � �� � � � � �� r% c C sB t j| dd�}z |�d�W S ty< t j|ddd� Y S 0 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. T)�quietzutf-8F)r& �decodeN)r Zload_binary_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�} |�d�} zt � |�} W nN t t fy� } z2| jtkr�t�d|� n t�d |� � W Y d } ~ n d } ~ 0 0 t �| �} zt �t � | ��} W n t t f�y4 d } Y n0 |dk�rrd t|f | d < d t| f | d < d t| 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. r r Zinstance_data_sensitivez4Missing root-readable %s. Using redacted %s instead.z user-data.txtzvendor-data.txt�combined_cloud_configz$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 Zload_text_file�IOError�OSError�errnor �errorZ load_jsonr r) )r � user_data� vendor_data�uid�pathsZinstance_data_fnZredacted_data_fnZsensitive_data_fnZ user_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 C s� d}| }|� d�D ]�}z || }W nL tyn } z4|rFdj||d�}n d�|�}t|�|�W Y d}~n d}~0 0 ||v r�|| }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_partr: �msg�keyr# r# r$ �(_find_instance_data_leaf_by_varname_path� s, � rH c C s� t |j|j|j|jg�s0t�d� t� �� dS zt |j |j |j �}W n t tfy^ Y dS 0 |jr�dj|jd�}z t|d||jr�dndd�}W n6 ty� } z t�d t|�� W Y d }~dS d }~0 0 |r�t|� d S dS t|�}|j�rJt|dd �}zt|||j|jd �}W n6 ttf�yH } zt�|� W Y d }~dS d }~0 0 |j�r�t|t��spt�d|j� dS d�t|� � ��}t|t��s�t � |�}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})�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? r r r r0 r5 r% � print_helpr; r r6 r7 r2 r3 r rL r r �printr rH rC rD � isinstance�dictr1 �sorted�keysr Z json_dumps)�name�argsr rK Zrendered_payloadr: rE r> r# r# r$ � handle_args sr � � � ��� � rW c C s t � } t�tt| �� �� dS )z,Tool to query specific instance-data values.N)r% �sys�exitrW r � parse_argsr! r# r# r$ �mainK s r[ �__main__)N)!r r Zloggingr, rX r4 r Z cloudinitr r Zcloudinit.cmd.develr Z!cloudinit.handlers.jinja_templater r r Zcloudinit.sourcesr Zcloudinit.templaterr r Z getLogger�__name__r0 r% r) rR r; r �boolrH rW r[ r# r# r# r$ �<module> s0 XQ� +C