File: //proc/self/root/usr/share/netplan/netplan/__pycache__/configmanager.cpython-38.pyc
U
��~f1 � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZG dd� de�Z G dd� de
�Z dS )z netplan configuration manager� Nc @ sR e Zd Zdi fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Zed d
� �Z edd� �Z
edd� �Z edd� �Z edd� �Z
edd� �Zedd� �Zedd� �Zed d � �Zed d � �Zed d!� �Zed"d#� �Zed$d%� �Zed&d'� �Zg fd(d)�Zd*d+� ZdAd-d.�Zd/d0� Zd1d2� Z d3d4� Z d5d6� Z dBd8d9�Z d:d;� Z d<d=� Z!d>d?� Z"d@S )C�
ConfigManager�/c C sP || _ tjdd�| _tj�| jd�| _tj�| jd�| _|| _ i | _
t � | _ d S )NZnetplan_)�prefix�etc�run)
r �tempfileZmkdtemp�tempdir�os�path�join�temp_etc�temp_run� extra_files�config�set�new_interfaces)�selfr r � r �+/usr/share/netplan/netplan/configmanager.py�__init__ s zConfigManager.__init__c C s
| j d S )N�network)r �r r r r r ( s zConfigManager.networkc C sh i }|� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |� | j� |S �N) �update� ovs_ports� ethernets�modems�wifis�bridges�bonds�tunnels�vlans�r �
interfacesr r r r# , s zConfigManager.interfacesc C s, i }|� | j� |� | j� |� | j� |S r )r r r r r" r r r �physical_interfaces9 s
z!ConfigManager.physical_interfacesc C s8 i }|� | j� |� | j� |� | j� |� | j� |S r )r r r r r! r" r r r �virtual_interfacesA s z ConfigManager.virtual_interfacesc C s
| j d S )Nr �r r r r r r K s zConfigManager.ovs_portsc C s
| j d S )N� openvswitchr& r r r r r' O s zConfigManager.openvswitchc C s
| j d S )Nr r& r r r r r S s zConfigManager.ethernetsc C s
| j d S )Nr r& r r r r r W s zConfigManager.modemsc C s
| j d S )Nr r& r r r r r [ s zConfigManager.wifisc C s
| j d S )Nr r& r r r r r _ s zConfigManager.bridgesc C s
| j d S )Nr r& r r r r r c s zConfigManager.bondsc C s
| j d S )Nr r& r r r r r g s zConfigManager.tunnelsc C s
| j d S )Nr! r& r r r r r! k s zConfigManager.vlansc C s
| j d S )N�
nm-devicesr& r r r r �
nm_deviceso s zConfigManager.nm_devicesc C s
| j d S )N�versionr& r r r r r* s s zConfigManager.versionc C s
| j d S )N�rendererr& r r r r r+ w s zConfigManager.rendererc C s | � | j�S r )�
strip_treer r r r r �tree{ s zConfigManager.treec C sD i }| � � D ]2\}}t|t�r(t�|�}|ddi fkr |||< q |S )zclear empty branches� N)�items�
isinstance�dictr r, )�dataZnew_data�k�vr r r r, s
zConfigManager.strip_treec
s� i � dD ]4}t � tj�| j|dd��D ]}|� tj�|�< q&q� fdd�t� �� �D �}i i i i i i i i i i ddd� | jd< |D ]}| � |� q�|D ]}| j
| � |�O _
q�t � d �
tj| jd
d ��� dS ) z�
Parse all our config files to return an object that describes the system's
entire configuration, so that it can later be interrogated.
Returns a dict that contains the entire, collated and merged YAML.
)�libr r �netplanz*.yamlc s g | ] }� | �qS r r )�.0�name�Znames_to_pathsr r �
<listcomp>� s z'ConfigManager.parse.<locals>.<listcomp>N) r r' r r r r r r r! r( r* r+ r zMerged config:
{}F)Zdefault_flow_style)�globr r
r r �basename�sorted�keysr �_merge_yaml_configr �logging�debug�format�yaml�dumpr- )r Z extra_configZyaml_dir� yaml_file�filesr r9 r �parse� s. � zConfigManager.parsec C s* |D ]}| � ||| � q| j�|� d S r )�
_copy_filer r )r Z config_dictZ config_filer r r �add� s zConfigManager.addTc C s� |r(| � tj�| jd�tj�| jd�� | j tj�| jd�tj�| jdd�dd� | j tj�| jd�tj�| jd d
�dd� d S ) Nz etc/netplanr6 �%run/NetworkManager/system-connectionsZNetworkManagerzsystem-connectionsT)�
missing_ok�run/systemd/networkZsystemdr )�
_copy_treer r
r r r r
)r Zbackup_config_dirr r r �backup� s ���zConfigManager.backupc
C s z�t | j�D ] }t�| j| � | j|= q d�| j�}d�| j�}tj�|�r|t� tj�
| j d�� | � |tj�
| j d�� tj�|�r�t� tj�
| j d�� | � |tj�
| j d�� W nF t
k
r� } z(t�d�|�� t�d� t�d� W 5 d }~X Y nX d S )Nz${}/NetworkManager/system-connectionsz{}/systemd/networkrJ rL z8Something really bad happened while reverting config: {}zZYou should verify the netplan YAML in /etc/netplan and probably run 'netplan apply' again.���)r1 r r �unlinkrB r
r
�exists�shutil�rmtreer r rM � Exceptionr@ �error�sys�exit)r Z
extra_fileZ temp_nm_pathZtemp_networkd_path�er r r �revert� s(
� �
zConfigManager.revertc C s t �| j� d S r )rR rS r r r r r �cleanup� s zConfigManager.cleanupc C s& z | � � W n tk
r Y nX d S r )rZ �FileNotFoundErrorr r r r �__del__� s zConfigManager.__del__c C s t �||� d S r )rR �copy)r �src�dstr r r rH � s zConfigManager._copy_fileFc C s2 zt �||� W n tk
r, |r&n� Y nX d S r )rR �copytreer[ )r r^ r_ rK r r r rM � s zConfigManager._copy_treec
C s� t � }t� }d|krD|�d�D ]$\}}|i d�||< |i d�||< q t|�� �}|D ]\}|�|�} ||kr�t�d�||�� || � | � qTt�d�||�� | ||< |�
|� qT|S )N�ports)Zpeerr' �{} exists in {}�{} not found in {}) r r1 �get�listr> �popr@ rA rB r rI )
r �orig�newr ra Zp1Zp2�changed_ifaces�ifname�ifacer r r �_merge_ovs_ports_config� s
z%ConfigManager._merge_ovs_ports_configc C sx t � }t|�� �}|D ]\}|�|�}||krNt�d�||�� || �|� qt�d�||�� |||< |�|� q|S )Nrb rc ) r re r> rf r@ rA rB r rI )r rg rh r ri rj rk r r r �_merge_interface_config
s
z%ConfigManager._merge_interface_configc C s t � }�z�t|����}tj|tjd�}d }|d k r<|�d�}|�r�d|krv| �| j|�d��}||O }|�d�| jd< d|kr�| � | j
|�d��}||O }d|kr�| � | j |�d��}||O }d|kr�| � | j |�d��}||O }d|k�r| � | j
|�d��}||O }d|k�r.| � | j|�d��}||O }d |k�rT| � | j|�d ��}||O }d
|k�rz| � | j|�d
��}||O }d |k�r�| � | j|�d ��}||O }d |k�r�|�d �| jd < d
|k�r�|�d
�| jd
< W 5 Q R X |W S ttjfk
�r t�d�|�� t�d� Y nX d S )N)�Loaderr r' r r r r r r r! r( r* r+ z!Error while loading {}, aborting.� )r �openrC �loadZ CSafeLoaderrd rl r r rm r r r r r r r! r) �IOErrorZ YAMLErrorr@ rU rB rV rW )r rE r �fZ yaml_datar rh r r r r? sX
z ConfigManager._merge_yaml_configN)T)F)#�__name__�
__module__� __qualname__r �propertyr r# r$ r% r r' r r r r r r r! r) r* r+ r- � staticmethodr, rG rI rN rY rZ r\ rH rM rl rm r? r r r r r s`
,
r c @ s e Zd ZdZdS )�ConfigurationErrorzL
Configuration could not be parsed or has otherwise failed to apply
N)rt ru rv �__doc__r r r r ry M s ry ) rz r; r@ r rR rV r rC �objectr rT ry r r r r �<module> s 2