3
\�me� � @ st d dl Z d dljjjZd dlmZmZ e je �Z
ddiZ dZ dZ
dZdZG dd � d ej�ZeejffgZd
d � ZdS ) � N)�sources�util� metadata_urlz'http://169.254.169.254/metadata/v1.json� � Tc s\ e Zd ZdZdd� Zedd�� fdd� Zdd � Zd
d � Zd d
� Z dd� Z
e dd� �Z � Z
S )�DataSourceDigitalOcean� DigitalOceanc C s� t jj| |||� || _t� | _tjtj|ddgi �t g�| _
| j � | j
d | _ | j
j
dt�| _| j
j
dt�| _| j
j
dt�| _| j
j
dt�| _d | _d S )NZ
datasourcer r �retries�timeout� use_ip4LL�
wait_retry)r �
DataSource�__init__�distro�dict�metadatar Z
mergemanydictZget_cfg_by_path�BUILTIN_DS_CONFIGZds_cfg�
_deprecate�metadata_address�get�
MD_RETRIESr �
MD_TIMEOUTr
�
MD_USE_IPV4LLr �
MD_WAIT_RETRYr �_network_config)�self�sys_cfgr �paths� r �,/usr/lib/python3.6/DataSourceDigitalOcean.pyr s
z DataSourceDigitalOcean.__init__N)�ci_pkl_version�returnc s t � j|� | j� d S )N)�super� _unpickler )r r )� __class__r r r# 4 s z DataSourceDigitalOcean._unpicklec C s t jdddd� d S )Nr z23.2z.Deprecated in favour of DataSourceConfigDrive.)Z
deprecatedZdeprecated_versionZ
extra_message)r Z deprecate)r r r r r 8 s z!DataSourceDigitalOcean._deprecatec C s t j� S )N)� do_helperZ read_sysinfo)r r r r � _get_sysinfo? s z#DataSourceDigitalOcean._get_sysinfoc C s� | j � \}}|sdS tjd|� d }| jr6tj| j�}tj| j| j | j
| j d�}|| _ |j
d|�| jd< |j
d|�| jd< |j
d�| jd< |j
d �| jd
< |j
d d �| jd
< |j
dd �| _|j
dd �| _|r�tj|� dS )NFz&Running on DigitalOcean. droplet_id=%s)r
Z sec_betweenr �
droplet_idz instance-idZhostnamezlocal-hostname�
interfacesZ public_keysz public-keysZregion�defaultZavailability_zoneZ vendor_dataZ user_dataT)r& �LOG�infor r% Zassign_ipv4_link_localr Z
read_metadatar r
r r �
metadata_fullr r Zvendordata_rawZ userdata_rawZdel_ipv4_link_local)r Zis_dor' Z
ipv4LL_nicZmdr r r � _get_dataB s.
z DataSourceDigitalOcean._get_datac C s t j| j� d�S )Nzsystem-serial-number)r Z instance_id_matches_system_uuidZget_instance_id)r r r r r �check_instance_idd s z(DataSourceDigitalOcean.check_instance_idc C sP | j r | j S | jjd�}tj|� |s.td��| jd d }tj||�| _ | 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.
r( z'Unable to get meta-data from server....Zdns� nameservers) r r r r* �debug� RuntimeErrorr, r% Z convert_network_configuration)r r( r/ r r r �network_configi s
z%DataSourceDigitalOcean.network_config)�__name__�
__module__� __qualname__Zdsnamer �intr# r r& r- r. �propertyr2 �
__classcell__r 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 r: )ZloggingZ&cloudinit.sources.helpers.digitaloceanr ZhelpersZ digitaloceanr% Z cloudinitr Z getLoggerr3 r* r r r r r r
r ZDEP_FILESYSTEMr9 r: r r r r �<module> s
e