3
\�me� � @ s� d dl Z d dlmZ d dlmZmZ d dlmZ d dlm Z d dl
m Z e j
e�ZddiZd Zd
Zd
ZG d d � d ej�ZG d
d� de�ZeejffeejejffgZdd� ZdS )� N)�net)�sources�util)�NoDHCPLeaseError)�EphemeralDHCPv4)�upcloud� metadata_urlz'http://169.254.169.254/metadata/v1.json� � c @ sH e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z e
d
d� �Z dS )�DataSourceUpCloud�UpCloudFc C s� t jj| |||� || _t� | _tjtj|ddgi �t g�| _
| j
d | _ | j
j dt
�| _| j
j dt�| _| j
j dt�| _d | _d S )NZ
datasourcer r �retries�timeout�
wait_retry)r �
DataSource�__init__�distro�dict�metadatar Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfg�metadata_address�get�
MD_RETRIESr
�
MD_TIMEOUTr �
MD_WAIT_RETRYr �_network_config)�self�sys_cfgr �paths� r �'/usr/lib/python3.6/DataSourceUpCloud.pyr " s
zDataSourceUpCloud.__init__c C s t j� S )N)� uc_helperZ read_sysinfo)r r r r � _get_sysinfo2 s z DataSourceUpCloud._get_sysinfoc C s t j| j| j| j| jd�S )N)r Z sec_betweenr
)r! Z
read_metadatar r r r
)r r r r �_read_metadata5 s
z DataSourceUpCloud._read_metadatac C s� | j � \}}|sdS tjd|� | jr�yNtjd� tj� }tjd|� t| j|�� t j
tjd| j d�}W d Q R X W n6 t t
jfk
r� } zt jtt|�� dS d }~X nX ndy"tjd� t j
tjd| j d�}W n@ t
jk
�r } z t jtt|�� tjd|� dS d }~X nX || _|jd |�| jd
< |jd �| jd < |jd
�| jd
< |jd�| jd< |jdd�| jd< |jdd �| _|jdd �| _dS )NFz"Running on UpCloud. server_uuid=%szFinding a fallback NICz*Discovering metadata via DHCP interface %sz Reading from metadata service)Zlogfunc�msg�funcz7Discovering metadata without DHCP-configured networkingzENo DHCP-enabled interfaces available, unable to fetch metadata for %sZ instance_idz instance-idZhostnamezlocal-hostname�networkZ public_keysz public-keysZregion�defaultZavailability_zoneZ vendor_dataZ user_dataT)r" �LOG�info�perform_dhcp_setup�debug�cloudnetZfind_fallback_nicr r r Zlog_timer# r r ZInvalidMetaDataExceptionZlogexc�strZ
metadata_fullr r Zvendordata_rawZ userdata_raw)r Z
is_upcloudZ server_uuidZnicZmd�er r r � _get_data= sN
zDataSourceUpCloud._get_datac C s t j| j� �S )N)r Z instance_id_matches_system_uuidZget_instance_id)r r r r r �check_instance_idr s z#DataSourceUpCloud.check_instance_idc C s6 | j r | j S | jjd�}|s$td��tj|�| _ | j S )z�
Configure the networking. This needs to be done each boot,
since the IP and interface information might have changed
due to reconfiguration.
r& z/Unable to get network meta-data from server....)r r r � RuntimeErrorr! Zconvert_network_config)r Zraw_network_configr r r �network_configu s z DataSourceUpCloud.network_configN) �__name__�
__module__� __qualname__Zdsnamer* r r" r# r/ r0 �propertyr2 r r r r r s 5r c @ s e Zd ZdZdZdS )�DataSourceUpCloudLocalaz
Run in init-local using a DHCP discovery prior to metadata crawl.
In init-local, no network is available. This subclass sets up minimal
networking with dhclient on a viable nic so that it can talk to the
metadata service. If the metadata service provides network configuration
then render the network configuration for that instance based on metadata.
TN)r3 r4 r5 �__doc__r* r r r r r7 � s r7 c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_list� s r: )ZloggingZ cloudinitr r, r r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr Zcloudinit.sources.helpersr r! Z getLoggerr3 r( r r r r r r r7 ZDEP_FILESYSTEMZ DEP_NETWORKr9 r: r r r r �<module> s
r