3
\meW"
@ s d dl Z d dlZd dlZd dlmZmZ ejeZe ej
gZ G dd dZ G dd dZ
dd Zdd d
ZdddZdddZdddZdS ) N)
url_helperutilc @ s e Zd ZdZdd Zdd ZdS )MetadataLeafDecoderz.Decodes a leaf blob into something meaningful.c C s, |sdS |j }|jdr(|jdr(dS dS )NF{}T)strip
startswithendswith)selftext r /usr/lib/python3.6/ec2.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
ddkr|j S |S )N z6Field %s looked like a json object, but it was not: %s
) r
decode_binaryUnicodeDecodeErrorr jsonloads
ValueError TypeErrorLOGwarningfind
splitlines)r
fieldblober 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
ddZdd Zdd Zdd ZdS ) MetadataMaterializerNc C s4 || _ d | _|| _|| _|d kr*t | _n|| _d S )N)_blob_md _base_url_callerr
_leaf_decoder)r
r base_urlcaller 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 rd|
} | }| ||< q8W ||fS ) Nc S s | j drdS dS d S )N/TF)r )itemr r r
has_childrenI s
z1MetadataMaterializer._parse..has_childrenc S s | j dr| jdS | S )Nr/ )r rstrip)r0 r r r
get_nameO s
z-MetadataMaterializer._parse..get_namezsecurity-credentials=r z%s/openssh-key)r r r r appendsplitlenZsafe_int) r
r leaveschildrenr1 r3 r Z
field_namecontentsresourceZidentZ 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||}|jds:|d7 }| j|}| j||||< qW i } x:|j D ].\}
} tj|| } | j| }
| j|
|
| |
< qfW i }|j| x2| j D ]&}
|
|krt
j d| q| |
||
< qW |S )Nr/ z&Duplicate key found in results from %s) r< r combine_urlr r) r= itemsr* updatekeysr r )r
r r+ r8 r9 Zchild_contentscZ 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_argscauser r r
skip_retry_on_codes s rF latesthttp://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-data ) ssl_detailstimeoutretries exception_cb
headers_cbheaders_redactz$Failed fetching userdata from url %s)r r? functoolspartialrF SKIP_USERDATA_CODESread_file_or_urlr: UrlErrorrD r logexcr Exception) api_versionmetadata_addressrK rL rM rO rP rN Zud_urlZ user_dataresponser r r r
get_instance_userdata s,
r[ c s t j||| } tjt j|||||| d fdd} y4 | }
t|
j| | |d}|j }t|t shi }|S t
k
r t j t
d| | i S X d S )N)rK rL rM rO rP rN c sX y
| j S tjk
rR } z* s.| r2 ntjd| dS W Y d d }~X nX d S )Nz&Skipped retrieval of the content of %sz (skipped))r: r rU r r )Zurlr )r, retrieval_exception_ignore_cbr r
mcaller s
z'_get_instance_metadata..mcaller)r- z Failed fetching %s from url %s)r r? rQ rR rT r% r: r>
isinstancedictrW r rV r )treerX rY rK rL rM r- rO rP rN r\ Zmd_urlr] rZ Z materializerZmdr )r, r\ r
_get_instance_metadata s*
ra c
C s t d| ||||||||| d S )Nz
meta-data/) r` rX rY rK rL rM r- rP rO rN r\ )ra )
rX rY rK rL rM r- rO rP rN r\ r r r
get_instance_metadata s rb c C s t d| ||||||||d
S )Nzdynamic/instance-identity)
r` rX rY rK rL rM r- rP rO rN )ra ) rX rY rK rL rM r- rO rP rN r r r
get_instance_identity s rc )rG rH NrI rI NNN)
rG rH NrI rI NNNNN)
rG rH NrI rI NNNNN) rG rH NrI rI NNNN)rQ r ZloggingZ cloudinitr r Z getLoggerr! r frozensetZ NOT_FOUNDrS r r% rF r[ ra rb rc r r r r
sZ
#T
!
)