U .eA8@sdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z m Z m Z mZmZmZmZmZmZddlmZddlmZddlmZmZmZddlmZmZmZdd l m!Z!dd l"m#Z#dd l$m%Z%dd l&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,dd l-m.Z.m/Z/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;me?eddZ@dddddZAedZBGdddZCGddde/ZDedd d!ZEGd"d#d#e eEZFdS)$z Progress bar implementation on top of prompt_toolkit. :: with ProgressBar(...) as pb: for item in pb(data): ... ) annotationsN) CallableGenericIterableIteratorSequenceSizedTextIOTypeVarcast) Application)get_app_session) Conditionis_donerenderer_height_is_known)AnyFormattedTextStyleAndTextTuplesto_formatted_text)Input) KeyBindings) KeyPressEvent)ConditionalContainerFormattedTextControlHSplitLayoutVSplitWindow) UIContent UIControl) AnyDimensionD) ColorDepthOutput) BaseStyle)in_main_thread) Formattercreate_default_formatters ProgressBarSIGWINCHCallable[[], None] | Noner)cancel_callbackreturncsJt}|dddddd}dk rF|ddddfd d }|S) zu Key bindings handled by the progress bar. (The main thread is not supposed to handle any key bindings.) zc-lENone)eventr,cSs|jjdSN)appZrendererclearr/r4N/tmp/pip-unpacked-wheel-_x9ru1vw/prompt_toolkit/shortcuts/progress_bar/base.py_clearHsz#create_key_bindings.._clearNzc-ccsdk s tdS)zMKill the 'body' of the progress bar, but only if we run from the main thread.N)AssertionErrorr3r+r4r5 _interruptNs z'create_key_bindings.._interrupt)radd)r+kbr6r9r4r8r5create_key_bindingsAsr<_Tc @sxeZdZdZd"dddddddd d d d d ddZddddZdd dddZd#ddddddddZd dd d!ZdS)$r(a. Progress bar context manager. Usage :: with ProgressBar(...) as pb: for item in pb(data): ... :param title: Text to be displayed above the progress bars. This can be a callable or formatted text as well. :param formatters: List of :class:`.Formatter` instances. :param bottom_toolbar: Text to be displayed in the bottom toolbar. This can be a callable or formatted text. :param style: :class:`prompt_toolkit.styles.BaseStyle` instance. :param key_bindings: :class:`.KeyBindings` instance. :param cancel_callback: Callback function that's called when control-c is pressed by the user. This can be used for instance to start "proper" cancellation if the wrapped code supports it. :param file: The file object used for rendering, by default `sys.stderr` is used. :param color_depth: `prompt_toolkit` `ColorDepth` instance. :param output: :class:`~prompt_toolkit.output.Output` instance. :param input: :class:`~prompt_toolkit.input.Input` instance. NrzSequence[Formatter] | NonezBaseStyle | NonezKeyBindings | Noner*z TextIO | NonezColorDepth | Nonez Output | Nonez Input | Noner.) title formattersbottom_toolbarstyle key_bindingsr+file color_depthoutputinputr,c Cs||_|pt|_||_g|_||_||_||_|jdkrTtrTdddd} | |_||_ | pdt j |_ | prt j |_ d|_ d|_t|_dS)Nr.r,cSstttjdSr0)oskillgetpidsignalSIGINTr4r4r4r5!keyboard_interrupt_to_main_threadsz?ProgressBar.__init__..keyboard_interrupt_to_main_threadF)r>r'r?r@countersrArBr+r$rDr rErF_threadZ _has_sigwinch threadingEvent _app_started) selfr>r?r@rArBr+rCrDrErFrMr4r4r5__init__us  zProgressBar.__init__rGc stttfdddddtfddd}tttfddd d d dd tt@tfd d@d}dddfdd fddjD}tdtt |t |fdddt|gj j dj jjd_ddfdd }t}tj|j|fd_jS)NcsjSr0r>r4rSr4r5z'ProgressBar.__enter__..r%zclass:progressbar,title)heightrAcs jdk Sr0rUr4rVr4r5rWrX)filtercsjSr0r@r4rVr4r5rWrXzclass:bottom-toolbar.text)rAzclass:bottom-toolbar)rArYcs jdk Sr0r[r4rVr4r5rWrXr&r) formatterr,cs |jdS)N) progress_bar)Z get_width)r\rVr4r5width_for_formattersz2ProgressBar.__enter__..width_for_formattercs*g|]"}tt|jt|dqS))contentwidth)r_ProgressControlr+ functoolspartial).0frSr^r4r5 s   z)ProgressBar.__enter__..g?csttjtjdS)N) preferredmax)r lenrNr4rVr4r5rWs)rYg333333?)Zmin_redraw_intervalZlayoutrArBZrefresh_intervalrDrErFr.rGc sNzjjjjdWn2tk rH}ztt|W5d}~XYnXdS)N)Zpre_run)r1runrRset BaseException traceback print_excprint)erVr4r5rks z"ProgressBar.__enter__..run)targetargs)rrrrrrr?r rrrrArBrDrErFr1 contextvarsZ copy_contextrPThreadrkrOstart)rSZ title_toolbarr@Zprogress_controlsrkctxr4rfr5 __enter__sj      zProgressBar.__enter__object)ar,cGsH|j|jjr0|jjdk r0|jj|jj|jdk rD|jdSr0) rRwaitr1Z is_runningZloopZcall_soon_threadsafeexitrOjoin)rSrzr4r4r5__exit__s   zProgressBar.__exit__FzIterable[_T] | Nonebool int | NonezProgressBarCounter[_T])datalabelremove_when_donetotalr,cCs"t|||||d}|j||S)aL Start a new counter. :param label: Title text or description for this progress. (This can be formatted text as well). :param remove_when_done: When `True`, hide this progress bar. :param total: Specify the maximum value if it can't be calculated by calling ``len``. )rrr)ProgressBarCounterrNappend)rSrrrrcounterr4r4r5__call__s zProgressBar.__call__cCs|jdSr0)r1 invalidaterVr4r4r5rszProgressBar.invalidate) NNNNNNNNNN)NrFN) __name__ __module__ __qualname____doc__rTrxr~rrr4r4r4r5r(Zs($*Mc@sReZdZdZdddddddZd d d d d d ZddddZddddZdS)raz, User control for the progress bar. r(r&r*r.)r]r\r+r,cCs||_||_t||_dSr0)r]r\r< _key_bindings)rSr]r\r+r4r4r5rTsz_ProgressControl.__init__intr)r`rYr,c s~g|jjD]J}z|j|j||}Wn tk rFtd}YnXt|q dddfdd }t |t ddS) NERRORrr)ir,cs|Sr0r4)ritemsr4r5get_line.sz1_ProgressControl.create_content..get_lineF)r line_countZ show_cursor) r]rNr\formatrmrnrorrrrj)rSr`rYprtextrr4rr5create_content"s  z_ProgressControl.create_contentrrGcCsdS)NTr4rVr4r4r5 is_focusable3sz_ProgressControl.is_focusablercCs|jSr0)rrVr4r4r5get_key_bindings6sz!_ProgressControl.get_key_bindingsN)rrrrrTrrrr4r4r4r5ras  ra _CounterItemT) covariantc@seZdZdZd$ddddd d d d d ZddddZd dddZeddddZej dd dddZeddddZ e j dd dddZ eddddZ edddd Z ed!dd"d#Z dS)%rzL An individual counter (A progress bar can have multiple counters). NrFr(zIterable[_CounterItem] | Nonerrrr.)r]rrrrr,cCs~tj|_d|_||_||_d|_||_||_d|_ ||dkrtzt t t ||_ Wqztk rpd|_ YqzXn||_ dS)NrF)datetimenow start_time stop_timer]ritems_completedrr_donerjr rr TypeError)rSr]rrrrr4r4r5rTBs zProgressBarCounter.__init__zIterator[_CounterItem]rGccsF|jdk r:z$|jD]}|V|qd|_W5d|_XntddS)NTz No data defined to iterate over.)rstoppeditem_completeddoneNotImplementedError)rSitemr4r4r5__iter__\s     zProgressBarCounter.__iter__cCs|jd7_|jdS)z Start handling the next item. (Can be called manually in case we don't have a collection to loop through.) r%N)rr]rrVr4r4r5rlsz!ProgressBarCounter.item_completedcCs|jS)a6Whether a counter has been completed. Done counter have been stopped (see stopped) and removed depending on remove_when_done value. Contrast this with stopped. A stopped counter may be terminated before 100% completion. A done counter has reached its 100% completion. )rrVr4r4r5rus zProgressBarCounter.done)valuer,cCs(||_||_|r$|jr$|jj|dSr0)rrrr]rNremoverSrr4r4r5rs cCs |jdk S)aWhether a counter has been stopped. Stopped counters no longer have increasing time_elapsed. This distinction is also used to prevent the Bar formatter with unknown totals from continuing to run. A stopped counter (but not done) can be used to signal that a given counter has encountered an error but allows other counters to continue (e.g. download X of Y failed). Given how only done counters are removed (see remove_when_done) this can help aggregate failures from a large number of successes. Contrast this with done. A done counter has reached its 100% completion. A stopped counter may be terminated before 100% completion. N)rrVr4r4r5rszProgressBarCounter.stoppedcCs"|r|jstj|_nd|_dSr0)rrrrr4r4r5rsfloatcCs(|jdkrdS|jdt|jdSdS)Nrdr%)rrrirVr4r4r5 percentages zProgressBarCounter.percentagezdatetime.timedeltacCs*|jdkrtj|jS|j|jSdS)zH Return how much time has been elapsed since the start. N)rrrrrVr4r4r5 time_elapseds zProgressBarCounter.time_elapsedzdatetime.timedelta | NonecCsD|jdks|jsdS|js |jr*tdS|jd|j|jSdS)z7 Timedelta representing the time left. Nrr)rrrrr timedeltarrVr4r4r5 time_lefts   zProgressBarCounter.time_left)NrFN)rrrrrTrrpropertyrsetterrrrrr4r4r4r5r=s,    r)Gr __future__rrtrrbrHrKrPrntypingrrrrrrr r r Zprompt_toolkit.applicationr Z"prompt_toolkit.application.currentr Zprompt_toolkit.filtersrrrZprompt_toolkit.formatted_textrrrZprompt_toolkit.inputrZprompt_toolkit.key_bindingrZ(prompt_toolkit.key_binding.key_processorrZprompt_toolkit.layoutrrrrrrZprompt_toolkit.layout.controlsrrZprompt_toolkit.layout.dimensionrr Zprompt_toolkit.outputr!r"Zprompt_toolkit.stylesr#Zprompt_toolkit.utilsr$r?r&r'__all__r-getattrZ _SIGWINCHr<r=r(rarrr4r4r4r5sB ,         :'