File: //lib/python3.6/site-packages/cloudinit/__pycache__/ec2_utils.cpython-36.opt-1.pyc
3
�� b� � @ s� d dl Z d dlZd dlmZ d dlmZmZ eje�Z e
ej g�Z G dd� de
�ZG dd� de
�Zdd � Zdd
d�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 |j � }|jd�r(|jd�r(dS dS )NF�{�}T)�strip�
startswith�endswith)�self�text� r
� /usr/lib/python3.6/ec2_utils.py�_maybe_json_object s z&MetadataLeafDecoder._maybe_json_objectc C s� |sdS yt j|�}W n tk
r* |S X | j|�rvy
tj|�S ttfk
rt } ztj d||� W Y d d }~X nX |j
d�dkr�|j � 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 kr*t� | _n|| _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 j|�}|s ||fS dd� }dd� }x�|j� D ]�}|j� }||�}| s8| rZq8|dkrdq8||�r�||kr�|j|� q8|jdd�}|} t|�dkr�|\}
} t j|
�}
|
d k r�d|
} | }| ||< q8W ||fS ) Nc S s | j d�rdS dS d S )N�/TF)r
)�itemr
r
r � has_childrenI s
z1MetadataMaterializer._parse.<locals>.has_childrenc S s | j d�r| jd�S | S )Nr0 )r
�rstrip)r1 r
r
r �get_nameO s
z-MetadataMaterializer._parse.<locals>.get_namezsecurity-credentials�=r z%s/openssh-key)r r r r �append�split�lenZsafe_int) r r �leaves�childrenr2 r4 r Z
field_name�contents�resourceZidentZ sub_contentsr
r
r �_parseA s6
zMetadataMaterializer._parsec C s( | j d k r| j S | j| j| j�| _ | j S )N)r( � _materializer' r) )r r
r
r � materializel s
z MetadataMaterializer.materializec C s� | j |�\}}i }xD|D ]<}tj||�}|jd�s:|d7 }| j|�}| j||�||< qW i } x:|j� D ].\}
} tj|| �} | j| �}
| j|
|
�| |
< qfW i }|j|� x2| j � D ]&}
|
|kr�t
j d|� q�| |
||
< q�W |S )Nr0 z&Duplicate key found in results from %s) r= r � combine_urlr
r* r> �itemsr+ �update�keysr r )r r r, r9 r: Zchild_contents�cZ child_urlZ
child_blobZ
leaf_contentsr r< Zleaf_urlZ leaf_blobZjoinedr
r
r r> r s(
z!MetadataMaterializer._materialize)N)r" r# r$ r/ r= r? r> r
r
r
r r& 6 s
+r& c C s
|j | kS )z/Returns False if cause.code is in status_codes.)�code)Z status_codesZ
_request_args�causer
r
r �skip_retry_on_codes� s rG �latest�http://169.254.169.254� c C s� t j|| �}t j|d�}d} y2|s.tjtt�}t j|||||||d�}
|
j} W n\ t jk
r� } z | j tkrzt
j t d|� W Y d d } ~ X n" t
k
r� t
j t d|� Y nX | S )Nz user-datar )� ssl_details�timeout�retries� exception_cb�
headers_cb�headers_redactz$Failed fetching userdata from url %s)r r@ � functools�partialrG �SKIP_USERDATA_CODES�read_file_or_urlr; ZUrlErrorrE r �logexcr � Exception) � api_version�metadata_addressrK rL rM rO rP rN Zud_urlZ user_data�responser r
r
r �get_instance_userdata� s,
rZ c
s� t j||| �}
tjt j|||||| d�� � fdd�} y4� |
�} t| j|
| |d�}
|
j� }t|t �sfi }|S t
k
r� t j t
d| |
� i S X d S )N)rK rL rM rO rP rN c s
� | �j S )N)r; )Zurl)r- r
r �mcaller� s z'_get_instance_metadata.<locals>.mcaller)r. z Failed fetching %s from url %s)r r@ rQ rR rT r& r; r? �
isinstance�dictrV r rU r )�treerW rX rK rL rM r. rO rP rN Zmd_urlr[ rY Z materializerZmdr
)r- r �_get_instance_metadata� s*
r_ c C s t d| ||||||||d�
S )Nz
meta-data/)
r^ rW rX rK rL rM r. rP rO rN )r_ ) rW rX rK rL rM r. rO rP rN r
r
r �get_instance_metadata� s
r` c C s t d| ||||||||d�
S )Nzdynamic/instance-identity)
r^ rW rX rK rL rM r. rP rO rN )r_ ) rW rX rK rL rM r. rO rP rN r
r
r �get_instance_identity� s ra )rH rI NrJ rJ NNN) rH rI NrJ rJ NNNN) rH rI NrJ rJ NNNN) rH rI NrJ rJ NNNN)rQ r Z cloudinitr Zloggingr r Z getLoggerr" r � frozensetZ NOT_FOUNDrS �objectr r& rG rZ r_ r` ra r
r
r
r �<module> sV
#T
!