File: //lib/python3.6/site-packages/cloudinit/sources/__pycache__/DataSourceOVF.cpython-36.opt-1.pyc
3
�q�h�) � @ s� d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZm Z m
Z
m Z ej e
�ZG dd� de j�ZG dd� de�Zd d
d �Zd d
� Zdd� Zd dd�Zdd� Zdd� Zdd� ZG dd� de�Zee jffee je j fffZ dd � Z d d � Z dS )!� N)�minidom)�log)�safeyaml�sources�subp�utilc @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z e
dd� �Z dS )�
DataSourceOVFZOVFc C sF t jj| |||� d | _tjj|jd�| _d | _i | _ d| _
d | _ d S )NZovf�/�file://)r r
) r �
DataSource�__init__�seed�os�path�join�seed_dir� environment�cfg�supported_seed_starts�_network_config)�self�sys_cfg�distro�paths� r �#/usr/lib/python3.6/DataSourceOVF.pyr s zDataSourceOVF.__init__c C s t jj| �}d|| jf S )Nz %s [seed=%s])r r �__str__r
)r �rootr r r r # s zDataSourceOVF.__str__c C s� g }i }d}d}ddi}t | jj�\}}|r^tjj| jj|�}t|�\}}} || _|j|� npdt fdt
fg}
d } x |
D ]\} } | � }|rxP qxW |r�t|d�\}}} || _d|kr�|d r�|d | _ |j| � t |�dkr�d S d
|ko�|d
�rp|d
}
d }x"| j
D ]}|
j|��r|}P �qW |�s8tjd |
| � d S tj|
d d �\}}}tjd
|
� tj||g�}|j|
� tj||g�}dj|�| _|| _|| _|| _| | _dS )N� z instance-idz iid-dsovfzcom.vmware.guestInfoZisoTznetwork-configr F�seedfromz Seed from %s not supported by %s)Ztimeoutz Using seeded cache data from %s�,)� get_ovf_envr r r r r �read_ovf_environmentr �append�transport_vmware_guestinfo�transport_iso9660r �lenr �
startswith�LOG�debugr Z read_seededZ
mergemanydictr
�metadataZ userdata_rawZvendordata_rawr )r �found�md�udZvdZdefaultsZseedfile�contentsr
r Znp�nameZ transfuncr Z seedfound�protoZmd_seedr r r � _get_data' s^
zDataSourceOVF._get_datac C s
d| j S )Nzovf (%s))r
)r r r r �_get_subplatformi s z DataSourceOVF._get_subplatformc C s0 d| j krg S | j d }t|t�r&|S |gS d S )Nz public-keys)r* �
isinstance�list)r Zpksr r r �get_public_ssh_keysl s
z!DataSourceOVF.get_public_ssh_keysc C s | j S )N)r )r r r r �get_config_objx s z DataSourceOVF.get_config_objc C s | j S )N)r )r r r r �network_config{ s z DataSourceOVF.network_configN) �__name__�
__module__� __qualname__Zdsnamer r r1 r2 r5 r6 �propertyr7 r r r r r s B r c @ s e Zd Zdd� ZdS )�DataSourceOVFNetc C s, t j| |||� tjj|jd�| _d| _d S )Nzovf-net�http://�https://)r= r> )r r r r r r r )r r r r r r r r � s zDataSourceOVFNet.__init__N)r8 r9 r: r r r r r r< � s r< Fc C s t | �}i }i }d }dg}ddddg}dg}x�|j� D ]�\} }
| dkrJd} | |kr\|
|| < q6| |krn|
|| < q6| |kr�|r�y$tj|
j� �} t| �jd�|| < W q� tk
r� tj d � Y q�X q6| d
kr6ytj|
j� �}W q6 tk
r� |
j� }Y q6X q6W |||fS ) NZpasswordr zlocal-hostnamez public-keysz instance-idznetwork-configZhostnameZnetworkz%Ignore network-config in wrong formatz user-data)
�get_properties�items�base64Z b64decode�encode�safeload_yaml_or_dict�get� Exceptionr( r) ) r. Z read_network�propsr, r r- Z cfg_propsZmd_propsZ
network_propsZprop�valr7 r r r r" � s4
r" c C sf d}x\|D ]T}t jj| |�}t jj|�r
ytj|�}||fS tk
r\ tjtd|� Y q
X q
W dS ) N� ovf-env.xml� ovf_env.xml� OVF_ENV.XML� OVF-ENV.XMLzFailed loading ovf file %sF)rH rI rJ rK )NF) r r r �isfiler Z load_filerE Zlogexcr( )�dirnameZ env_namesZfnameZfull_fnr. r r r r! � s
r! c C s� | sdS t | t�s td| ��tjj| �} | jd�rB| jd�d
} | jd�r\| jd�d } n| j d�dkrndS | svdS d}tj
j d|�}t j
|�}|j| �d k S ) z�Test if devname matches known list of devices which may contain iso9660
filesystems.
Be helpful in accepting either knames (with no leading /dev/) or full path
names, but do not allow paths outside of /dev/, like /dev/foo/bar/xxx.
Fz Unexpected input for devname: %sz/dev/� r r z^(sr[0-9]+|hd[a-z]|xvd.*)ZCLOUD_INIT_CDROM_DEV_REGEXN���rO )r3 �str�
ValueErrorr r �normpathr' � partition�split�count�environrD �re�compile�match)ZdevnameZ
default_regexZ
devname_regexZcdmatchr r r �maybe_cdrom_device� s"
rZ Tc
C s� t j� }xT|j� D ]H\}}|d }|dkr0| r0qt|�s:q|d }t|�\}}|dk r|S qW | rhd}nd }dd� t j|r�d| nd �D �} xV| D ]N}yt j|t|d�\}}W n$ t jk
r� tj d |� w�Y nX |dk r�|S q�W d S )
N�fstypeZiso9660Z
mountpointFc S s g | ]}t |�r|�qS r )rZ )�.0�devr r r �
<listcomp>� s z%transport_iso9660.<locals>.<listcomp>zTYPE=%s)�mtypez%s not mountable as iso9660)
r �mountsr@ rZ r! Zfind_devs_withZmount_cbZMountFailedErrorr( r) )
Z require_isor` r] �infor[ ZmpZ_fnamer. r_ Zdevsr r r r% � s2
r% c C s� d} d }t j| �s|S | dg}y(t j |�\}}|r6|S tjd||� W nH t jk
r� } z*|jdkr~tjd| |j� tj|� W Y d d }~X nX |S )Nzvmware-rpctoolzinfo-get guestinfo.ovfEnvz%cmd %s exited 0 with empty stdout: %srN z%s exited with code %d)r Zwhichr( r) ZProcessExecutionErrorZ exit_codeZwarning)ZrpctoolZ not_found�cmd�outZ_err�er r r r$ s
r$ c C s6 g }| j � s|S x | jD ]}||�r|j|� qW |S )N)�
hasChildNodesZ
childNodesr# )ZnodeZ filter_func�retZchildr r r �
find_child s rg c C s� t j| �}|jjdkr td��|jj� s0td��d}t|jdd� �}t|�dkrXtd��i }t|d d d� �}x4|D ],}|jj |d
�j
}|jj |d �j
}|||< qtW |S ) NZ EnvironmentzNo Environment NodezNo Child Nodesz)http://schemas.dmtf.org/ovf/environment/1c S s
| j dkS )NZPropertySection)� localName)�nr r r �<lambda>/ s z get_properties.<locals>.<lambda>r zNo 'PropertySection'sc S s
| j dkS )NZProperty)rh )ri r r r rj 7 s �key�value) r Z parseStringZdocumentElementrh �XmlErrorre rg r& Z
attributesZgetNamedItemNSrl ) r. ZdomZenvNsURIZ propSectionsrF Z propElems�elemrk rG r r r r? ! s$
r? c @ s e Zd ZdS )rm N)r8 r9 r: r r r r rm B s rm c C s t j| t�S )N)r Zlist_from_depends� datasources)Zdependsr r r �get_datasource_listN s rp c C s | si S t j| �S )z�
The meta data could be JSON or YAML. Since YAML is a strict superset of
JSON, we will unmarshal the data as YAML. If data is None then a new
dictionary is returned.
)r �load)�datar r r rC R s rC )F)T) rA r rW Zxml.domr Z cloudinitr Zloggingr r r r Z getLoggerr8 r( r r r< r" r! rZ r% r$ rg r? rE rm ZDEP_FILESYSTEMZ DEP_NETWORKro rp rC r r r r �<module> s(
j
)
'
!