�
*C�e� � �r � d Z ddlZddlZddlZddlZddlZ G d� de� Z G d� de� Z G d� de� Z
y) z
Terminal / input handling
� Nc �F � e Zd ZdZd� Zd� Zd� Zd� Zd� Zd d�Z d d �Z
d d
�Z y)
�Terminalz@
Do minimal terminal mangling to prompt users for input
c �N � || _ d | _ d | _ | j � y �N)�fd�
orig_flags� orig_term�save)�selfr s �*/usr/share/netplan/netplan_cli/terminal.py�__init__zTerminal.__init__" s � ���� ��� ��� � � � � c �B � t j j � r�t j | j
� }|d t j z |d< |d t j z |d< t j | j
t j |� y y �N� �
�sys�stdin�isatty�termios� tcgetattrr �ICANON�ECHO� tcsetattr�TCSANOW�r �attrss r � enable_echozTerminal.enable_echo( sq � � �9�9� � � ��%�%�d�g�g�.�E� �Q�x�'�.�.�0�E�!�H� �Q�x�'�,�,�.�E�!�H� � � �d�g�g�w���� >� r c �F � t j j � r�t j | j
� }|d t j z |d< |d t j z |d< t j | j
t j |� y y r r r s r � disable_echozTerminal.disable_echo/ su � � �9�9� � � ��%�%�d�g�g�.�E� �Q�x�7�>�>�/�1�E�!�H� �Q�x�7�<�<�-�/�E�!�H� � � �d�g�g�w���� >� r c �� � t j | j t j � }t j | j t j |t j
z � y r ��fcntlr �F_GETFL�F_SETFL�os�
O_NONBLOCK�r �flagss r �enable_nonblocking_ioz Terminal.enable_nonblocking_io6 s; � �� � �D�G�G�U�]�]�3��
� � �D�G�G�U�]�]�E�B�M�M�,A�Br c �� � t j | j t j � }t j | j t j |t j
z � y r r" r( s r �disable_nonblocking_ioz Terminal.disable_nonblocking_io: s= � �� � �D�G�G�U�]�]�3��
� � �D�G�G�U�]�]�E�R�]�]�N�,B�Cr Nc
�n � t d� t � }| j |� | j � | j � |sd}t dj |� � |}|dkD r�t dj |t
t |� � � d�� t j t j gg g d� t j j � }|d k( r| j |� t � � |dz }|dkD r��| j |� t � �# t $ r Y �1w xY w)
a�
Get a "confirmation" input from the user, for at most (timeout)
seconds. Optionally, customize the message to be displayed.
timeout -- timeout to wait for input (default 120)
message -- optional customized message ("Press ENTER to (message)")
raises:
InputAccepted -- the user confirmed the changes
InputRejected -- the user rejected the changes
z%Do you want to keep these settings?
z accept the new configurationz&Press ENTER before the timeout to {}
r z%Changes will revert in {:>{}} seconds�
)�end� �
)�print�dictr
r r* �format�len�str�selectr r �read�reset�
InputAccepted� TypeError�
InputRejected)r �timeout�message�settings� timeout_now�cs r �get_confirmation_inputz Terminal.get_confirmation_input> s � � �7�8��6�� � � �(�� ���� �"�"�$��4�G�
�8�?�?��H�I� � ��Q�� �9�@�@��c�RU�V]�R^�N_�`�fj� k�
�M�M�3�9�9�+�r�2�q� 1�
� �I�I�N�N�$����I��J�J�x�(�'�/�)� �
�1� �K�% �Q��.
�
�
�8� ��o� �� �
� �
�s �>D( �( D4�3D4c �( � t j | j t j � }d}t j j � r t
j | j � }|�|j ||d�� y|| _ || _
y)z�
Save the terminal's current attributes and flags
Optional argument:
- dest: if set, save settings to this dict
N)r) �term) r# r r$ r r r r r �updater r )r �destr r s r r
z
Terminal.savep sq � � �[�[����%�-�-�8�
�� � �9�9� � � � �)�)�$�'�'�2�I� � � �K�K�*�!*�,�
-� )�D�O� &�D�Nr c �~ � d}d}|�#|j d� }|j d� }n| j }| j }t j j � r/t
j | j t j |� t j | j t j |� y)z�
Reset the terminal to its original attributes and flags
Optional argument:
- orig: if set, reset to settings from this dict
NrD r) ) �getr r r r r r r r � TCSAFLUSHr# r% )r �origr r s r r9 zTerminal.reset� s� � � � ��
� � � ����(�I� ���'�*�J� ���I� � � �J� �9�9� � � � � � �d�g�g�w�'8�'8�)� D�
� � �D�G�G�U�]�]�J�7r )�x Nr ) �__name__�
__module__� __qualname__�__doc__r
r r r* r, rB r
r9 � r r r r s2 � ��� ?�?�C�D�0 �d'�$8r r c � � e Zd ZdZy)r: z Denotes has accepted inputN�rL rM rN rO rP r r r: r: � s � �%�r r: c � � e Zd ZdZy)r< z) Denotes that the user has rejected inputNrR rP r r r< r< � s � �3�r r< ) rO r# r& r r7 r �objectr � Exceptionr: r< rP r r �<module>rV sB ��$�
� � �
�
�v8�v� v8�r �I� �
�I� r |