3
\me(# @ s^ d dl Z d dlmZmZmZmZmZmZ d dlm Z e j
e Z dZ
dZdZdZdZd ZG d
d d ejZeejejffgZd d
Ze
eeeefddZe
eeeefddZe dkrZd dlZej ddZ e j ddde
d e j ddded e j d d e d ed e j d d e d!ed e j d"d#e d$ed e j Z!ee!j"e!j#e!j$e!j%e!j&d%Z'e(ej)e' dS )& N)
atomic_helperdmihelperssources
url_helperutil)ec2zhttp://169.254.169.254z1.0i
Exoscalec sX e Zd ZdZdZ fddZdd Zdd Zd d
Zd d Z d
d Z
e dd Z Z
S )DataSourceExoscaler x c s| t t| j||| tjd | jjdt| _| jjdt | _
t | jjdt | _
| jjdt| _| jjdt| _i | _d S )Nz$Initializing the Exoscale datasource metadata_url api_versionpassword_server_porttimeoutretries)superr __init__LOGdebugZds_cfgget METADATA_URLr API_VERSIONr intPASSWORD_SERVER_PORTr URL_TIMEOUT url_timeout URL_RETRIES url_retries extra_config)selfZsys_cfgZdistropaths) __class__ (/usr/lib/python3.6/DataSourceExoscale.pyr s
zDataSourceExoscale.__init__c C s8 d}| j jd}tj|}|jd| dr4tjd dS )zdS tj|dddi||d |S )
zObtain the VM's password if set.
Once fetched the password is marked saved. Future calls to this method may
return empty string or 'saved_password'.z {}:{}/{}/NZ DomU_RequestZsend_my_password)Z ssl_detailsZheadersr r zutf-8 saved_password)rB rC )r) r Zread_file_or_urlcontentsdecode)r r r r r Z password_urlZresponser4 r$ r$ r% get_password s$
rF c C s i }||d< y0t j|| ||d|d< t j|| ||d|d< W n0 tk
rl } ztjtd| | i S d}~X nX yt|| |||d|d< W n4 tk
r } ztjtd | || W Y dd}~X nX |S )
z8Query the metadata server and return the retrieved data.Z_metadata_api_version)r r z user-dataz meta-dataz(failed reading from metadata url %s (%s)N)r r r r r r4 z2failed to read from password server url %s:%s (%s))r Zget_instance_userdataZget_instance_metadata Exceptionr Zlogexcr rF )r r r r r Zcrawled_metadataer$ r$ r% r2 s4 r2 __main__zQuery Exoscale Metadata) descriptionz
--endpointZURLz The url of the metadata service.)metavarhelpdefaultz --versionVERSIONz.The version of the metadata endpoint to query.z --retriesZNUMz,The number of retries querying the endpoint.)rK typerL rM z --timeoutz.The time in seconds to wait before timing out.z--password-portZPORTz/The port on which the password endpoint listens)r r r r r )*ZloggingZ cloudinitr r r r r r Zcloudinit.sources.helpersr Z getLoggerr; r r r r r r r9 Z
DataSourcer ZDEP_FILESYSTEMZ DEP_NETWORKr@ rA rF r2 argparseArgumentParserparser add_argumentr
parse_argsargsZendpointversionZ
password_portr r r5 printZ
json_dumpsr$ r$ r$ r% s|
" &