CasperSecurity
o
쑛h � @ s d dl Z d dlmZ d dlm m mZ d dlmZmZm Z m
Z
e � e �Z
ddddde
�� e �� d e �� d
f d �ZG d d
� d
ej�ZeejffgZdd� Ze dkr�d dlZe�� sged� e�d� ejdd�jZe� eed ed ed ed ed �Z e d Z e� � Z dS dS )� N)�Tuple)�sources�stages�util�versionzhttp://169.254.169.254� �
� z"Cloud-Init/%s - OS: %s Variant: %s�system�variant)�url�retries�timeout�wait�
user-agentc s� e Zd ZU dZejjd Zeedf e d< � fdd�Z
e dd� �Z d d
� Z
d d � Zd
d� Zdd� Zedd� �Zedd� �Z� ZS )�DataSourceVultr�Vultr)zstartup-script.�sensitive_metadata_keysc s< t t| ��|||� t�t�|ddgi �tg�| _d | _d S )N�
datasourcer ) �superr �__init__r �
mergemanydict�get_cfg_by_path�BUILTIN_DS_CONFIG�ds_cfg�netcfg)�self�sys_cfg�distro�paths�� __class__� �C/usr/lib/python3/dist-packages/cloudinit/sources/DataSourceVultr.pyr % s ��
zDataSourceVultr.__init__c C s t �� S �N)�vultr�is_vultrr" r" r" r# � ds_detect/ s zDataSourceVultr.ds_detectc C sT t �d� | �� | _| jd | _| �| j� t �d| jd � t �d| jd � dS )NzMachine is a Vultr instance� user-dataz SUBID: %sz instance-idz Hostname: %s�local-hostnameT)�LOG�debug� get_metadata�metadata� userdata_raw�get_datasource_data�r r" r" r# � _get_data4 s
zDataSourceVultr._get_datac C sv d|v rt �|d � |d | _nt �|d �| _|d | _| jd dkr*d| jd< |d | _| jdkr9d | _d S d S )N�cloud_interfaces�
interfaces� vendor-datar) � �guestr( )r% �add_interface_namesr �generate_network_config�vendordata_rawr- r. )r �mdr" r" r# r/ F s
�z#DataSourceVultr.get_datasource_datac C s> t j| j| jd | jd | jd | jd | jd | j�� d�S )Nr r r
r r )�tmp_dir)r% r, r r �get_tmp_exec_pathr0 r" r" r# r, [ s �z DataSourceVultr.get_metadatac C s. t �� sdS t �� r dS t �� d }t�|�S )NF�subid)r% r& � is_baremetal� get_sysinfor � instance_id_matches_system_uuid)r r r= r" r" r# �check_instance_idg s
z!DataSourceVultr.check_instance_idc C s d S r$ r" r0 r" r" r# � launch_indexs s z DataSourceVultr.launch_indexc C s | j S r$ )r r0 r" r" r# �network_configw s z DataSourceVultr.network_config)�__name__�
__module__� __qualname__�dsnamer �
DataSourcer r �str�__annotations__r � staticmethodr' r1 r/ r, rA �propertyrB rC �
__classcell__r" r" r r# r s
�
r c C s t �| t�S r$ )r �list_from_depends� datasources)�dependsr" r" r# �get_datasource_list� s rQ �__main__z Machine is not a Vultr instance� T)�reporterr r r
r r r4 )!�logging�typingr � cloudinit.sources.helpers.vultrr �helpersr% � cloudinitr r r � getLoggerrD r* �version_string� system_infor rH r �DEP_FILESYSTEMrO rQ �sysr& �print�exit�Initr r, r: �configr? �sysinfor" r" r" r# �<module> sJ
���
a�
� �