File: //usr/share/netplan/netplan/cli/commands/__pycache__/try_command.cpython-38.pyc
U
��~f� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m m Z ddl
mZ ddlZdZG dd� de j�ZdS )znetplan try command line� N)�
ConfigManager)� NetplanApply�x c sx e Zd Z� fdd�Zedd� �Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�
NetplanTryc sZ t � jdddd� d| _d | _d | _d | _d | _tj� dd�| _
tj � | j
dd d
�| _
d S ) N�tryzLTry to apply a new netplan config to running system, with automatic rollbackT)�
command_id� description�leafFZDBUS_TEST_NETPLAN_ROOT�/�run�netplanznetplan-try.ready)�super�__init__�configuration_changed�new_interfaces�_config_manager�
t_settings�t�os�environ�get�_rootdir�path�join�_netplan_try_stamp��self�� __class__� �6/usr/share/netplan/netplan/cli/commands/try_command.pyr ( s �zNetplanTry.__init__c C s | j st� | _ | j S �N)r r r r r r �config_manager5 s zNetplanTry.config_managerc C s" t j�| j�r t �| j� dS dS )NTF)r r �isfiler �remover r r r �clear_ready_stamp; s z NetplanTry.clear_ready_stampc C s* t j| jd ddd� t| jd��� d S )Nz /run/netplani� T)�mode�exist_ok�w)r �makedirsr �openr �closer r r r �touch_ready_stampA s z NetplanTry.touch_ready_stampc C sP | j jddd� | j jdttdd� | j jddd� | j| _| �� | �� d S ) Nz
--config-filezGApply the config file in argument in addition to current configuration.)�helpz --timeoutz=Maximum number of seconds to wait for the user's confirmation)�type�defaultr- z--statez0Directory containing previous YAML configuration)�parser� add_argument�int�DEFAULT_INPUT_TIMEOUT� command_try�func�
parse_args� run_commandr r r r r E s � ��zNetplanTry.runc
C sV | � � st�tj� �zz�tj
� � }t j
�|�| _| j�| j� t�tj| j� t�tj| j� | �� | �� t� jddd| jd� | �� | jj| jd� W n� t j
j k
r� t d� | � � Y n^ t j
j k
r� t d� Y n@ t k
�r$ } z t d| � t d� | � � W 5 d }~X Y nX W 5 | j�r@| j�| j� | �� | � � X d S )NTF�� run_generate�sync�
exit_on_error� state_dir)�timeoutz
Reverting.z
Configuration accepted.z
An error occurred: %s)!�
is_revertable�sys�exitr � EX_CONFIGr �resetr �cleanupr% �stdin�filenor �terminalZTerminalZsave�signal�SIGINT�_signal_handler�SIGUSR1�backup�setupr �
command_apply�stater, Zget_confirmation_inputr= �
InputRejected�print�revert�
InputAccepted� Exception)r �fd�er r r r4 S s4
zNetplanTry.command_tryc C s d}| j rd}| jj|d� d S )NFT)�backup_config_dir)� config_filer" rK )r rV r r r rK x s zNetplanTry.backupc C sd | j rZtj�ddd�}tj�| j ��d�}t�� }tj�|d�||��}| j� | j |i� d| _
d S )Nr
�etcr z.yamlz
{}.{}.yamlT) rW r r r �basename�rstrip�time�formatr" �addr )r Zdest_dirZ dest_nameZ dest_suffixZ dest_pathr r r rL ~ s zNetplanTry.setupc C s^ t �� }tj�|dd�}t�|� tjd|dd� | j� � t
� j ddd|d� t� |� d S )NrX r z /etc/netplanT)�
dirs_exist_okFr8 )
�tempfile�mkdtempr r r r) �shutil�copytreer" rQ r rM �rmtree)r �tempdirZconfdirr r r rQ � s
zNetplanTry.revertc C s | j �� d S r! )r" rC r r r r rC � s zNetplanTry.cleanupc C s� g }| j r|�| j � | jj|d� | jj| _t�d�| j�� g }i }|�| jj � |�| jj
� |� � D ]&\}}|rld|krld}|�||f� ql|r�|D ]\}}t d�||�� q�t d� dS dS ) a
Check if the configuration is revertable, if it doesn't contain bits
that we know are likely to render the system unstable if we apply it,
or if we revert.
Returns True if the parsed config is "revertable", meaning that we
can actually rely on backends to re-apply /all/ of the relevant
configuration to interfaces when their config changes.
Returns False if the parsed config contains options that are known
to not cleanly revert via the backend.
)� extra_configzNew interfaces: {}�
parameterszBreverting custom parameters for bridges and bonds is not supportedz{}: {}zL
Please carefully review the configuration and use 'netplan apply' directly.FT)
rW �appendr" �parser �logging�debugr\ �update�bridges�bonds�itemsrP )r re Zrevert_unsupportedZ multi_iface�ifname�settings�reasonr r r r> � s(
zNetplanTry.is_revertablec C s* |t jkrtj�� �n| jr&tj�� �d S r! )rG rJ r rF rR r rO )r �sig�framer r r rI � s
zNetplanTry._signal_handler)�__name__�
__module__� __qualname__r �propertyr" r% r, r r4 rK rL rQ rC r> rI �
__classcell__r r r r r & s
% -r )�__doc__r r[ ra rG r? r_ ri �netplan.configmanagerr �netplan.cli.utils�cli�utils�netplan.cli.commands.applyr Znetplan.terminalr r3 �NetplanCommandr r r r r �<module> s