�
���g� � �r � S r SSKrSSKrSSKrSSKrSSKr " S S\5 r " S S\5 r " S S\5 r
g) z
Terminal / input handling
� Nc �T � \ rS rSrSrS rS rS rS rS r S
S jr
SS
jr SS jr S r
g)�Terminal� z8
Do minimal terminal mangling to prompt users for input
c �L � Xl S U l S U l U R 5 g �N)�fd�
orig_flags� orig_term�save)�selfr s �*/usr/share/netplan/netplan_cli/terminal.py�__init__�Terminal.__init__" s � ��� ��� ��� � � � � c �P � [ R R 5 ( a� [ R " U R
5 nUS [ R - US' US [ R - US' [ R " U R
[ R U5 g g �N� �
�sys�stdin�isatty�termios� tcgetattrr �ICANON�ECHO� tcsetattr�TCSANOW�r �attrss r
� enable_echo�Terminal.enable_echo( st � � �9�9� � � � ��%�%�d�g�g�.�E� �Q�x�'�.�.�0�E�!�H� �Q�x�'�,�,�.�E�!�H� � � �d�g�g�w���� >� r c �T � [ R R 5 ( a� [ R " U R
5 nUS [ R ) - US' US [ R ) - US' [ R " U R
[ R U5 g g r r r s r
� disable_echo�Terminal.disable_echo/ sx � � �9�9� � � � ��%�%�d�g�g�.�E� �Q�x�7�>�>�/�1�E�!�H� �Q�x�7�<�<�-�/�E�!�H� � � �d�g�g�w���� >� r c �� � [ R " U R [ R 5 n[ R " U R [ R U[ R
- 5 g r ��fcntlr �F_GETFL�F_SETFL�os�
O_NONBLOCK�r �flagss r
�enable_nonblocking_io� Terminal.enable_nonblocking_io6 s; � �� � �D�G�G�U�]�]�3��
� � �D�G�G�U�]�]�E�B�M�M�,A�Br c �� � [ R " U R [ R 5 n[ R " U R [ R U[ R
) - 5 g r r& r, s r
�disable_nonblocking_io� Terminal.disable_nonblocking_io: s= � �� � �D�G�G�U�]�]�3��
� � �D�G�G�U�]�]�E�R�]�]�N�,B�Cr Nc
�~ � [ S5 [ 5 nU R U5 U R 5 U R 5 U( d Sn[ SR U5 5 UnUS:� a� [ SR U[
[ U5 5 5 SS9 [ R " [ R // / S5 [ R R 5 nUS :X a U R U5 [ 5 e US- nUS:� a M� U R U5 [ 5 e! [ a N4f = f)
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_input� 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/ �/
D<�;D<c �4 � [ R " U R [ R 5 nSn[ R R 5 ( a [ R " U R 5 nUb UR UUS.5 gX l X0l
g)zu
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 �
Terminal.savep sp � � �[�[����%�-�-�8�
�� � �9�9� � � � � �)�)�$�'�'�2�I� � � �K�K�*�!*�,�
-� )�O� &�Nr c � � SnSnUb# UR S5 nUR S5 nOU R nU R n[ R R 5 ( a0 [ R " U R [ R U5 [ R " U R [ R U5 g)z�
Reset the terminal to its original attributes and flags
Optional argument:
- orig: if set, reset to settings from this dict
NrK r- ) �getr
r r r r r r r � TCSAFLUSHr' r) )r �origr
r s r
r? �Terminal.reset� s� � � � ��
� � � ����(�I� ���'�*�J� ���I� � � �J� �9�9� � � � � � � �d�g�g�w�'8�'8�)� D�
� � �D�G�G�U�]�]�J�7r )r r r
)�x Nr )�__name__�
__module__� __qualname__�__firstlineno__�__doc__r r r# r. r1 rH r r? �__static_attributes__� r r
r r s2 � ��� ?�?�C�D�0 �d'�$8r r c � � \ rS rSrSrSrg)r@ � zDenotes has accepted inputr[ N�rU rV rW rX rY rZ r[ r r
r@ r@ � s � �%�r r@ c � � \ rS rSrSrSrg)rB � z(Denotes that the user has rejected inputr[ Nr^ r[ r r
rB rB � s � �3�r rB ) rY r' r* r r= r �objectr � Exceptionr@ rB r[ r r
�<module>rc sB ��$�
� � �
�
�v8�v� v8�r �I� �
�I� r