o ˷e5@sdZddlZddlZddlZddlZddlZddlZddlmZm Z ddl m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZddlmZmZmZdd lm Z m!Z!m"Z"dd 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/ddl0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:ddl;mZ>Wne?yddlm@Z@e@Z>YnwdgZAe(ZBeCeddZDde&fddZEedZFGdddZGGddde2ZHeddd ZIGd!d"d"e eIZJdS)#z Progress bar implementation on top of prompt_toolkit. :: with ProgressBar(...) as pb: for item in pb(data): ... N)new_event_loopset_event_loop) GenericIterableIteratorListOptionalSequenceSizedTextIOTypeVarcast) Application)get_app_session)get_event_loop) Conditionis_donerenderer_height_is_known)AnyFormattedTextStyleAndTextTuplesto_formatted_text)Input) KeyBindings) KeyPressEvent)ConditionalContainerFormattedTextControlHSplitLayoutVSplitWindow) UIContent UIControl) AnyDimensionD) ColorDepthOutput) BaseStyle) Formattercreate_default_formatters)dummy_contextvars ProgressBarSIGWINCHreturncCsBt}|ddtddfdd}|ddtddfdd }|S) zu Key bindings handled by the progress bar. (The main thread is not supposed to handle any key bindings.) zc-leventr-NcSs|jjdSN)apprendererclearr.r4a/var/www/ideatree/venv/lib/python3.10/site-packages/prompt_toolkit/shortcuts/progress_bar/base.py_clearOsz#create_key_bindings.._clearzc-ccSstttjdSr/)oskillgetpidsignalSIGINTr3r4r4r5 _interruptSsz'create_key_bindings.._interrupt)raddE)kbr6r<r4r4r5create_key_bindingsHs r@_Tc@seZdZdZ         ddedeeededeedee dee d ee d ee d ee d dfd dZd ddZded dfddZ    d!deeedededeed df ddZd"ddZdS)#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 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. Ntitle formattersbottom_toolbarstyle key_bindingsfile color_depthoutputinputr-c Csv||_|pt|_||_g|_||_||_||_|ptj |_ | p$tj |_ d|_ t |_ t|_d|_t|_dS)NF)rBr)rCrDcountersrErFrHrrIrJ_threadr_loopr _app_loop _has_sigwinch threadingEvent _app_started) selfrBrCrDrErFrGrHrIrJr4r4r5__init__vs zProgressBar.__init__c stttfdddddtfddd}tttfddd d d dd tt@tfd d@d}dtdtffdd fddjD}t dt t |t |fdddt|gj jdjjjd_dfdd }t}tj|j|fd_jS)NcjSr/rBr4rSr4r5z'ProgressBar.__enter__..r'zclass:progressbar,title)heightrEc jduSr/rVr4rWr4r5rX )filtercrUr/rDr4rWr4r5rXrYzclass:bottom-toolbar.text)rEzclass:bottom-toolbar)rErZcr[r/r^r4rWr4r5rXr\ formatterr-cs |jdS)N) progress_bar) get_width)r_rWr4r5width_for_formatters z2ProgressBar.__enter__..width_for_formattercs&g|]}tt|t|dqS))contentwidth)r_ProgressControl functoolspartial).0frSrbr4r5 s z)ProgressBar.__enter__..g?csttjtjdS)N) preferredmax)r#lenrKr4rWr4r5rXs)rZg333333?)min_redraw_intervallayoutrErFrefresh_intervalrHrIrJc sZtjz jjjjdWdSty,}ztt |WYd}~dSd}~ww)N)pre_run) rrNr0runrRset BaseException traceback print_excprint)erWr4r5rss z"ProgressBar.__enter__..run)targetargsr-N)rrrrrrr(r"rCrrrrrErFrHrIrJr0 contextvars copy_contextrPThreadrsrLstart)rS title_toolbarrDprogress_controlsrsctxr4rjr5 __enter__sh     zProgressBar.__enter__acGsD|j|jjr|j|jj|jdur|j|j dSr/) rRwaitr0 is_runningrNcall_soon_threadsafeexitrLjoinclose)rSrr4r4r5__exit__s   zProgressBar.__exit__Fdatalabelremove_when_donetotalzProgressBarCounter[_T]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)ProgressBarCounterrKappend)rSrrrrcounterr4r4r5__call__s   zProgressBar.__call__cCs|jdSr/)r0 invalidaterWr4r4r5r szProgressBar.invalidate) NNNNNNNNN)r-r+NrFNr|)__name__ __module__ __qualname____doc__rrr r(r&rr r$r%rrTrobjectrrrAboolintrrr4r4r4r5r+^sd    !N  c@sXeZdZdZdededdfddZded edefd d Z de fd d Z de fddZ dS)rez, User control for the progress bar. r`r_r-NcCs||_||_t|_dSr/)r`r_r@ _key_bindings)rSr`r_r4r4r5rTs z_ProgressControl.__init__rdrZc s~g|jjD]$}z |j|j||}Wnty"td}Ynwt|qdt dt ffdd }t |t ddS)NERRORir-cs|Sr/r4)ritemsr4r5get_line#sz1_ProgressControl.create_content..get_lineF)r line_count show_cursor) r`rKr_formatrurvrwrrrrr rn)rSrdrZprtextrr4rr5create_contents  z_ProgressControl.create_contentcCsdS)NTr4rWr4r4r5 is_focusable(sz_ProgressControl.is_focusablecC|jSr/)rrWr4r4r5get_key_bindings+sz!_ProgressControl.get_key_bindings)rrrrr+r(rTrr rrrrrr4r4r4r5re s re _CounterItemT) covariantc @seZdZdZ    ddedeeedede d ee d df d d Z d e efd dZ dddZed e fddZejde d dfddZed e fddZejde d dfddZed efddZed ejfddZed eejfddZdS) rzL An individual counter (A progress bar can have multiple counters). NrFr`rrrrr-cCs~tj|_d|_||_||_d|_||_||_d|_ ||dur:z t t t ||_ WdSty9d|_ YdSw||_ dS)NrF)datetimenow start_time stop_timer`ritems_completedrr_donernr r r TypeError)rSr`rrrrr4r4r5rT7s    zProgressBarCounter.__init__ccsJ|jdur!z|jD] }|V|q d|_Wd|_dSd|_wtd)NTz No data defined to iterate over.)ritem_completeddonestoppedNotImplementedError)rSitemr4r4r5__iter__Rs   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`rrWr4r4r5rbsz!ProgressBarCounter.item_completedcCr)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. )rrWr4r4r5rks zProgressBarCounter.donevaluecCs0||_||_|r|jr|jj|dSdSdSr/)rrrr`rKremoverSrr4r4r5rws  cCs |jduS)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)rrWr4r4r5rs zProgressBarCounter.stoppedcCs(|r|js tj|_dSdSd|_dSr/)rrrrr4r4r5rs  cCs$|jdurdS|jdt|jdS)Nrdr')rrrmrWr4r4r5 percentages zProgressBarCounter.percentagecCs&|jdur tj|jS|j|jS)zH Return how much time has been elapsed since the start. N)rrrrrWr4r4r5 time_elapseds  zProgressBarCounter.time_elapsedcCs@|jdus|js dS|js|jrtdS|jd|j|jS)z7 Timedelta representing the time left. Nrr)rrrrr timedeltarrWr4r4r5 time_lefts   zProgressBarCounter.time_leftrr|)rrrrr+rrrrrrrTrrrpropertyrsetterrfloatrrrrrr4r4r4r5r2sF       r)Krrrfr7r:rPrvasynciorrtypingrrrrrr r r r r prompt_toolkit.applicationr"prompt_toolkit.application.currentrprompt_toolkit.eventlooprprompt_toolkit.filtersrrrprompt_toolkit.formatted_textrrrprompt_toolkit.inputrprompt_toolkit.key_bindingr(prompt_toolkit.key_binding.key_processorrprompt_toolkit.layoutrrrrrrprompt_toolkit.layout.controlsr r!prompt_toolkit.layout.dimensionr"r#prompt_toolkit.outputr$r%prompt_toolkit.stylesr&rCr(r)r} ImportErrorr*__all__r>getattr _SIGWINCHr@rAr+rerrr4r4r4r5sN 0           0 "