Mini Shell
3
<�Zcm � @ s� d dl Z d dlmZ d dlmZ d dlmZ d dlmZm Z d dl
m Z d dl m
Z
mZmZ dd d
d d d
gZG dd� de�ZG dd � d e�ZG dd
� d
e�ZG dd � d e�ZG dd � d e�ZG dd
� d
e�ZdS )� N)�widgets)�WindowContainer)�Prompt)�UIScreen�
InputState)�PasswordInputHandler)�_�N_�C_� ErrorDialog�GetInputScreen�GetPasswordInputScreen�
HelpScreen�PasswordDialog� YesNoDialogc s@ e Zd ZdZ� fdd�Zd � fdd� Zd dd�Zd d
� Z� ZS )
r z+Dialog screen for reporting errors to user.c s t � j� td�| _|| _dS )z\
:param message: the message to show to the user
:type message: str
�ErrorN)�super�__init__r �title�_message)�self�message)� __class__� �!/usr/lib/python3.6/adv_widgets.pyr # s
zErrorDialog.__init__Nc s. t � j|� tj| j�}| jjtj|�� d S )N)r �refreshr �
TextWidgetr �window�add_with_separator� CenterWidget)r �args�text)r r r r , s zErrorDialog.refreshc C s t td�t j �S )NzPress %s to exit)r r �ENTER)r r r r r �prompt1 s zErrorDialog.promptc C s t jd� dS )zUThis dialog is closed by any input.
And causes the program to quit.
� N)�sys�exit)r r �keyr r r �input4 s zErrorDialog.input)N)N) �__name__�
__module__� __qualname__�__doc__r r r# r( �
__classcell__r r )r r r s
c sN e Zd ZdZd
� fdd� Zd� fdd� Zddd�Zed d
� �Zd d � Z � Z
S )r z!Dialog screen for password input.Nc s, t � j� td�| _|p td�| _d| _dS )zX
:param message: password prompt question
:type message: string
ZPasswordzEnter your passphraseN)r r r r r r � _password)r r )r r r r ? s
zPasswordDialog.__init__c s. t � j|� tj| j�}| jjtj|�� d S )N)r r r r r r r r )r r r! )r r r r I s zPasswordDialog.refreshc C sR t | d�}| jr |j| j� |jtd�� |j� |j� s>d S |j| _| j � d S )N)�sourcez Passphrase: )
r Z
password_funcZ
set_pass_funcZ get_inputr Z
wait_on_inputZinput_successful�valuer. �close)r r Zhandlerr r r r# N s
zPasswordDialog.promptc C s | j S )z?The response can be None (no response) or the password entered.)r. )r r r r �answerb s zPasswordDialog.answerc C s |r|| _ tjS tjS d S )N)r. r �PROCESSED_AND_CLOSE� DISCARDED)r r r' r r r r( g s zPasswordDialog.input)N)N)N) r) r* r+ r, r r r# �propertyr2 r( r- r r )r r r < s
c sL e Zd ZdZ� fdd�Zd
� fdd� Zddd�Zd d
� Zed d � �Z � Z
S )r z%Dialog screen for Yes - No questions.c s$ t � j� td�| _|| _d| _dS )z`
:param message: the message to show to the user
:type message: unicode
ZQuestionN)r r r r r � _response)r r )r r r r r s
zYesNoDialog.__init__Nc s. t � j|� tj| j�}| jjtj|�� d S )N)r r r r r r r r )r r r! )r r r r | s zYesNoDialog.refreshc C s" t td�tdd�tdd�d� �S )Nz$Please respond '%(yes)s' or '%(no)s'zTUI|Spoke Navigation�yes�no)r7 r8 )r r r
)r r r r r r# � s zYesNoDialog.promptc C s> |t dd�krd| _tjS |t dd�kr4d| _tjS tjS d S )NzTUI|Spoke Navigationr7 Tr8 F)r
r6 r r3 r4 )r r r' r r r r( � s zYesNoDialog.inputc C s | j S )zAThe response can be True (yes), False (no) or None (no response).)r6 )r r r r r2 � s zYesNoDialog.answer)N)N) r) r* r+ r, r r r# r( r5 r2 r- r r )r r r o s
c s@ e Zd ZdZ� fdd�Zd � fdd� Zdd� Zd d d
�Z� ZS )
r z!Screen to display a help message.c s t � j� td�| _|| _dS )zO
:param help_path: help file name
:type help_path: str
ZHelpN)r r r r � help_path)r r9 )r r r r � s
zHelpScreen.__init__Nc
sP t � j|� td�}| jr:t| jd��}|j� }W dQ R X | jjtj |�� dS )z Show the help. zThe help is not available.�rN)
r r r r9 �open�readr r r r )r r Z help_message�f)r r r r � s zHelpScreen.refreshc C s t jS )z Handle user input. )r r3 )r r r' r r r r( � s zHelpScreen.inputc C s t td�t j �S )NzPress %s to return)r r r" )r r r r r r# � s zHelpScreen.prompt)N)N) r) r* r+ r, r r r( r# r- r r )r r r � s
c sf e Zd ZdZ� fdd�Zedd� �Zddd�Zd d
� Zd� fd d � Z dd
d�Z
dd� Z dd� Z � Z
S )r z Screen for getting user input.c s t � j� || _d| _g | _dS )z^
:param message: Prompt printed before user input.
:type message: str
N)r r r �_value� _conditions)r r )r r r r � s
zGetInputScreen.__init__c C s | j S )z User input.)r> )r r r r r0 � s zGetInputScreen.valueNc C s | j j||f� dS )a6 Add acceptance condition to the conditions list.
:param acceptance_function: Functions that accepts or rejects a user input.
:type acceptance_function: `function(input, args) -> bool` - function which takes
user input (string) and arguments (`args`) and return True when input is accepted or
False if rejected so we will ask for a new input.
:param args: Second argument for `acceptance_function` the first one will be user input.
:type args: Anything.
N)r? �append)r Zacceptance_functionr r r r �add_acceptance_condition� s z'GetInputScreen.add_acceptance_conditionc C s | j j� dS )z(Clear list of the acceptance conditions.N)r? �clear)r r r r �clear_acceptance_conditions� s z*GetInputScreen.clear_acceptance_conditionsc s t � j|� t� | _d S )N)r r r Z_window)r r )r r r r � s zGetInputScreen.refreshc C s t | jd�S )N)r )r r )r r r r r r# � s zGetInputScreen.promptc C s | j |�stjS || _tjS )N)� _test_inputr r4 r> r3 )r r r' r r r r( � s
zGetInputScreen.inputc C s& x | j D ]\}}|||�sdS qW dS )NFT)r? )r r' r= r r r r rD � s
zGetInputScreen._test_input)N)N)N)r) r* r+ r, r r5 r0 rA rC r r# r( rD r- r r )r r r � s
c s e Zd ZdZ� fdd�Z� ZS )r
z'Screen for getting user password input.c s t � j|� d| _d S )NT)r r Zhide_user_input)r r )r r r r � s z GetPasswordInputScreen.__init__)r) r* r+ r, r r- r r )r r r
� s )r% Zsimpleline.renderr Z simpleline.render.containersr Zsimpleline.render.promptr Zsimpleline.render.screenr r Z simpleline.input.input_handlerr Zsimpleline.utils.i18nr r r
�__all__r r r r r r
r r r r �<module> s 3. :