3
\�met � @ s� d dl Z d dlZd dlmZ d dlmZmZ d dlmZ d dl m
Z
m Z e j e
�ZG dd� dej�ZeZeejffgZdd � ZdS )
� N)� b64decode)�dmi�sources)�DataSourceHostname)� SERIAL_PORT�Cepkoc @ sR e Zd ZdZdZdd� Zedd� �Zdd� Zd d
� Z dd d
�Z
dd� Z dd� Z dS )�DataSourceCloudSigmaz�
Uses cepko in order to gather the server context from the VM.
For more information about CloudSigma's Server Context:
http://cloudsigma-docs.readthedocs.org/en/latest/server_context.html
Z
CloudSigmac C s$ t � | _d| _tjj| |||� d S )N� )r �cepko�ssh_public_keyr �
DataSource�__init__)�selfZsys_cfgZdistro�paths� r �*/usr/lib/python3.6/DataSourceCloudSigma.pyr
s z DataSourceCloudSigma.__init__c C s> t jd� tjd�} | s&t jd� dS t jd| � d| j� kS )z
Uses dmi data to detect if this instance of cloud-init is running
in the CloudSigma's infrastructure.
z0determining hypervisor product name via dmi datazsystem-product-namez-system-product-name not available in dmi dataFzdetected hypervisor as %sZ
cloudsigma)�LOG�debugr Z
read_dmi_data�lower)Zsys_product_namer r r � ds_detect! s
z DataSourceCloudSigma.ds_detectc C s� d}y| j j� j}|d }W n tk
r: tjd� dS X | j|jd�g�| _|t j
kr^dS |jdd�j d�}|jd d�| _ d |kr�t
| j �| _ d
|jd i �kr�|d d
| _|| _|d | _d
S )zj
Metadata is the whole server context and /meta/cloud-config is used
as userdata.
N�metaz+CloudSigma: Unable to read from serial portFzcloudinit-dsmode�
base64_fieldsr �,zcloudinit-user-data� cloudinitZ vendor_datar T)r
�all�result� Exceptionr r Z_determine_dsmode�get�dsmoder ZDSMODE_DISABLED�splitZ userdata_rawr Zvendordata_raw�metadatar )r r Zserver_contextZ server_metar r r r � _get_data0 s(
z DataSourceCloudSigma._get_datac C s dt S )z/Return the subplatform metadata source details.z
cepko (%s))r )r r r r �_get_subplatformR s z%DataSourceCloudSigma._get_subplatformFc C sD t jd| jd �r&| jd dd� }n| jd jd�d }t|d�S ) z�
Cleans up and uses the server's name if the latter is set. Otherwise
the first part from uuid is being used.
z^[A-Za-z0-9 -_\.]+$�nameN�= �uuid�-r F)�re�matchr r r )r ZfqdnZ
resolve_ipZ
metadata_only�retr r r � get_hostnameV s z!DataSourceCloudSigma.get_hostnamec C s | j gS )N)r )r r r r �get_public_ssh_keysa s z(DataSourceCloudSigma.get_public_ssh_keysc C s
| j d S )Nr% )r )r r r r �get_instance_idd s z$DataSourceCloudSigma.get_instance_idN)FFF)
�__name__�
__module__� __qualname__�__doc__Zdsnamer
� staticmethodr r! r" r* r+ r, r r r r r s "
r c C s t j| t�S )zK
Return a list of data sources that match this set of dependencies
)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listr s r3 )Zloggingr' �base64r r r r Zcloudinit.sourcesr Z$cloudinit.sources.helpers.cloudsigmar r Z getLoggerr- r r r ZDataSourceCloudSigmaNetZDEP_FILESYSTEMr2 r3 r r r r �<module> s
W