File: //lib/python3/dist-packages/cloudinit/sources/__pycache__/DataSourceHetzner.cpython-38.pyc
U
Ӈg� � @ s� d Z ddlZddlm m mZ ddlmZm Z mZm
Z
ddl m Z ddl
mZ e�e�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 j�| |||� || _t� | _t�t�|ddgi �t g�| _
| j
d | _ | j
d | _ | j
�
dt�| _| j
�
dt�| _| j
�
dt�| _t j| _t j| _d | _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�
metadata_full)�self�sys_cfgr �paths� r# �E/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceHetzner.pyr " s �� zDataSourceHetzner.__init__c
C s t � \}}|sdS zbt| jt�� dtd igd��: tj| j| j | j
| j d�}tj | j
| j | j
| j d�}W 5 Q R X W n0 tk
r� } zt�d|� � W 5 d }~X Y nX t�|�| _|| _|d | jd< |d | jd < |�d
d �| jd
< |�d d �| jd < |�d d �| _| �� |k�r td
��dS )NFZurlz/metadata/instance-id)ZifaceZconnectivity_urls_data)r Z sec_betweenr zBailing, DHCP Exception: %sz instance-idZhostnamezlocal-hostname�network-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_rawr r r Zvendordata_raw�get_instance_id� RuntimeError)r Z
on_hetzner�serialZmdZud�er# r# r$ � _get_data5 sN
��� �� �zDataSourceHetzner._get_datac C s t �| �� d�S )N�system-serial-number)r Z instance_id_matches_system_uuidr+ )r r! r# r# r$ �check_instance_idp s �z#DataSourceHetzner.check_instance_idc C sT | j dkr t�dtj� 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 %sr% z'Unable to get meta-data from server....)r r) Zwarningr r r r, )r Z _net_configr# r# r$ �network_configu s
�
z DataSourceHetzner.network_configN) �__name__�
__module__� __qualname__Zdsnamer r/ r1 �propertyr2 r# r# r# r$ r s ;r c C sB t �d�} | dkrdS t �d�}|r2t�d|� ntd��d|fS )Nzsystem-manufacturerr
)FNr0 z#Running on Hetzner Cloud: serial=%sz+Hetzner Cloud detected, but no serial foundT)r Z
read_dmi_datar) �debugr, )Z vendor_namer- r# r# r$ r& � s
r& c C s t �| t�S )N)r Zlist_from_depends� datasources)Zdependsr# r# r$ �get_datasource_list� s r9 ) �__doc__ZloggingZ!cloudinit.sources.helpers.hetznerr ZhelpersZhetznerr( Z cloudinitr r r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Z getLoggerr3 r) r' r r r r r r r& ZDEP_FILESYSTEMr8 r9 r# r# r# r$ �<module> s$
�q
�