CasperSecurity
o
쑛hJ � @ s� d Z ddlZddlZddlZddlZddlZddlZddlmZm Z ddl
m Z ddl m
Z
mZmZmZmZmZ ddlmZ ddlmZ ddlmZ d Zdd
d �Zd d
� Z e dkree� � � Z e ee � dS dS )z(Debug network config format conversions.� N)�distros�safeyaml)�loggers)�eni�netplan�network_manager�
network_state�networkd� sysconfig)�DataSourceAzure)� openstack)�guestcust_utilz net-convertc C s� | s t jttd�} | jddtdddd� | jdd g d
�dd d � | jd
ddddd� | jdddd� tj�� D �dd� | jdddddd� | jd d d d � | jd d!g d"�dd#d � | S )$a Build or extend and arg parser for net-convert utility.
@param parser: Optional existing ArgumentParser instance representing the
subcommand which will be extended to support the args of this utility.
@returns: ArgumentParser with proper argument configuration.
)�prog� descriptionz-pz--network-data�PATHTz!The network configuration to read)�type�metavar�required�helpz-kz--kind)r �network_data.json�yaml�
azure-imds�
vmware-imcz&The format of the given network config)�choicesr r z-dz --directoryz directory to place output in)r r r z-Dz--distroc S s g | ] }|D ]}|�qqS � r )�.0�sublist�itemr r �A/usr/lib/python3/dist-packages/cloudinit/cmd/devel/net_convert.py�
<listcomp>G s
��z get_parser.<locals>.<listcomp>)r r z-mz--maczname,mac�appendz interface name to mac mapping)r �actionr z--debug�
store_truez enable debug logging to stderr.)r! r z-Oz
--output-kind)r r r r
�network-managerz!The network config format to emit) �argparse�ArgumentParser�NAME�__doc__� add_argument�openr �
OSFAMILIES�values)�parserr r r �
get_parser sd ��
������r- c C s� |j �d�s
| j d7 _ tj�|j �st�|j � |jr%tjt j
d� ntjt j d� |j rDi }|j D ]}|�
dd�\}}|||< q4nd }|j�� }|jdkrVt�|�}n]|jdkr|t�|�}d|v ri|�d�}|jr{tj�d�d t�|�d
g�� n7|jd kr�t j t � |�|d �}n'|jd
kr�t j!t � |�d dd�}n|jdkr�t"�#t"�$|jj%��}t"�&|d�}t'�(|j)�} | |j)i d �}
|j*dkr�tj+} |
j,�d�}nX|j*dkr�t-j+} |
j,�di �}d|d< |d dd � |d< ddg|d< n4|j*dk�rt.j+} |
j,�d�}n$|j*dk�rt/j+} |
j,�d�}n|j*dk�r#t0j+} |
j,�d�}nt1d��| |d �} t2j3|| d �}
|j�rGtj�d�d
d t�4|
�d
g�� tj�d
�d |j|jj%f d |j*|j f g�d � | j5|
|j d!� d S )"N�/)�level�,� r r �network�
z
Input YAML� r )�
known_macsr T)�&apply_network_config_for_secondary_ipsr Fr �postcmds� netplan_pathzdhcp-use-domainszipv6-mtu�featuresr r
r# zInvalid output_kind)�config)�rendererzInternal Statez"Read input format '%s' from '%s'.
z!Wrote output format '%s' to '%s'
)r �target)6� directory�endswith�os�path�isdir�makedirs�debugr �setup_basic_logging�logging�DEBUG�WARN�mac�split� network_data�read�kindr �convert_eni_datar � safe_load�get�sys�stderr�write�joinr �dumpsr �convert_net_json�json�loads�azure�6generate_network_config_from_instance_network_metadatar
�Config�
ConfigFile�name�%get_network_data_from_vmware_cust_cfgr �fetch�distro� output_kind�Renderer�renderer_configsr r r
r � RuntimeErrorr �parse_net_config_data�dump�render_network_state)r\ �argsr5 r �
iface_name� iface_mac�net_data�pre_nsr: �
distro_clsr_ �r_cls�r�nsr r r � handle_args` s�
�
��
�
�
��
�
����� rp �__main__)N) r' r$ rV rE r? rP r � cloudinitr r �
cloudinit.logr �
cloudinit.netr r r r r r
�cloudinit.sourcesr rX �cloudinit.sources.helpersr �$cloudinit.sources.helpers.vmware.imcr
r& r- rp �__name__�
parse_argsrg r r r r �<module> s(
B]
�