File: //usr/share/netplan/netplan/cli/commands/__pycache__/set.cpython-38.pyc
U
��~f�" � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z ddl m
Z
dZddgZG dd � d e �ZdS )
znetplan set command line� N)�NetplanCommand)�
ConfigManagerz70-netplan-set�renderer�versionc sb e Zd Z� fdd�Zdd� Zdd� Zeefd�dd �Zd
d � Z d d
� Z
ddd�Z ddd�Z � Z
S )�
NetplanSetc s t � jdddd� d S )N�setzTAdd new setting by specifying a dotted key=value pair like ethernets.eth0.dhcp4=trueT)�
command_id� description�leaf)�super�__init__��self�� __class__� �./usr/share/netplan/netplan/cli/commands/set.pyr & s �zNetplanSet.__init__c C sR | j jdtdd� | j jdtdd� | j jdddd � | j| _| �� | �� d S )
N� key_valuezNThe nested key=value pair in dotted format. Value can be NULL to delete a key.)�type�helpz
--origin-hintz�Can be used to help choose a name for the overwrite YAML file. A .yaml suffix will be appended automatically.z
--root-dir�/zAOverwrite configuration files in this root directory instead of /)�defaultr )�parser� add_argument�str� command_set�func�
parse_args� run_commandr
r r r �run+ s
�
�
�zNetplanSet.runc C s0 t |t�r,t|�� �dgkr,|d d kr,dS dS )N�networkTF)�
isinstance�dict�list�keys)r �treer r r �
is_emtpy_yaml9 s (zNetplanSet.is_emtpy_yaml)�returnc s2 |� di �� t� }� D ]�}|tkr$q� � |g �}|d krXdd� t�|| j�D �}|� |< |D ]b}t}t�|| j�}|r�tj �
|�d d� }d||� � |�� |�iii}| � |� |i �|�||< q\qt � fdd�tD ���r*t
|�dkr�t|�d nt}tD ].} d| � � | �ii}
| � |� |i �|
�||< q�|�� S ) Nr c S s i | ]
}|d �qS �Nr )�.0�devr r r �
<dictcomp>J s z1NetplanSet.split_tree_by_hint.<locals>.<dictcomp>���c 3 s | ]}� � |�V qd S r( )�get)r) �key�r r r � <genexpr>X s z0NetplanSet.split_tree_by_hint.<locals>.<genexpr>� r )r- r" � GLOBAL_KEYS�
libnetplan�netplan_get_ids_for_devtype�root_dir�
FALLBACK_HINT�netplan_get_filename_by_id�os�path�basename�merge�any�lenr# �items) r �set_treeZsubtrees�devtypeZdevtype_content�netdef�hint�filenameZ netdef_treer. r% r r/ r �split_tree_by_hint> s, z NetplanSet.split_tree_by_hintc C s | j d k r t| j �dkr td��| j�dd�}t|�dkrBtd��|\}}| �|t�|��}| �|�r�t j
� dd�}| j r�t j
� | j || j d �}t j
�
|�r�t �|� n,t�t j
� | j |d
��}|D ]}t �|� q�d S | j |fg} | j d kr�| �|�} | D ] \}
} | �| |
d | j � q�d S ) Nr zInvalid/empty origin-hint�=r1 � zInvalid value specified�etc�netplanz.yamlz*.yaml)Z origin_hintr= � Exceptionr �split� parse_key�yaml� safe_loadr& r8 r9 �joinr5 �isfile�remove�globrD �
write_file) r rJ r. �valuer? r9 Z hint_pathZ
yaml_files�f�hintsrB Zsubtreer r r r b s,
zNetplanSet.command_setc C st |� d�s|dksd| }t�d|�}i }d}|}|D ]8}|�dd�}i }|t|�krZ|}|�||�}|d7 }q6|S )Nznetwork.r z (?<!\\)\.r1 z\.�.)�
startswith�rerJ �replacer= �
setdefault) r r. rS rJ r% �i�t�part�valr r r rK � s
zNetplanSet.parse_keyNc C s� |dkr g }|D ]x}||kr|t || t�rZt || t�rZ| �|| || |t|�g � q�|| dkrn||= q�|| ||< q|| ||< q|S )z9
Merges tree/dict 'b' into tree/dict 'a'
N)r! r" r; r )r �a�br9 r. r r r r; � s " zNetplanSet.merger c C s� t jdd�}tj�dd�}t�tj�|j|�� di i}tj�|||�}tj�|�r�t�|�j dkr�t
|d�� }t � |�} | d k r�| }W 5 Q R X | �
||�}
t�|
�} t�d�|| �� d| kr�t| d �� �d gkr�t�d
�|�� tj�|�r�t�|� n�d| k�rltj�|j||�} t
| d �� }t j| d d
d�}
|�|
� W 5 Q R X t�| � t�| |� t�| � nB| i k�r�t�d�|�� tj�|��r�t�|� nt d�|���d S )Nz netplan-set_)�prefixrG rH r r �rzWriting file {}: {}r z Empty YAML, deleting file {}zw+rF F)�indent�default_flow_stylez,Removed last key from YAML, deleting file {}zInvalid input: {}) �tempfile�TemporaryDirectoryr8 r9 rN �makedirs�namerO �stat�st_size�openrL rM r; r �
strip_tree�logging�debug�formatr# r$ rP �dump�writer3 �
netplan_parse�shutil�copy2rI )r r? rh �rootdirZtmprootr9 �configZabsprT �cZnew_tree�strippedZtmppZnew_yamlr r r rR � s<
zNetplanSet.write_file)N)r )�__name__�
__module__� __qualname__r r r& r r" rD r rK r; rR �
__classcell__r r r r r $ s $
r )�__doc__r8 rL re rX rm rs rQ �netplan.cli.utilsr �netplan.libnetplanr3 �netplan.configmanagerr r6 r2 r r r r r �<module> s