�
*C�e� � �p � d Z ddlZddlZddlZddlZddlZddlZddlmZ G d� de � Z
G d� de � Z y)z netplan configuration manager� N)�Optionalc �t � e Zd Zdi fd�Zd� Zed� � Zed� � Zdd�Zd� Z dd �Z
d
� Z d � Z d � Z
d
� Zdd�Zy)�
ConfigManager�/c �> � || _ t j d�� | _ t j
j
| j d� | _ t j
j
| j d� | _ || _ t � | _ d | _ y )N�netplan_)�prefix�etc�run)
r �tempfile�mkdtemp�tempdir�os�path�join�temp_etc�temp_run� extra_files�set�new_interfaces�np_state)�selfr r s �//usr/share/netplan/netplan_cli/configmanager.py�__init__zConfigManager.__init__ sh � � �� � �'�'�z�:�� ���� � �T�\�\�5�9��
���� � �T�\�\�5�9��
�&��� !�e���15��
� c �T � | j �J d� �t | j |� S �Nz.Must call parse() before accessing the config.)r �getattr)r �attrs r � __getattr__zConfigManager.__getattr__( s) � ��}�}�(�Z�*Z�Z�(��t�}�}�d�+�+r c � � | j �J d� �i }|j | j j � |j | j j � |j | j j � |S r )r �update� ethernets�modems�wifis�r �
interfacess r �physical_interfacesz!ConfigManager.physical_interfaces, sk � ��}�}�(�Z�*Z�Z�(��
����$�-�-�1�1�2����$�-�-�.�.�/����$�-�-�-�-�.��r c �6 � | j �J d� �i }|j | j j � |j | j j � |j | j j � |j | j j
� |j | j j � |j | j j � |j | j j � |S r ) r r" �bridges�bonds�
dummy_devices�tunnels�virtual_ethernets�vlans�vrfsr& s r �virtual_interfacesz ConfigManager.virtual_interfaces5 s� � ��}�}�(�Z�*Z�Z�(��
����$�-�-�/�/�0����$�-�-�-�-�.����$�-�-�5�5�6����$�-�-�/�/�0����$�-�-�9�9�:����$�-�-�-�-�.����$�-�-�,�,�-��r Nc �z � t j � } |j | j �� |r|D ] }|j |� � t j
� | _ | j j |� t j � 5 }| j j |�� t j dj! |j# � � � ddd� | j S # t j $ r}t t |� � �d}~ww xY w# 1 sw Y | j S xY w)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 libnetplan State wrapper
)�rootdirN)� output_filezMerged config:
{})�netplan�Parser�load_yaml_hierarchyr � load_yaml�Stater �import_parser_results�NetplanException�ConfigurationError�strr �
TemporaryFile�
_dump_yaml�logging�debug�format�read)r � extra_config�parser�f�e�tmps r �parsezConfigManager.parseC s � � � � �!��
-� � &� &�t�{�{� &� ;��%� (�A��$�$�Q�'�(� $�M�M�O�D�M� �M�M� /� /�� 7�
�
#�
#�
%� C�� �M�M� $� $�� $� 5� �M�M�.�5�5�c�h�h�j�A� B� C� �}�}� �� �'�'� -�$�S��V�,� ,�� -�� C� �}�}� �s% �A*C7 �AD&�7D#�
D � D#�&D:c � � |D ] }| j ||| � � | j j |� d | _ y �N)�
_copy_filer r" r )r � config_dict� config_files r �addzConfigManager.add` sC � �&� C�K� �O�O�K��[�)A� B� C� ��� � � �,� ��
r c �` � |rb| j t j j | j d� t j j | j
d� � | j t j j | j d� t j j | j dd� d�� | j t j j | j d� t j j | j d d
� d�� y ) Nz etc/netplanr5 �%run/NetworkManager/system-connections�NetworkManagerzsystem-connectionsT)�
missing_ok�run/systemd/network�systemd�network)�
_copy_treer r r r r r )r �backup_config_dirs r �backupzConfigManager.backuph s� � � � �O�O�B�G�G�L�L����m� D� �G�G�L�L�� � � � B�
D� ������ � �T�[�[�2Y�Z���� � �T�]�]�4D�FZ�[�#'� � )�
������ � �T�[�[�2G�H���� � �T�]�]�I�y�I�#'� � )r c �& � t | j � D ]1 }t j | j | � | j |= �3 dj | j
� }dj | j
� }t j j |� rwt j t j j | j d� � | j |t j j | j d� � t j j |� rxt j t j j | j d� � | j |t j j | j d� � y y # t $ rX}t j dj |� � t j d� t! j" d� Y d }~y d }~ww xY w)Nz${}/NetworkManager/system-connectionsz{}/systemd/networkrQ rT z8Something really bad happened while reverting config: {}zZYou should verify the netplan YAML in /etc/netplan and probably run 'netplan apply' again.���)�dictr r �unlinkrB r r �exists�shutil�rmtreer r rW � Exceptionr@ �error�sys�exit)r �
extra_file� temp_nm_path�temp_networkd_pathrG s r �revertzConfigManager.reverts sl � � � "�4�#3�#3� 4�
1�
�� � �$�*�*�:�6�7��$�$�Z�0�
1� B�H�H�� � �W�L�!5�!<�!<�T�]�]�!K� ��w�w�~�~�l�+��
�
�b�g�g�l�l�4�;�;�8_� `�a���� � "��� � �T�[�[�:a� b�d��w�w�~�~� 0�1��
�
�b�g�g�l�l�4�;�;�8M� N�O���� 2� "��� � �T�[�[�:O� P�R� 2�� � �
�M�M�T�[�[�\]�^� _� �M�M�v� w� �H�H�R�L�L�� �s �F+F/ �/ H�8AH � Hc �B � t j | j � y rK )r_ r` r �r s r �cleanupzConfigManager.cleanup� s � ��
�
�d�l�l�#r c �D � | j � y # t $ r Y y w xY wrK )rk �FileNotFoundErrorrj s r �__del__zConfigManager.__del__� s" � � � �L�L�N�� � � � �s � � � c �0 � t j ||� y rK )r_ �copy)r �src�dsts r rL zConfigManager._copy_file� s � �� � �C�� r c �V � t j ||� y # t $ r |rY y � w xY wrK )r_ �copytreerm )r rq rr rS s r rW zConfigManager._copy_tree� s- � � � �O�O�C�� %�� � ���� �s � � (�(rK )T)F)�__name__�
__module__� __qualname__r r �propertyr( r1 rI rO rY rh rk rn rL rW � r r r r sa � � !�r� 6�,� �� �� � � � � �: � )��4$�� �r r c � � e Zd ZdZy)r< zL
Configuration could not be parsed or has otherwise failed to apply
N)ru rv rw �__doc__ry r r r< r< � s � �� r r< )
r{ r@ r5 r r_ rc r �typingr �objectr ra r< ry r r �<module>r~ s: ��$ $� � � �
�
� � �C�F� C�L �� r |