← Back
Editing: ec2.cpython-38.pyc
U ӇgM" � @ s� d dl Z d dlZd dlZd dlmZmZ e�e�Ze ej g�Z G dd� d�Z G dd� d�Z dd� Zddd d dddd �d d�Zddd�Zddd�Zddd�ZdS )� N)� 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 �?/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 tk r. | Y S X | �|�r|z t�|�W S ttfk rz } zt� d||� W 5 d }~X Y nX |� 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 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 �|�}|s ||fS dd� }dd� }|�� D ]�}|�� }||�}|r6|sTq6|dkr^q6||�rz||kr�|�|� q6|�dd�}|} t|�dkr�|\} } t �| �} | d k r�d| } | }| ||< q6||fS ) Nc S s | � d�rdS dS d S )N�/TF)r ��itemr r r � has_childrenI s z1MetadataMaterializer._parse.<locals>.has_childrenc S s | � d�r| �d�S | S )Nr/ )r �rstripr0 r r r �get_nameO s z-MetadataMaterializer._parse.<locals>.get_namezsecurity-credentials�=� 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 S r% )r' � _materializer& r( )r r r r � materializel s z MetadataMaterializer.materializec C s� | � |�\}}i }|D ]<}t�||�}|�d�s8|d7 }| �|�}| �||�||< qi } |�� D ].\} } t�|| �} | �| �} | �| | �| | < q`i }|�|� | � � D ]&} | |kr�t � d|� q�| | || < q�|S )Nr/ z&Duplicate key found in results from %s) r> r � combine_urlr r) r? �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 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_codes�causer r r �skip_retry_on_codes� s rH �latest�http://169.254.169.254� �� ssl_details�timeout�retries� headers_cb�headers_redact� exception_cbc C s� t �|| �}t �|d�}d} z2|s.t�tt�}t j|||||||d�} | j} W n\ t jk r� } z | j tkrzt � t d|� W 5 d } ~ X Y n" t k r� t � t d|� Y nX | S )Nz user-data� )rM rN rO rR rP rQ z$Failed fetching userdata from url %s)r rA � functools�partialrH �SKIP_USERDATA_CODES�read_file_or_urlr< �UrlErrorrF r �logexcr � Exception) � api_version�metadata_addressrM rN rO rP rQ rR Zud_urlZ user_data�responser r r r �get_instance_userdata� s2 �� r^ c s� t �||| �} tjt j|||||| d�� � �fdd�} z6� | �} t| j| | |d�}|�� }t|t �shi }|W S t k r� t � t d| | � i Y S X d S )NrL c s\ z � | �j W S tjk rV } z*�r,�|�s0� nt�d| � W Y �dS W 5 d }~X Y nX d S )Nz&Skipped retrieval of the content of %sz (skipped))r< r rX r r )Zurlr �r, � retrieval_exception_ignore_cbr r �mcaller� s �� z'_get_instance_metadata.<locals>.mcaller)r- z Failed fetching %s from url %s)r rA rT rU rW r$ r< r@ � isinstance�dictrZ r rY r )�treer[ r\ rM rN rO r- rP rQ rR r` Zmd_urlra r] Z materializerZmdr r_ r �_get_instance_metadata� s4 � � re c C s t d| ||||||||| d� S )Nz meta-data/) rd r[ r\ rM rN rO r- rQ rP rR r` �re ) r[ r\ rM rN rO r- rP rQ rR r` r r r �get_instance_metadata� s �rg c C s t d| ||||||||d� S )Nzdynamic/instance-identity) rd r[ r\ rM rN rO r- rQ rP rR rf ) r[ r\ rM rN rO r- rP rQ rR r r r �get_instance_identity s �rh )rI rJ ) rI rJ NrK rK NNNNN) rI rJ NrK rK NNNNN) rI rJ NrK rK NNNN)rT r ZloggingZ cloudinitr r Z getLoggerr r � frozensetZ NOT_FOUNDrV r r$ rH r^ re rg rh r r r r �<module> sd #T �� * � 4 � �
Save File
Cancel