File: //lib/python3/dist-packages/cloudinit/sources/helpers/__pycache__/ec2.cpython-310.pyc
o
�Ad�" � @ s� d dl Z d dlZd dlmZ d dlmZmZ e�e�Z e
ej g�Z G dd� d�Z
G dd� d�Zdd � Z
dd
d�Z
ddd�Z
ddd�Z
ddd�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�rdS dS )NF�{�}T)�strip�
startswith�endswith)�self�text� r
�?/usr/lib/python3/dist-packages/cloudinit/sources/helpers/ec2.py�_maybe_json_object s z&MetadataLeafDecoder._maybe_json_objectc
C s� |sdS zt �|�}W n ty | Y S w | �|�r>zt�|�W S ttfy= } z
t� d||� W Y d }~nd }~ww |�
d�dkrI|� � 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 s6 || _ d | _|| _|| _|d u rt� | _d S || _d S �N)�_blob�_md� _base_url�_callerr �
_leaf_decoder)r r �base_url�caller� leaf_decoderr
r
r �__init__7 s
z MetadataMaterializer.__init__c C s� i }g }t �|�}|s||fS dd� }dd� }|�� D ]G}|�� }||�}|r)|s*q|dkr/q||�r=||vr<|�|� q|�dd�}|} t|�dkr^|\}
} t �|
�}
|
d ur^d|
} | }| ||< q||fS ) Nc S s | � d�rdS dS )N�/TF)r
��itemr
r
r � has_childrenI s
z1MetadataMaterializer._parse.<locals>.has_childrenc S s | � d�r
| �d�S | S )Nr0 )r
�rstripr1 r
r
r �get_nameO 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�ident� sub_contentsr
r
r �_parseA s8
�
zMetadataMaterializer._parsec C s( | j d ur| j S | �| j| j�| _ | j S r& )r( � _materializer' r) )r r
r
r � materializel s
z MetadataMaterializer.materializec C s� | � |�\}}i }|D ] }t�||�}|�d�s |d7 }| �|�}| �||�||< q i } |�� D ]\}
} t�|| �} | �| �}
| �|
|
�| |
< q0i }|�|� | � � D ]}
|
|v r`t
� d|� qS| |
||
< qS|S )Nr0 z&Duplicate key found in results from %s) rC r � combine_urlr
r* rD �itemsr+ �update�keysr r )r r r, r< r= �child_contents�c� child_url�
child_blob�
leaf_contentsr r@ �leaf_url� leaf_blob�joinedr
r
r rD r s(
z!MetadataMaterializer._materializer& )r! r" r# r/ rC rE rD r
r
r
r r% 6 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
rV �latest�http://169.254.169.254� c
C s� t �|| �}t �|d�}d} z|st�tt�}t j|||||||d�}
|
j} W | S t jyO } z| j tvrDt
� t d|� W Y d } ~ | S W Y d } ~ | S d } ~ w t
y_ t
� t d|� Y | S w )Nz user-datar )� ssl_details�timeout�retries� exception_cb�
headers_cb�headers_redactz$Failed fetching userdata from url %s)r rF � functools�partialrV �SKIP_USERDATA_CODES�read_file_or_urlr? �UrlErrorrR r �logexcr � Exception) � api_version�metadata_addressrZ r[ r\ r^ r_ r] �ud_url� user_data�responser r
r
r �get_instance_userdata� s<
�� �
�� ��rl c s� t �||| �} tjt j|||||| d�� � �fdd�} z� | �}
t|
j| | |d�}|�� }t|t �s4i }|W S t
yI t � t
d| | � i Y S w )N)rZ r[ r\ r^ r_ r] c
sP z� | �j W S tjy' } z�r�|�s� t�d| � W Y d }~dS d }~ww )Nz&Skipped retrieval of the content of %sz (skipped))r? r rd r r )�urlr �r- � retrieval_exception_ignore_cbr
r �mcaller� s �� ��z'_get_instance_metadata.<locals>.mcaller)r. z Failed fetching %s from url %s)r rF r` ra rc r% r? rE �
isinstance�dictrf r re r )�treerg rh rZ r[ r\ r. r^ r_ r] ro �md_urlrp rk � materializer�mdr
rn r �_get_instance_metadata� s0
�
�
�rw c
C s t d| ||||||||| d� S )Nz
meta-data/) rs rg rh rZ r[ r\ r. r_ r^ r] ro �rw )
rg rh rZ r[ r\ r. r^ r_ r] ro r
r
r �get_instance_metadata� s �ry c C s t d| ||||||||d�
S )Nzdynamic/instance-identity)
rs rg rh rZ r[ r\ r. r_ r^ r] rx ) rg rh rZ r[ r\ r. r^ r_ r] r
r
r �get_instance_identity s �rz )rW rX NrY rY NNN)
rW rX NrY rY NNNNN) rW rX NrY rY NNNN)r` r � cloudinitr �loggingr r � getLoggerr! r � frozenset� NOT_FOUNDrb r r% rV rl rw ry rz r
r
r
r �<module> sd
#T
�)
�4
� �