Google
Edit File: DataSourceExoscale.cpython-39.pyc
a '�Dg8" � @ s^ d dl Z d dlmZmZmZmZmZmZ d dlm Z e � e �Z dZ dZdZdZdZd ZG d d � d ej�ZeejejffgZd d � Ze eeeefdd�Ze eeeefdd�Ze dk�rZd dlZej dd�Z 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 � � Z!ee!j"e!j#e!j$e!j%e!j&d%�Z'e(e�)e'�� dS )&� N)� atomic_helper�dmi�helpers�sources� url_helper�util)�ec2zhttp://169.254.169.254z1.0i� � � �Exoscalec sX e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d � Zd d � Z d d� Z e dd� �Z � Z S )�DataSourceExoscaler �x c s| t t| ��|||� t�d� | j�dt�| _| j�dt �| _ t | j�dt ��| _ | j�dt�| _| j�dt�| _i | _d S )Nz$Initializing the Exoscale datasource� metadata_url� api_version�password_server_port�timeout�retries)�superr �__init__�LOG�debugZds_cfg�get� METADATA_URLr � API_VERSIONr �int�PASSWORD_SERVER_PORTr � URL_TIMEOUT� url_timeout� URL_RETRIES� url_retries� extra_config)�selfZsys_cfgZdistro�paths�� __class__� �H/usr/lib/python3.9/site-packages/cloudinit/sources/DataSourceExoscale.pyr s �zDataSourceExoscale.__init__c C s8 d}| j �d�}t�|�}|�d| d�r4t�d� dS )z<Adjust set_passwords module to run 'always' during each bootZ set_passwordsZsemZconfig_Nz5Overriding module set_passwords with frequency always)r" Z get_ipath_curr ZFileSemaphores�clearr r )r! ZcfgZis_new_instance�modZsem_pathZ sem_helperr% r% r& �activate* s zDataSourceExoscale.activatec C s6 d� | j| j�}tj|g| j| jtjd�\}}t |�S )z.Wait for the metadata service to be reachable.z{}/{}/meta-data/instance-id)ZurlsZmax_waitr Z status_cb) �formatr r r Z wait_for_url� url_max_waitr r Zcritical�bool)r! r �urlZ _responser% r% r& �wait_for_metadata_service9 s �� z,DataSourceExoscale.wait_for_metadata_servicec C s* | � � }|si S t| j| j| j| j| j�S )zw Crawl the metadata service when available. @returns: Dictionary of crawled metadata content. )r. � read_metadatar r r r r )r! Zmetadata_readyr% r% r& �crawl_metadataI s �z!DataSourceExoscale.crawl_metadatac C s\ | � � }|sdS |d | _|d | _|�d�}i }|rFd|ddid�}t�| j|g�| _dS )z�Fetch 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. F� user-data� meta-data�passwordTZexpire)Z ssh_pwauthr3 Zchpasswd)r0 Z userdata_raw�metadatar r Z mergemanydictr )r! �datar3 Zpassword_configr% r% r& � _get_data\ s" �� �z DataSourceExoscale._get_datac C s | j S �N)r )r! r% r% r& �get_config_obj� s z!DataSourceExoscale.get_config_objc C s t �d��t�S )Nzsystem-product-name)r Z read_dmi_data� startswith�EXOSCALE_DMI_NAMEr% r% r% r& � ds_detect� s �z DataSourceExoscale.ds_detect)�__name__� __module__� __qualname__Zdsnamer+ r r) r. r0 r6 r8 � staticmethodr; � __classcell__r% r% r# r& r s 'r c C s t �| t�S r7 )r Zlist_from_depends� datasources)Zdependsr% r% r& �get_datasource_list� s rB c C sZ d� | ||�}tj|dddi||d�}|j�d�}|dv r>dS tj|dddi||d� |S ) z�Obtain 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_passwordrD )r* r Zread_file_or_url�contents�decode)r r r r r Z password_urlZresponser3 r% r% r& � get_password� s* �� �rG c C s� i }||d< z0t j|| ||d�|d< t j|| ||d�|d< W n8 tyt } z t�td| |� i W Y d}~S d}~0 0 zt|| |||d�|d< W n6 ty� } z t�td | ||� W Y d}~n d}~0 0 |S ) z8Query the metadata server and return the retrieved data.Z_metadata_api_version)r r r1 r2 z(failed reading from metadata url %s (%s)N)r r r r r r3 z2failed to read from password server url %s:%s (%s))r Zget_instance_userdataZget_instance_metadata� Exceptionr Zlogexcr rG )r r r r r Zcrawled_metadata�er% r% r& r/ � s> � ����r/ �__main__zQuery Exoscale Metadata)� descriptionz --endpointZURLz The url of the metadata service.)�metavar�help�defaultz --versionZVERSIONz.The version of the metadata endpoint to query.z --retriesZNUMz,The number of retries querying the endpoint.)rL �typerM rN 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 r: Z DataSourcer ZDEP_FILESYSTEMZ DEP_NETWORKrA rB rG r/ �argparse�ArgumentParser�parser� add_argumentr � parse_args�argsZendpoint�versionZ password_portr r r5 �printZ json_dumpsr% r% r% r& �<module> s� x�� '� + ������