File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceExoscale.cpython-310.pyc
o
�Ad�# � @ sh d dl mZmZ d dl mZ d dl mZmZmZ d dlm Z e�
e �Z dZ
dZdZdZd Zd
ZG d d � d ej�ZeejejffgZd
d� Ze
eeeefdd�Ze
eeeefdd�Ze dkr�d dlZej dd�Z e j ddde
d� e j dd d ed� 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 dS )(� )�dmi�helpers)�log)�sources�
url_helper�util)�ec2zhttp://169.254.169.254z1.0i� �
� �Exoscalec sT e Zd ZdZdZ� fdd�Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
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�debug�ds_cfg�get� METADATA_URLr � API_VERSIONr �int�PASSWORD_SERVER_PORTr � URL_TIMEOUT� url_timeout� URL_RETRIES� url_retries� extra_config)�self�sys_cfg�distro�paths�� __class__� �F/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceExoscale.pyr s
�
zDataSourceExoscale.__init__c C s< d}| j �d�}t�|�}|�d| d�r t�d� dS dS )z<Adjust set-passwords module to run 'always' during each boot�
set_passwords�sem�config_Nz5Overriding module set-passwords with frequency always)r% �
get_ipath_curr �FileSemaphores�clearr r )r" �cfg�is_new_instance�mod�sem_path�
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)�urls�max_waitr � status_cb)
�formatr r r � wait_for_url� url_max_waitr r �critical�bool)r" r �url� _responser( r( r) �wait_for_metadata_service9 s �
�z,DataSourceExoscale.wait_for_metadata_servicec C s6 t jtjd| jd�}|si S t| j| j| j| j | j
�S )zw
Crawl the metadata service when available.
@returns: Dictionary of crawled metadata content.
z waiting for the metadata service��logfunc�msg�func) r �log_timer �infor@ �
read_metadatar r r r r )r" �metadata_readyr( r( r) �crawl_metadataI s ��z!DataSourceExoscale.crawl_metadatac C st | � � sdS tjtjd| jd�}|sdS |d | _|d | _|�d�}i }|r/d|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.
FzCrawl of metadata servicerA � user-data� meta-data�passwordT�expire)�
ssh_pwauthrL �chpasswd) �_is_platform_viabler rE r r rI � userdata_raw�metadatar �
mergemanydictr! )r" �datarL �password_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 �
read_dmi_data�
startswith�EXOSCALE_DMI_NAMErX r( r( r) rP � s
�z&DataSourceExoscale._is_platform_viable)
�__name__�
__module__� __qualname__�dsnamer; r r5 r@ rI rV rY rP �
__classcell__r( r( r& r) r s
.r c C s t �| t�S rW )r �list_from_depends� datasources)�dependsr( r( r) �get_datasource_list� s re 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 {}:{}/{}/N� DomU_Request�send_my_password)� ssl_details�headersr r zutf-8)� �saved_passwordrk )r9 r �read_file_or_url�contents�decode)r r r r r � password_url�responserL r( r( r) � get_password� s* �� �rq c
C s� i }||d< zt j|| ||d�|d< t j|| ||d�|d< W n ty: } zt�td| |� i W Y d}~S d}~ww zt|| |||d�|d< W |S tye } zt�td | ||� W Y d}~|S d}~ww )
z8Query the metadata server and return the retrieved data.�_metadata_api_version)r r rJ rK z(failed reading from metadata url %s (%s)N)r r r r r rL z2failed to read from password server url %s:%s (%s))r �get_instance_userdata�get_instance_metadata� Exceptionr �logexcr rq )r r r r r �crawled_metadata�er( r( r) rG � sH
����� �����rG �__main__NzQuery Exoscale Metadata)� descriptionz
--endpoint�URLz The url of the metadata service.)�metavar�help�defaultz --version�VERSIONz.The version of the metadata endpoint to query.z --retries�NUMz,The number of retries querying the endpoint.)r| �typer} r~ z --timeoutz.The time in seconds to wait before timing out.z--password-port�PORTz/The port on which the password endpoint listens)r r r r r )*� cloudinitr r r �loggingr r r �cloudinit.sources.helpersr � getLoggerr] r r r r r r r\ �
DataSourcer �DEP_FILESYSTEM� DEP_NETWORKrc re rq rG �argparse�ArgumentParser�parser� add_argumentr �
parse_args�args�endpoint�version�
password_portr r rT �print�
json_dumpsr( r( r( r) �<module> s�
�
�'
�+ �������