3
\�meB � @ s� d Z ddlZddljjjZddlmZm Z mZm
Z
ddl m Z ddl
mZ eje�ZdZed ed d �Zd
Zd Zd ZG d d
� d
ej�Zdd� ZeejffgZdd� Z dS )z>Hetzner Cloud API Documentation
https://docs.hetzner.cloud/� N)�dmi�net�sources�util)�NoDHCPLeaseError)�EphemeralDHCPv4z!http://169.254.169.254/hetzner/v1z /metadataz /userdata)� metadata_url� userdata_url�< � c @ s4 e Zd ZdZdd� Zdd� Zdd� Zedd � �Zd
S ) �DataSourceHetzner�Hetznerc C s� t jj| |||� || _t� | _tjtj|ddgi �t g�| _
| j
d | _ | j
d | _ | j
j
dt�| _| j
j
dt�| _| j
j
dt�| _t j| _t j| _d S )NZ
datasourcer
r r �retries�timeout�
wait_retry)r �
DataSource�__init__�distro�dict�metadatar Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfg�metadata_address�userdata_address�get�
MD_RETRIESr �
MD_TIMEOUTr �
MD_WAIT_RETRYr �UNSET�_network_configZDSMODE_NETWORKZdsmode)�self�sys_cfgr �paths� r" �'/usr/lib/python3.6/DataSourceHetzner.pyr " s
zDataSourceHetzner.__init__c C s t � \}}|sdS y`t| jtj� dtd id��: tj| j| j | j
| j d�}tj | j
| j | j
| j d�}W d Q R X W n0 tk
r� } ztjd|� � W Y d d }~X nX tj|�| _|| _|d | jd< |d | jd < |jd
d �| jd
< |jd d �| jd < |jd d �| _| j� |k�rtd
��dS )NFZurlz/metadata/instance-id)ZifaceZconnectivity_url_data)r Z sec_betweenr zBailing, DHCP Exception: %sz instance-idZhostnamezlocal-hostnameznetwork-configz public-keysZ vendor_dataz6SMBIOS serial does not match instance ID from metadataT)�get_hcloud_datar r r Zfind_fallback_nic� BASE_URL_V1� hc_helperZ
read_metadatar r r r Z
read_userdatar r �LOG�errorr Zmaybe_b64decodeZ userdata_rawZ
metadata_fullr r Zvendordata_raw�get_instance_id� RuntimeError)r Z
on_hetzner�serialZmdZud�er" r" r# � _get_data4 s@
zDataSourceHetzner._get_datac C s t j| j� d�S )Nzsystem-serial-number)r Z instance_id_matches_system_uuidr) )r r r" r" r# �check_instance_idm s z#DataSourceHetzner.check_instance_idc C sT | j dkr tjdtj� tj| _ | j tjkr2| j S | jd }|sHtd��|| _ | j S )z�Configure the networking. This needs to be done each boot, since
the IP information may have changed due to snapshot and/or
migration.
Nz5Found None as cached _network_config. Resetting to %sznetwork-configz'Unable to get meta-data from server....)r r' Zwarningr r r r* )r Z _net_configr" r" r# �network_configr s
z DataSourceHetzner.network_configN) �__name__�
__module__� __qualname__Zdsnamer r- r. �propertyr/ r" r" r" r# r s
9r c C sB t jd�} | dkrdS t jd�}|r2tjd|� ntd��d|fS ) Nzsystem-manufacturerr
Fzsystem-serial-numberz#Running on Hetzner Cloud: serial=%sz+Hetzner Cloud detected, but no serial foundT)FN)r Z
read_dmi_datar' �debugr* )Z vendor_namer+ r" r" r# r$ � s
r$ c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr" r" r# �get_datasource_list� s r6 ) �__doc__ZloggingZ!cloudinit.sources.helpers.hetznerr ZhelpersZhetznerr& Z cloudinitr r r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Z getLoggerr0 r' r% r r r r r r r$ ZDEP_FILESYSTEMr5 r6 r" r" r" r# �<module> s
n