3
\�me� � @ s� d dl Z d dlmZ d dlmZmZ d dlmZmZ d dl m
Z d dl m Z e j
e�ZdZG dd � d e j
�Zd
d � Zd d
� ZG dd� de�ZeejffeejejffgZdd� ZdS )� N)�List)�dmi�sources)�
EventScope� EventType)�
DataSourceEc2)�DataSourceHostnamezAlibaba Cloud ECSc s^ e Zd ZU dZdgZdZg Zee e dd� �Z
� fdd�Z dd d
�Z d d � Z
d
d� Z� ZS )�DataSourceAliYunZAliYunzhttp://100.100.100.200z
2016-01-01c C s dS )NzX-aliyun-ecs-metadata-token� )�selfr
r
�&/usr/lib/python3.6/DataSourceAliYun.py�imdsv2_token_put_header s z(DataSourceAliYun.imdsv2_token_put_headerc s, t t| �j|||� | jtj jtj� d S )N) �superr �__init__Zdefault_update_eventsr ZNETWORK�addr ZBOOT)r Zsys_cfgZdistro�paths)� __class__r
r r s zDataSourceAliYun.__init__Fc C s* | j jd�}d}|d kr d}d}t||�S )N�hostnameFzlocalhost.localdomainT)�metadata�getr )r ZfqdnZ
resolve_ipZ
metadata_onlyr Z
is_defaultr
r
r � get_hostname" s z DataSourceAliYun.get_hostnamec C s t | jjdi ��S )Nz public-keys)�parse_public_keysr r )r r
r
r �get_public_ssh_keys* s z$DataSourceAliYun.get_public_ssh_keysc C s t � rtjjS tjjS d S )N)�
_is_aliyun�EC2Z
CloudNamesZALIYUNZNO_EC2_METADATA)r r
r
r �_get_cloud_name- s z DataSourceAliYun._get_cloud_name)FFF)�__name__�
__module__� __qualname__ZdsnameZ
metadata_urlsZmin_metadata_versionZextended_metadata_versionsr �str�propertyr
r r r r �
__classcell__r
r
)r r r s
r c C s t jd�tkS )Nzsystem-product-name)r Z
read_dmi_data�ALIYUN_PRODUCTr
r
r
r r 4 s r c C s� g }x�| j � D ]|\}}t|t�r0|j|j� � qt|t�rF|j|� qt|t�r|jdg �}t|t�rv|j|j� � qt|t�r|j|� qW |S )Nz openssh-key) �items�
isinstancer �append�strip�list�extend�dictr )Z public_keys�keysZ_key_idZkey_body�keyr
r
r r 8 s
r c @ s e Zd ZdZdZdS )�DataSourceAliYunLocalay Datasource run at init-local which sets up network to query metadata.
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)r r r �__doc__Zperform_dhcp_setupr
r
r
r r, H s r, c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr
r
r �get_datasource_list[ s r/ )ZloggingZtypingr Z cloudinitr r Zcloudinit.eventr r Zcloudinit.sourcesr r r Z getLoggerr ZLOGr" r r r r, ZDEP_FILESYSTEMZ DEP_NETWORKr. r/ r
r
r
r �<module> s
$