"T`# @sIddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ej e Z dZ d Zd Zd Zd Zd ZGdddejZeejejffgZddZe eeeeddZe eeeeddZe dkrEddlZejddZejdddddde ejd dd!dd"deejd#dd$d%edd&deejd'dd$d%edd(deejd)dd*d%edd+deej Z!ed,e!j"d-e!j#d.e!j$d/e!j%d0e!j&Z'e(e j)e'dS)1)dmi) ec2_utils)log)sources)helpers) url_helper)utilzhttp://169.254.169.254z1.0i ExoscalecsveZdZdZdZfddZddZddZd d Zd d Z d dZ ddZ S)DataSourceExoscaler xcstt|j|||tjd|jjdt|_|jjdt |_ t |jjdt |_ |jjdt|_|jjdt|_i|_dS)Nz$Initializing the Exoscale datasource metadata_url api_versionpassword_server_porttimeoutretries)superr __init__LOGdebugZds_cfgget METADATA_URLr API_VERSIONrintPASSWORD_SERVER_PORTr URL_TIMEOUT url_timeout URL_RETRIES url_retries extra_config)selfZsys_cfgZdistropaths) __class__F/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceExoscale.pyr s zDataSourceExoscale.__init__cCsNd}|jjd}tj|}|jd|drJtjddS)zdS|d|_|d|_|jd}i}|rd d d|d d dii}tj |j |g|_ d S) zFetch the user data, the metadata and the VM password from the metadata service. Please refer to the datasource documentation for details on how the metadata server and password server are crawled. Fr-r.zCrawl of metadata servicer/z user-dataz meta-datapasswordZ ssh_pwauthTZchpasswdZexpire) _is_platform_viablerr0rrr3Z userdata_rawZmetadatarZ mergemanydictr )r!datar4Zpassword_configr$r$r% _get_data[s(      zDataSourceExoscale._get_datacCs|jS)N)r )r!r$r$r%get_config_objsz!DataSourceExoscale.get_config_objcCstjdjtS)Nzsystem-product-name)rZ read_dmi_data startswithEXOSCALE_DMI_NAME)r!r$r$r%r5sz&DataSourceExoscale._is_platform_viable) __name__ __module__ __qualname__Zdsnamer*rr(r,r3r7r8r5r$r$)r#r%r s     , r cCstj|tS)N)rZlist_from_depends datasources)Zdependsr$r$r%get_datasource_listsr?c Csdj|||}tj|dddddid|d|}|jjd }|d krddStj|ddddd id|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 {}:{}/{}/Z ssl_detailsNZheadersZ DomU_RequestZsend_my_passwordrrzutf-8saved_password)r@rA)r)rZread_file_or_urlcontentsdecode)rrrrrZ password_urlZresponser4r$r$r% get_passwords$      rDc Csi}||dtk r }ztjtd|||WYdd}~XnX|S)z8Query the metadata server and return the retrieved data.Z_metadata_api_versionrrz user-dataz meta-dataz(failed reading from metadata url %s (%s)Nrrrrrr4z2failed to read from password server url %s:%s (%s))ec2Zget_instance_userdataZget_instance_metadata ExceptionrZlogexcrrD)rrrrrZcrawled_metadataer$r$r%r2s8      r2__main__NZ descriptionzQuery Exoscale Metadataz --endpointmetavarZURLhelpz The url of the metadata service.defaultz --versionVERSIONz.The version of the metadata endpoint to query.z --retriesZNUMtypez,The number of retries querying the endpoint.z --timeoutz.The time in seconds to wait before timing out.z--password-portZPORTz/The port on which the password endpoint listensrrrrr)*Z cloudinitrrrErZloggingrrrrZ getLoggerr;rrrrrrr:Z DataSourcer ZDEP_FILESYSTEMZ DEP_NETWORKr>r?rDr2argparseArgumentParserparser add_argumentr parse_argsargsZendpointversionZ password_portrrr6printZ json_dumpsr$r$r$r%sw "