3
\�me� � @ s� d dl Z d dlmZ d dlmZmZmZmZmZm Z d dl
m Z d dl m
Z
e je�ZdZded iZd Zd
Zd
ZG d d � d ej�Zd
d� Zdd� Zddd�ZeejffgZ dS )� N)�
exceptions)�dmi�net�sources�subp�
url_helper�util)�NoDHCPLeaseError)�EphemeralDHCPv4z http://169.254.169.254/api/v1� metadata_urlz /metadata� � c @ s@ e Zd ZdZdd� Zdd� Zdd� Zedd � �Ze d
d � �Z
d S )
�DataSourceNWCS�NWCSc C s� t jj| |||� || _t� | _tjtj|ddgi �t g�| _
| j
d | _ | j
j dt
�| _| j
j dt�| _| j
j dt�| _t j| _t j| _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 �UNSET�_network_configZDSMODE_NETWORKZdsmode)�selfZsys_cfgr �paths� r" �$/usr/lib/python3.6/DataSourceNWCS.pyr s
zDataSourceNWCS.__init__c C sv | j � }|d krtd��|| _|d | jd< |d | jd< |d | jd< |d | jd< |jdd �| _|jdd �| _d S )
Nzfailed to get metadataz instance-idz public-keys�networkZhostnamezlocal-hostnameZuserdataZ
vendordataT)� get_metadata� RuntimeErrorZ
metadata_fullr r Z userdata_rawZvendordata_raw)r Zmdr" r" r# � _get_data/ s zDataSourceNWCS._get_datac C s� yJt jd� t| jtj� dtd id�� t| j| j | j
| j d�S Q R X W n< t t
jttjfk
r� } zt jd|� � W Y d d }~X nX d S )Nz#Attempting to get metadata via DHCP�urlz/metadata/instance-id)ZifaceZconnectivity_url_data)r � sec_betweenr zDHCP failure: %s)�LOG�infor
r r Zfind_fallback_nic� BASE_URL_V1�
read_metadatar r r r r r ZProcessExecutionErrorr& r ZRequestException�error)r �er" r" r# r% B s$
zDataSourceNWCS.get_metadatac C s� t jd� | jd kr*t jdtj� tj| _| jtjkr<| jS | jd d sRtd��x@| jd d D ].}t|d �}|rbt j d|d |� ||d< qbW | jd | _| jS ) Nz Attempting network configurationz5Found None as cached _network_config, resetting to %sr$ �configz"Unable to get metadata from serverZ mac_addresszOverriding %s with %s�name)
r* �debugr Zwarningr r r r& �get_interface_namer+ )r �iZ
iface_namer" r" r# �network_config] s"
z DataSourceNWCS.network_configc C s dt jd�kS )Nr zsystem-manufacturer)r Z
read_dmi_datar" r" r" r# � ds_detectz s zDataSourceNWCS.ds_detectN) �__name__�
__module__� __qualname__Zdsnamer r' r% �propertyr5 � staticmethodr6 r" r" r" r# r s r c C s t j� }| |krd S |j| �S )N)r Zget_interfaces_by_macr )ZmacZ macs_to_nicr" r" r# r3 s r3 c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr" r" r# �get_datasource_list� s r= � c C s6 t j| |||d�}|j� s&td| ��tj|jj� �S )N)r r) r z unable to read metadata at %s)r Zreadurl�okr& r Z load_json�contents�decode)r( r r) r Zresponser" r" r# r- � s
r- )r> r> r ) ZloggingZrequestsr Z cloudinitr r r r r r Zcloudinit.net.dhcpr Zcloudinit.net.ephemeralr
Z getLoggerr7 r* r, r r r r r r r3 r= r- ZDEP_FILESYSTEMr< r" r" r" r# �<module> s
e