File: //usr/share/netplan/netplan/__pycache__/configmanager.cpython-38.pyc
U
ci�_
. � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZG dd� de�Z G dd� de
�Z dS )z netplan configuration manager� Nc @ s2 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#� �Zg fd$d%�Zd&d'� Zd;d)d*�Zd+d,� Zd-d.� Zd/d0� Zd<d2d3�Z d4d5� Z d6d7� Z d8d9� Z d:S )=�
ConfigManager�/c C sP || _ tjdd�| _tj�| jd�| _tj�| jd�| _|| _ i | _
t � | _ d S )NZnetplan_)�prefix�etc�run)
r �tempfile�mkdtemp�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 s
| j d S )Nr �r r r r r r A s zConfigManager.ovs_portsc C s
| j d S )N� openvswitchr& r r r r r' E s zConfigManager.openvswitchc C s
| j d S )Nr r& r r r r r I s zConfigManager.ethernetsc C s
| j d S )Nr r& r r r r r M s zConfigManager.modemsc C s
| j d S )Nr r& r r r r r Q s zConfigManager.wifisc C s
| j d S )Nr r& r r r r r U s zConfigManager.bridgesc C s
| j d S )Nr r& r r r r r Y s zConfigManager.bondsc C s
| j d S )Nr! r& r r r r r! ] s zConfigManager.tunnelsc C s
| j d S )Nr" r& r r r r r" a s zConfigManager.vlansc C s
| j d S )N�versionr& r r r r r( e s zConfigManager.versionc C s
| j d S )N�rendererr& r r r r r) i s zConfigManager.rendererc C s | � | j�S r )�
strip_treer r r r r �treem 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* q 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 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 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 r7 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/netplanr4 �%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/networkrH rJ z8Something really bad happened while reverting config: {}zZYou should verify the netplan YAML in /etc/netplan and probably run 'netplan apply' again.���)r/ r r
�unlinkr@ r r �exists�shutil�rmtreer r rK � 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 )rP rQ r r r r r �cleanup� s zConfigManager.cleanupc C s t �||� d S r )rP �copy)r �src�dstr r r rF � s zConfigManager._copy_fileFc C s2 zt �||� W n tk
r, |r&n� Y nX d S r )rP �copytree�FileNotFoundError)r rZ r[ rI r r r rK � 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 r/ �get�listr< �popr> r? r@ r rG )
r �orig�newr r^ 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 )Nr_ r` ) r rb r< rc r> r? r@ r rG )r rd re r rf rg rh 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�|�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)ZLoaderr r' r r r r r r! r" r( r) z!Error while loading {}, aborting.� )r �openrA �loadZ CSafeLoaderra ri r r rj r r r r r r! r" �IOErrorZ YAMLErrorr> rS r@ rT rU )r rC r �fZ yaml_datar re r r r r= sR
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+ � staticmethodr* rE rG rL rW rX rF rK ri rj r= r r r r r sV
+
r c @ s e Zd ZdZdS )�ConfigurationErrorzL
Configuration could not be parsed or has otherwise failed to apply
N)rp rq rr �__doc__r r r r ru 4 s ru ) rv r9 r> r
rP rT r rA �objectr rR ru r r r r �<module> s