File: //usr/lib/python3/dist-packages/cloudinit/__pycache__/ec2_utils.cpython-39.pyc
a
��`�" � @ s� d dl Z d dlZd dlmZ d dlmZ d dlmZ e�e�Z e
ej g�Z G dd� de
�ZG dd� de
�Zd d
� Zddd�Zddd�Zddd�Zddd�ZdS )� N)�log)�
url_helper)�utilc @ s e Zd ZdZdd� Zdd� ZdS )�MetadataLeafDecoderz.Decodes a leaf blob into something meaningful.c C s, |sdS |� � }|�d�r(|�d�r(dS dS )NF�{�}T)�strip�
startswith�endswith)�self�text� r
�5/usr/lib/python3/dist-packages/cloudinit/ec2_utils.py�_maybe_json_object s z&MetadataLeafDecoder._maybe_json_objectc
C s� |sdS zt �|�}W n ty, | Y S 0 | �|�r|z t�|�W S ttfyz } zt� d||� W Y d }~n
d }~0 0 |�
d�dkr�|� � S |S )N� z6Field %s looked like a json object, but it was not: %s�
���) r �
decode_binary�UnicodeDecodeErrorr �json�loads�
ValueError� TypeError�LOG�warning�find�
splitlines)r �field�blob�er
r
r �__call__ s
�z MetadataLeafDecoder.__call__N)�__name__�
__module__� __qualname__�__doc__r r r
r
r
r r s r c @ s. e Zd Zd
dd�Zdd� Zdd� Zdd � ZdS ) �MetadataMaterializerNc C s4 || _ d | _|| _|| _|d u r*t� | _n|| _d S �N)�_blob�_md� _base_url�_callerr �
_leaf_decoder)r r �base_url�caller� leaf_decoderr
r
r �__init__5 s
z MetadataMaterializer.__init__c C s� i }g }t �|�}|s ||fS dd� }dd� }|�� D ]�}|�� }||�}|r6|sTq6|dkr^q6||�rz||vr�|�|� q6|�dd�}|} t|�dkr�|\}
} t �|
�}
|
d ur�d|
} | }| ||< q6||fS ) Nc S s | � d�rdS dS d S )N�/TF)r
��itemr
r
r � has_childrenG s
z1MetadataMaterializer._parse.<locals>.has_childrenc S s | � d�r| �d�S | S )Nr0 )r
�rstripr1 r
r
r �get_nameM s
z-MetadataMaterializer._parse.<locals>.get_namezsecurity-credentials�=� z%s/openssh-key)r r r r �append�split�len�safe_int) r r �leaves�childrenr3 r5 r �
field_name�contents�resource�identZ sub_contentsr
r
r �_parse? s6
zMetadataMaterializer._parsec C s( | j d ur| j S | �| j| j�| _ | j S r& )r( � _materializer' r) )r r
r
r � materializej s
z MetadataMaterializer.materializec C s� | � |�\}}i }|D ]<}t�||�}|�d�s8|d7 }| �|�}| �||�||< qi } |�� D ].\}
} t�|| �} | �| �}
| �|
|
�| |
< q`i }|�|� | � � D ]&}
|
|v r�t
� d|� q�| |
||
< q�|S )Nr0 z&Duplicate key found in results from %s) rB r � combine_urlr
r* rC �itemsr+ �update�keysr r )r r r, r< r= Zchild_contents�cZ child_urlZ
child_blobZ
leaf_contentsr r@ Zleaf_urlZ leaf_blobZjoinedr
r
r rC p s,
�z!MetadataMaterializer._materialize)N)r! r" r# r/ rB rD rC r
r
r
r r% 4 s
+r% c C s
|j | vS )z/Returns False if cause.code is in status_codes.)�code)� status_codes�
_request_args�causer
r
r �skip_retry_on_codes� s rN �latest�http://169.254.169.254� c
C s� t �|| �}t �|d�}d} z2|s.t�tt�}t j|||||||d�}
|
j} W n\ t jy� } z$| j tvrxt
� t d|� W Y d } ~ n(d } ~ 0 t
y� t
� t d|� Y n0 | S )Nz user-datar )� ssl_details�timeout�retries� exception_cb�
headers_cb�headers_redactz$Failed fetching userdata from url %s)r rE � functools�partialrN �SKIP_USERDATA_CODES�read_file_or_urlr? �UrlErrorrJ r �logexcr � Exception) � api_version�metadata_addressrR rS rT rV rW rU �ud_url� user_data�responser r
r
r �get_instance_userdata� s( ��
" rd c
s� t �||| �}
tjt j|||||| d�� � fdd�} z6� |
�} t| j|
| |d�}
|
�� }t|t �sfi }|W S t
y� t � t
d| |
� i Y S 0 d S )N)rR rS rT rV rW rU c s
� | �j S r& )r? )�url�r- r
r �mcaller� s z'_get_instance_metadata.<locals>.mcaller)r. z Failed fetching %s from url %s)r rE rX rY r[ r% r? rD �
isinstance�dictr^ r r] r )�treer_ r` rR rS rT r. rV rW rU �md_urlrg rc Z materializer�mdr
rf r �_get_instance_metadata� s* � �
rm c C s t d| ||||||||d�
S )Nz
meta-data/�
rj r_ r` rR rS rT r. rW rV rU �rm � r_ r` rR rS rT r. rV rW rU r
r
r �get_instance_metadata� s �rq c C s t d| ||||||||d�
S )Nzdynamic/instance-identityrn ro rp r
r
r �get_instance_identity� s �rr )rO rP NrQ rQ NNN) rO rP NrQ rQ NNNN) rO rP NrQ rQ NNNN) rO rP NrQ rQ NNNN)rX r � cloudinitr �loggingr r � getLoggerr! r � frozenset� NOT_FOUNDrZ �objectr r% rN rd rm rq rr r
r
r
r �<module> sH
U �
�
�
�