File: //usr/share/netplan/netplan/cli/commands/__pycache__/try_command.cpython-310.pyc
o
|hEc� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlm Z ddl
m m Z ddl
mZ ddlZddlmZ dZG 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 s` t � jdddd� d| _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�leafF�DBUS_TEST_NETPLAN_ROOT�/�run�netplanznetplan-try.ready)�super�__init__�configuration_changed�new_interfaces� config_file�_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 s
t| jd�| _ | j S )N)�prefix)r r r r r! r! r" �config_manager7 s zNetplanTry.config_managerc C s" t j�| j�rt �| 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_stampC 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 G s ��� zNetplanTry.runc
C sp | � � s
t�tj� z�zDtj�� }tj� |�| _
| j
� | j � t
�
t
j| j� t
�
t
j| j� | �� | �� t� jddd| jd� | �� | j
j| jd� W n@ tjjyb td� | �� Y n. tjj yp td� Y n t y� } ztd| � td� | �� W Y d }~nd }~ww W | j
r�| j
� | j � | � � | � � d S | j
r�| j
� | j � | � � | � � w )NTF�� run_generate�sync�
exit_on_error� state_dir)�timeoutz
Reverting.z
Configuration accepted.z
An error occurred: %s)!�
is_revertable�sys�exitr � EX_CONFIG�stdin�filenor
�terminal�Terminalr �saver �signal�SIGINT�_signal_handler�SIGUSR1�backup�setupr �
command_apply�stater. �get_confirmation_inputr? �
InputRejected�print�revert�
InputAccepted� Exception�reset�cleanupr' )r �fd�er! r! r" r6 U sB
��� �
zNetplanTry.command_tryc C s d}| j rd}| jj|d� d S )NFT)�backup_config_dir)r r$ rM )r r[ r! r! r" rM z s zNetplanTry.backupc C sd | j r-tj�ddd�}tj�| j ��d�}t�� }tj�|d�||��}| j� | j |i� d| _
d S )Nr �etcr
z.yamlz
{}.{}.yamlT) r r r r �basename�rstrip�time�formatr$ �addr )r �dest_dir� dest_name� dest_suffix� dest_pathr! r! r" rN � s
zNetplanTry.setupc C s^ t �� }tj�|dd�}t�|� tjd|dd� | j� � t
� j ddd|d� t� |� d S )Nr\ r
z /etc/netplanT)�
dirs_exist_okFr: )
�tempfile�mkdtempr r r r+ �shutil�copytreer$ rT r rO �rmtree)r �tempdir�confdirr! r! r" rT � s
zNetplanTry.revertc C s | j �� d S �N)r$ rX r r! r! r" rX � s zNetplanTry.cleanupc
C s� g }| j r |�| j � d}z | jj|d�}W n tjy4 } zt�|� t� t
j � W Y d}~nd}~ww g }i }|� |j
� |� |j� |�� D ]}|jsXd}|�|j|f� qI|rp|D ] \}}td�||�� q]td� 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.
N)� extra_configzBreverting custom parameters for bridges and bonds is not supportedz{}: {}zL
Please carefully review the configuration and use 'netplan apply' directly.FT)r �appendr$ �parse�utils�
config_errors�logging�errorrA rB r rC �update�bridges�bonds�values�is_trivial_compound_itf�idrS r` ) r ro �np_staterZ �revert_unsupported� multi_iface�itf�reason�ifnamer! r! r" r@ � s4
�� � zNetplanTry.is_revertablec C s( |t jkr
tj�� �| jrtj�� �d S rn )rI rL r
rF rU r rR )r �sig�framer! r! r" rK � s
�zNetplanTry._signal_handler)�__name__�
__module__� __qualname__r �propertyr$ r' r. r r6 rM rN rT rX r@ rK �
__classcell__r! r! r r" r ' s
% -r )�__doc__rt r r_ ri rI rA rg �netplan.configmanagerr �netplan.cli.utils�clirr �netplan.cli.commands.applyr �netplan.terminalr
�netplan.libnetplan�
libnetplanr5 �NetplanCommandr r! r! r! r" �<module> s