Current File : //usr/share/netplan/netplan/__pycache__/configmanager.cpython-36.pyc
3
�Bd]G" � @ s` d Z ddlZddlZddlZddlZddlZddlZddlZG dd� de�Z G dd� de
�Z dS )z netplan configuration manager� Nc @ s� e Zd Zdi fdd�Zedd� �Zedd� �Zedd � �Zed
d � �Zed d
� �Z edd� �Z
edd� �Z edd� �Z g fdd�Z
dd� Zd)dd�Zdd � Zd d � Zd d � Zd*d"d#�Zd$d%� Zd&d'� Zd(S )+�
ConfigManager�/c C sP || _ tjdd�| _tjj| jd�| _tjj| 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 sD i }|j | j� |j | j� |j | j� |j | j� |j | j� |S )N)�update� ethernets�wifis�bridges�bonds�vlans)r �
interfacesr r r r , s zConfigManager.interfacesc C s i }|j | j� |j | j� |S )N)r r r )r r r r r �physical_interfaces6 s z!ConfigManager.physical_interfacesc C s
| j d S )Nr )r )r r r r r = s zConfigManager.ethernetsc C s
| j d S )Nr )r )r r r r r A s zConfigManager.wifisc C s
| j d S )Nr )r )r r r r r E s zConfigManager.bridgesc C s
| j d S )Nr )r )r r r r r I s zConfigManager.bondsc C s
| j d S )Nr )r )r r r r r M s zConfigManager.vlansc s� i � x@dD ]8}x2t j tjj| j|dd��D ]}|� tjj|�< q*W q
W � fdd�t� j� �D �}i i i i i d�| jd < x|D ]}| j |� q|W x |D ]}| j
| j |�O _
q�W t j d
j
tj| jd d ��� d
S )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)�names_to_pathsr r �
<listcomp>e s z'ConfigManager.parse.<locals>.<listcomp>)r r r r r r zMerged config:
{}F)Zdefault_flow_styleN)r r r )�globr r
r r �basename�sorted�keysr �_merge_yaml_configr �logging�debug�format�yaml�dump)r Z extra_configZyaml_dir� yaml_file�filesr )r# r �parseQ s
zConfigManager.parsec C s. x |D ]}| j ||| � qW | jj|� d S )N)�
_copy_filer r )r Z config_dictZ config_filer r r �addv s
zConfigManager.addTc C s� |r(| j tjj| jd�tjj| jd�� | j tjj| jd�tjj| jdd�dd� | j tjj| jd�tjj| jd d
�dd� d S ) Nz etc/netplanr z%run/NetworkManager/system-connectionsZNetworkManagerzsystem-connectionsT)�
missing_okzrun/systemd/networkZsystemdr )�
_copy_treer r
r r r r
)r Zbackup_config_dirr r r �backup{ s zConfigManager.backupc C s
y�x*t | j�D ] }tj| j| � | j|= qW dj| j�}dj| j�}tjj|�r�tj tjj
| j d�� | j |tjj
| j d�� tjj|�r�tj tjj
| j d�� | j |tjj
| j d�� W nH t
k
�r } z*tjdj|�� tjd� tjd� W Y d d }~X nX d S ) Nz${}/NetworkManager/system-connectionsz{}/systemd/networkz%run/NetworkManager/system-connectionszrun/systemd/networkz8Something really bad happened while reverting config: {}zZYou should verify the netplan YAML in /etc/netplan and probably run 'netplan apply' again.� ���)�dictr r �unlinkr, r
r
�exists�shutil�rmtreer r r5 � Exceptionr* �error�sys�exit)r Z
extra_fileZ temp_nm_pathZtemp_networkd_path�er r r �revert� s$
zConfigManager.revertc C s t j| j� d S )N)r< r= r )r r r r �cleanup� s zConfigManager.cleanupc C s t j||� d S )N)r< �copy)r �src�dstr r r r2 � s zConfigManager._copy_fileFc C s2 yt j||� W n tk
r, |r&n� Y nX d S )N)r< �copytree�FileNotFoundError)r rF rG r4 r r r r5 � s zConfigManager._copy_treec C s| t � }t|j� �}xd|D ]\}|j|�}||krPtjdj||�� || j|� qtjdj||�� |||< |j|� qW |S )Nz{} exists in {}z{} not found in {}) r �listr( �popr* r+ r, r r3 )r Zorig�newr Zchanged_ifacesZifname�ifacer r r �_merge_interface_config� s
z%ConfigManager._merge_interface_configc C s8 t � }y�t|���}tj|tjd�}d }|d k r8|jd�}|r�d|kr`| j| j|jd��}||O }d|kr�| j| j|jd��}||O }d|kr�| j| j |jd��}||O }d|kr�| j| j
|jd��}||O }d|kr�| j| j |jd��}||O }W d Q R X |S t tj
fk
�r2 tjdj|�� tjd � Y nX d S )
N)�Loaderr r r r r r z!Error while loading {}, aborting.r7 )r �openr- �loadZ CSafeLoader�getrN r r r r r �IOErrorZ YAMLErrorr* r? r, r@ rA )r r/ r �fZ yaml_datar rL r r r r) � s6
z ConfigManager._merge_yaml_configN)T)F)�__name__�
__module__� __qualname__r �propertyr r r r r r r r r1 r3 r6 rC rD r2 r5 rN 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)rU rV rW �__doc__r r r r rY � s rY ) rZ r% r* r r< r@ r r- �objectr r> rY r r r r �<module> s B
Mini Shell