Mini Shell
3
<�Zc�. � @ s� d dl Z d dlmZ d dlmZ d dlmZmZmZ d dl m
Z
d dl m Z d dl
mZmZmZ d dlmZ e� Ze j� Zd gZG d
d � d e�ZdS ) � N)�App)� ExitMainLoop)�ExceptionSignal�RenderScreenSignal�CloseScreenSignal)�RenderUnexpectedError)�UserInputAction)� ScreenStack�
ScreenData�ScreenStackEmptyException)�get_simpleline_logger�ScreenSchedulerc @ s� e Zd Zd)dd�Zdd� Zdd� Zedd � �Zejd
d � �Zed d � �Z d
d� Z
d*dd�Z dd� Z d+dd�Z
d,dd�Zd-dd�Zdd� Zd.dd �Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� ZdS )/r
Nc C s2 d| _ || _|r|| _nt� | _| j� d| _dS )a� Constructor where you can pass your own scheduler stack.
The ScreenStack will be used automatically if scheduler stack will be None.
:param event_loop: Event loop used for the scheduler.
:type event_loop: Class based on `simpleline.event_loop.AbstractEventLoop`.
:param scheduler_stack: Use custom scheduler stack if you need to.
:type scheduler_stack: `simpleline.screen_stack.ScreenStack` based class.
NF)� _quit_screen� _event_loop�
_screen_stackr �_register_handlers�_first_screen_scheduled)�selfZ
event_loopZscheduler_stack� r �&/usr/lib/python3.6/screen_scheduler.py�__init__, s
zScreenScheduler.__init__c C s dj dtj� jd g �S )N�
� �=)�joinr Zget_configuration�width)r r r r �_spacerA s zScreenScheduler._spacerc C s$ | j jt| j� | j jt| j� d S )N)r Zregister_signal_handlerr �_process_screen_callbackr �_close_screen_callback)r r r r r D s z"ScreenScheduler._register_handlersc C s | j S )zReturn quit UIScreen.)r )r r r r � quit_screenH s zScreenScheduler.quit_screenc C s
|| _ dS )a, Set the UIScreen based instance which will be showed before the Application will quit.
You can also use `simpleline.render.adv_widgets.YesNoDialog` or `UIScreen` based class
with the `answer` property. Without the `answer` property the application will always
close.
N)r )r r r r r r M s c C s
| j j� S )z�Is something for rendering in the scheduler stack?
:return: True if the rendering stack is empty
:rtype: bool
)r �empty)r r r r �nothing_to_renderW s z!ScreenScheduler.nothing_to_renderc C s
| j j� S )z1Get string representation of actual screen stack.)r �
dump_stack)r r r r r" ` s zScreenScheduler.dump_stackc C s. t jd|� t||�}| jj|� | j� dS )aP Add screen to the bottom of the stack.
This is mostly useful at the beginning to prepare the first screen hierarchy to display.
:param ui_screen: screen to show
:type ui_screen: UIScreen instance
:param args: optional argument, please see switch_screen for details
:type args: anything
zScheduling screen %sN)�log�debugr
r Z add_first�!_redraw_on_first_scheduled_screen)r � ui_screen�args�screenr r r �schedule_screend s
z ScreenScheduler.schedule_screenc C s | j s| j� d| _ d S )NT)r �redraw)r r r r r% s s z1ScreenScheduler._redraw_on_first_scheduled_screenc C s^ t jd|� y| jj� j}W n tk
r8 td��Y nX t|||�}| jj|� | j� dS )aN Schedules a screen to replace the current one.
:param ui_screen: screen to show
:type ui_screen: instance of UIScreen
:param args: optional argument to pass to ui's refresh and setup methods
(can be used to select what item should be displayed or so)
:type args: anything
zReplacing screen %sz@Switch screen is not possible when there is no screen scheduled!N) r# r$ r �pop�execute_new_loopr r
�appendr* )r r& r' r, r( r r r �replace_screenx s z ScreenScheduler.replace_screenc C s0 t jd|� t||d�}| jj|� | j� dS )a Schedules a screen to show, but keeps the current one in stack to
return to, when the new one is closed.
:param ui_screen: screen to show
:type ui_screen: UIScreen instance
:param args: optional argument
:type args: anything
zPushing screen %s to stackFN)r# r$ r
r r- r* )r r& r' r( r r r � push_screen� s zScreenScheduler.push_screenc C s8 t jd|� t||d�}| jj|� | jjt| �� dS )a� Starts a new screen right away, so the caller can collect data back.
When the new screen is closed, the caller is redisplayed.
This method does not return until the new screen is closed.
:param ui_screen: screen to show
:type ui_screen: UIScreen instance
:param args: optional argument, please see switch_screen for details
:type args: anything
z Pushing modal screen %s to stackTN)r# r$ r
r r- r r, r )r r& r' r( r r r �push_screen_modal� s z!ScreenScheduler.push_screen_modalc C s | j |j� d S )N)� close_screen�source)r �signal�datar r r r � s z&ScreenScheduler._close_screen_callbackc C s� | j j� }tjd||� |jj� |dk rF||jk rFtd||jf ��|jrV| jj � | j j
� rr|j rr| j � | j j
� r�t � �dS )z�Close the currently displayed screen and exit it's main loop if necessary.
Next screen from the stack is then displayed.
zClosing screen %s from %sNzXYou are trying to close screen %s from screen %s! This is most probably not intentional.)
r r+ r# r$ r& �closedr r, r Z
close_loopr r* r )r Z closed_fromr( r r r r1 � s
z ScreenScheduler.close_screenc C s | j jt| �� dS )z4Register rendering to the event loop for processing.N)r �enqueue_signalr )r r r r r* � s zScreenScheduler.redrawc C s | j � d S )N)�_process_screen)r r3 r4 r r r r � s z(ScreenScheduler._process_screen_callbackc C s� | j � }tjd|� |jjsL|jj|j�sL| jj� | j � tj
d|� dS yN|jj |j� || j � krldS | j |� |jj
r�tjd|� |jj|j� W n8 tk
r� � Y n$ tk
r� | jjt| �� dS X dS )z�Draws the current screen and returns True if user input is requested.
If modal screen is requested, starts a new loop and initiates redraw after it ends.
zProcessing screen %sz!Screen %s setup wasn't successfulNz Input is required by %s screenF)�_get_last_screenr# r$ r& Z screen_readyZsetupr' r r+ r* ZwarningZrefresh� _draw_screenZinput_required�get_input_with_error_checkr � Exceptionr r6 r )r Z
top_screenr r r r7 � s*
z ScreenScheduler._process_screenc C s` y"|j jst| j� � |j j� W n8 tk
r8 � Y n$ tk
rZ | jjt | �� Y nX dS )z�Draws the current `active_screen`.
:param active_screen: Screen which should be draw to the console.
:type active_screen: Classed based on `simpleline.render.screen.UIScreen`.
N)
r& Z no_separator�printr Zshow_allr r; r r6 r )r �
active_screenr r r r9 � s z ScreenScheduler._draw_screenc C s | j j� rt� �| j jd�S )NF)r r r r+ )r r r r r8 s
z ScreenScheduler._get_last_screenc C s� | j � }|j� s8|r | j� q�tjd� |jj|j� n�|tj krFd S |tj
krZ| j� nt|tj krn| j � n`|tj
kr�| jr�| j| j� y | jjdkr�t� �n| j� W q� tk
r� t� �Y q�X nt� �d S )Nz,Input was not successful, ask for new input.T)r8 Zwas_successfulr* r# r$ r& r: r' r ZNOOPZREDRAWZCLOSEr1 ZQUITr r0 Zanswerr �AttributeError)r Z input_resultZ
should_redrawr= r r r �process_input_result s,
z$ScreenScheduler.process_input_result)N)N)N)N)N)N)�__name__�
__module__� __qualname__r r r �propertyr �setterr! r" r) r% r. r/ r0 r r1 r* r r7 r9 r8 r? r r r r r
* s(
')Z threadingZ
simpleliner Zsimpleline.event_loopr Z simpleline.event_loop.signalsr r r Zsimpleline.renderr Z&simpleline.render.screen.input_managerr Z simpleline.render.screen_stackr r
r Zsimpleline.loggingr r# ZLockZRAW_INPUT_LOCK�__all__�objectr
r r r r �<module> s