B 't\c@s$dZddlmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZddlmZmZmZdd lmZmZdd lmZmZmZdd lmZdd lmZdd lm Z ddl!Z!ddl"Z"ddgZ#dddZ$Gddde%Z&Gddde'Z(Gddde)Z*Gddde)Z+dddZ,dS)z_ Renders the command line on the console. (Redraws parts of the input line that were changed.) )unicode_literals)FutureFrom ensure_futureget_event_loop) to_filter)to_formatted_text)Input) MouseHandlers)PointScreen WritePosition)Output ColorDepth) BaseStyleDummyStyleTransformationStyleTransformation) is_windows)deque)rangeNRendererprint_formatted_textFc s| j| j } g|j |j} |j|j|j|j|j| fdd  fdd} fdd}|s |r|s| |s|r| kr|t ddd | t }t|j| }tt|j|j| }d}x@t|D]2}|j|}|j|}|j|}t d |r.reset_attributescsjj}}|j|krBd|j|d}|j|S|j|krZ||j|dkrzd|jn>|j|ks|dkr||jn|j|kr|j||S)z; Move cursor to this `new` point. Returns the given Point. z r )xy)newZ current_xZ current_y)_output_cursor_backward_output_cursor_forward_output_cursor_up current_posrwidthwriterr move_cursorLs"      z(_output_screen_diff..move_cursorcsZd}||jkr|jn8|j}|r8||krB||j|jd<dS)z5 Write the output of this character. rN)stylechar)r*Zthe_last_styleZ new_attrs)_output_set_attributesattrs_for_style_string color_depthrr'rr output_chargs     z(_output_screen_diff..output_charr)rr r) columnsrowsr' write_rawset_attributesrZcursor_forward cursor_upcursor_backwardZ hide_cursorZdisable_autowrapr erase_downr minheightmaxrZ data_bufferZzero_width_escapeskeysr&r*r)rr Zerase_end_of_lineZget_cursor_positionlayoutZcurrent_windowenable_autowrap show_cursor)appoutputscreenr%r-Zprevious_screenris_done full_screenr,sizeprevious_widthr7r1r(r.Zcurrent_heightZ row_countcr Znew_rowZ previous_rowZzero_width_escapes_rowZnew_max_line_lenZprevious_max_line_lenZnew_charZold_charZ char_widthr) r"r#r$rr+r,r-r%rrr&r'r_output_screen_diffsx            rEc@seZdZdZdS)HeightIsUnknownErrorz@ Information unavailable. Did not yet receive the CPR response. N)__name__ __module__ __qualname____doc__rrrrrFsrFc@s eZdZdZddZddZdS)_StyleStringToAttrsCachezk A cache structure that maps style strings to :class:`.Attr`. (This is an important speed up.) cCs*t|s tt|tst||_||_dS)N)callableAssertionError isinstancerget_attrs_for_style_strstyle_transformation)selfrOrPrrr__init__s z!_StyleStringToAttrsCache.__init__cCs"||}|j|}|||<|S)N)rOrPZtransform_attrs)rQ style_strattrsrrr __missing__s  z$_StyleStringToAttrsCache.__missing__N)rGrHrIrJrRrUrrrrrKsrKc@seZdZdZdZdZdZdS) CPR_Supportz( Enum: whether or not CPR is supported. SUPPORTED NOT_SUPPORTEDUNKNOWNN)rGrHrIrJrWrXrYrrrrrVsrVc@seZdZdZdZdddZd dd Zed d Zed d Z eddZ ddZ ddZ eddZ d!ddZd"ddZd#ddZddZdS)$rz Typical usage: :: output = Vt100_Output.from_pty(sys.stdout) r = Renderer(style, output) r.render(app, layout=...) FNcCst|tstt|tstt|ts*tt|s>|dks>t||_||_||_||_ t ||_ ||_ d|_ d|_d|_t|_tj|_|jstj|_d|_d|_d|_d|_|jdddS)NFT)_scroll)rNrrMrr rLr)r>inputrAr mouse_supportcpr_not_supported_callback_in_alternate_screen_mouse_support_enabled_bracketed_paste_enabledr_waiting_for_cpr_futuresrVrY cpr_supportZresponds_to_cprrX_attrs_for_style_last_style_hash_last_transformation_hash_last_color_depthreset)rQr)r>r\rAr]r^rrrrR s, zRenderer.__init__TcCstddd|_d|_d|_d|_t|_d|_trB|rB|j |j r\|r\|j d|_ |j rr|j d|_ |jr|j d|_|j dS)Nr)rr F)r _cursor_pos _last_screen _last_size _last_styler mouse_handlers_min_available_heightrr>Zscroll_buffer_to_promptr_Zquit_alternate_screenr`disable_mouse_supportraZdisable_bracketed_pasteflush)rQr[leave_alternate_screenrrrrh.s$      zRenderer.resetcCs|jS)zn The `Screen` class that was generated during the last rendering. This can be `None`. )rj)rQrrrlast_rendered_screenXszRenderer.last_rendered_screencCs|jp|jdkptS)z True when the height from the cursor until the bottom of the terminal is known. (It's often nicer to draw bottom toolbars only if the height is known, in order to avoid flickering when the CPR response arrives.) r)rArnr)rQrrrheight_is_known`szRenderer.height_is_knowncCsN|jr dS|jdkrB|jj}|jr.|jjnd}|t|j|StddS)zU Return the number of rows visible in the terminal above the layout. rzRows above layout is unknown.N) r_rnr>get_sizer0rjr7r8rF)rQ total_rowsZlast_screen_heightrrrrows_above_layoutjs  zRenderer.rows_above_layoutcsjjdkstjr&jj_ntr:j _nnj t j krJdSfdd}j t j krj|n>j t jkrjs|fdd}tj|d}d|_|dS) aS Get current cursor position. We do this to calculate the minimum available height that we can consume for rendering the prompt. This is the available space below te cursor. For vt100: Do CPR request. (answer will arrive later.) For win32: Do API call. (Answer comes immediately.) rNcsjtjdS)N)rbappendrr>Z ask_for_cprr)rQrrdo_cprsz9Renderer.request_absolute_cursor_position..do_cprcs8tjjtjkr4tj_jr4t jdS)N) timesleep CPR_TIMEOUTrcrVrYrXr^rZcall_from_executorr)rQrrtimers   z8Renderer.request_absolute_cursor_position..timer)targetT)rir rMrAr>rtr0rnrZget_rows_below_cursor_positionrcrVrXrWrYwaiting_for_cpr threadingThreaddaemonstart)rQrxr|tr)rQr request_absolute_cursor_positionxs     z)Renderer.request_absolute_cursor_positioncCsXtj|_|jj}||d}||_y|j}Wnt k rHYn X| ddS)z To be called when we know the absolute cursor position. (As an answer of a "Cursor Position Request" response.) rN) rVrWrcr>rtr0rnrbpopleft IndexError set_result)rQrowruZrows_below_cursorfrrrreport_absolute_cursor_rows  z#Renderer.report_absolute_cursor_rowcCs t|jS)zi Waiting for CPR flag. True when we send the request, but didn't got a response. )boolrb)rQrrrr~szRenderer.waiting_for_cprrcsptjrjtjkr$tdStfdd}t|fdd}tj |d}d|_ | S)z* Wait for a CPR response. Nc3s.xD]}t|VqWs*ddS)N)rdoner)Z response_f) cpr_futuresrrrwait_for_responsess z;Renderer.wait_for_cpr_responses..wait_for_responsescs(ts$t_ddS)N)ryrzrrrbrr)rrQtimeoutrrwait_for_timeouts z9Renderer.wait_for_cpr_responses..wait_for_timeout)r}T) listrbrcrVrXrZsucceedrrrrr)rQrrrrr)rrrQrrwait_for_cpr_responsess    zRenderer.wait_for_cpr_responsesc Cs6|j}|jr |js d|_||js6|jd|_|}|rX|jsX|d|_n|sp|jrp| d|_| }t }d|_ t }|jr|j} nJ|r|j|j|jj} n0|jr|jjnd} t|j| |j|j|jj} t| |j} |j|krd|_|j|jks0|j|jks0|j|jkrrAr_Zenter_alternate_screenraZenable_bracketed_paster]r`Zenable_mouse_supportrortr r<r r0Z containerZpreferred_heightr/Z preferredrjr7r8rnr6rkr)Zinvalidation_hashrerPrfr-rgrdrKrOZwrite_to_screenr Zdraw_all_floatsZ exit_styleZappend_style_to_contentrErirlrmrpZvisible_windowsr:rh) rQr=r:r@r>Zneeds_mouse_supportrBr?rmr7Z last_heightrrrrenders~                zRenderer.rendercCsR|j}||jj||jj|||| |j |ddS)aR Hide all output and put the cursor back at the first line. This is for instance used for running a system command (while hiding the CLI) and later resuming the same CLI.) :param leave_alternate_screen: When True, and when inside an alternate screen buffer, quit the alternate screen. )rqN) r>r4rirr3r r5rr;rprh)rQrqr>rrrerase[s zRenderer.erasecCs6||j}||dd||dS)z, Clear screen and go to 0,0 rN)rr>Z erase_screenZ cursor_gotorpr)rQr>rrrclearos  zRenderer.clear)FFN)FT)r)F)T)rGrHrIrJr{rRrhpropertyrrrsrvrrr~rrrrrrrrrs  *  9  # h c Cst|tstt|tst|dks2t|ts2t|dksH|tjksHtt|}|pXt}|pdt }| | t |j |}xP|D]H\}}||} | r|| |n| d|kst||ddqW| |dS)zT Print a list of (style_str, text) tuples in the given style to the output. Nr z )rNrrMrrrZ_ALLrrdefaultrr;rKrOr2r'replacerp) r>Zformatted_textr)rPr-Z fragmentsr,rStextrTrrrrs*   )NNFFNNr)NN)-rJZ __future__rZprompt_toolkit.eventlooprrrrZprompt_toolkit.filtersrZprompt_toolkit.formatted_textrZprompt_toolkit.input.baser Z$prompt_toolkit.layout.mouse_handlersr Zprompt_toolkit.layout.screenr r r Zprompt_toolkit.outputrrZprompt_toolkit.stylesrrrZprompt_toolkit.utilsr collectionsrZ six.movesrryr__all__rE ExceptionrFdictrKobjectrVrrrrrrs:         D