wi412ddlZddlZddlmZmZddlmZmZddlmZmZm Z m Z ddl Z ddl m Z m ZddlmZmZddlmZddlmZdd lmZmZdd lmZdd lmZmZdd lmZ dd l m!Z!ddl"m#Z#ddl$m%Z%n#e&$r dZ!dZ#dZ%YnwxYwdZ'GddZ(dS)N)datetime timedelta)EventThread)AnyDictOptionalType)TqdmExperimentalWarningtqdm) WorkerComms POISON_PILL)remove_highlighting)WorkerInsights)WorkerMapParamsWorkerPoolParams)DisableKeyboardInterruptSignal)get_tqdm TqdmManager)format_seconds)DASHBOARD_STARTED_EVENT)get_function_details)get_manager_client_dictscdSN)_s j/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/mpire/progress_bar.pyrrs ctr)NotImplementedErrorrrrrrs!!rz%Y-%m-%d, %H:%M:%SceZdZdedededeeefde ede de dd fd Z dd Z d edd fd ZddZdedd fdZ ddedede edd fdZ ddedede edeeeffdZdedd fdZdedd fdZd S)ProgressBarHandler pool_params map_paramsshow_progress_barprogress_bar_optionsprogress_bar_style worker_commsworker_insightsreturnNc||_||_||_||_||_|rCt 5HinN^N^5_5_5_2"&*# rc|jrrt5t|j|_|j|jdddn #1swxYwY|S)z Enables the use of the ``with`` statement. Starts a new progress handler thread if a progress bar should be shown :return: self )targetN)r&rr_progress_bar_handlerr2startr3wait)r?s r __enter__zProgressBarHandler.__enter__Hs  ! +011 + +$D,FGGG  !!####((*** + + + + + + + + + + + + + + +  sA A//A36A3exc_typec|jr|jrj||j|js|j|jdSdSdS)zs Enables the use of the ``with`` statement. Terminates the progress handler thread if there is one N)r&r2is_aliver)signal_kill_signal_receivedexception_thrownsignal_progress_bar_shutdownjoin)r?rGrs r__exit__zProgressBarHandler.__exit__Ys  ! dk&:&:&<&< #!==???$5577 A!>>@@@ K           rct|j}t}|\}}|||||jdtj 5tj dtd|_ |d i|j}dddn #1swxYwYtj|j|_|j|| |j}|t*ur|js|jr|d|jr]|j5|j|jdddn #1swxYwY||d|jn1|jr||dd||dS|j rH|!|j"|||j#|d kr ||j$kr|%||j$z |j$|j"krG|j&|j'|||j$|j(kr||I) zj Keeps track of the progress made by the workers and updates the progress bar accordingly positionignoreFNTz$Exception occurred, terminating ... )failed traceback_strzKill signal receivedrr))rr(rget_connection_detailsset_lockset_main_progress_barregister_progress_bar_positionr'warningscatch_warnings simplefilterr monitor_intervalr fromtimestampr>r3set_register_progress_barr) get_tasks_completed_progress_barrrKkill_signal_receivedset_descriptionr;r7rE_send_dashboard_update final_refresh!get_highest_progress_bar_positionr6r/ update_totalr5clearnupdatesignal_progress_bar_complete#wait_until_progress_bar_is_complete last_print_n)r?r tqdm_manager tqdm_locktqdm_position_register progress_bartasks_completeds rrCz(ProgressBarHandler._progress_bar_handleris /00#}} ,8,O,O,Q,Q) ) i    "" " A A$B[\fBg h h    $ & & = =  !(,C D D D$)D !4<<$";<>@@@!EEGGG++L999~!:::++L999]. :s$/CCC2!GG#&G#roc|jttjrt\|_|_}|t|jdz|_|j |j|j fg| || dSdSdSdS)zv Register this progress bar to the dashboard :param progress_bar: tqdm progress bar instance N) r4rr/rr<r=acquirelenkeysrhr1rbrelease)r?rodashboard_tqdm_locks rr^z)ProgressBarHandler._register_progress_bars   ',C,OTkTrTtTt,OTlTnTn QD !<>Q ' ' ) ) )#&t':'?'?'A'A#B#BQ#FD  ' . .1EtG\0]/^ _ _ _  ' ' 5 5 5  ' ' ) ) ) ) ) ( ',O,O,O,OrFrRrSc|j9|j|j||||fgdSdS)aT Adds a progress bar update to the shared dict so the dashboard process can use it, only when a dashboard has started :param progress_bar: tqdm progress bar instance :param failed: Whether or not the operation failed or not :param traceback_str: Traceback string, if an exception was raised N)r4r<rh_get_progress_bar_update_dict)r?rorRrSs rrbz)ProgressBarHandler._send_dashboard_updatesg   +   & &)=)-)K)KLZ`bo)p)p)r(s t t t t t , +rc|j}|d}|d}tj}|dr|dn|dr ||dz nd}|r |r||z |z nd} |j| |||r||z ndt ||jz dddt| d |j|jt| |t| znd| +|t| ztnd || nd|j d S) aS Obtain update dictionary with all the information needed for displaying on the dashboard :param progress_bar: tqdm progress bar instance :param failed: Whether or not the operation failed or not :param traceback_str: Traceback string, if an exception was raised :return: Update dictionary rgr5rateelapsedN.rrrF)seconds) idsuccessrgr5 percentageduration remaining started_rawstarted finished_rawfinished tracebackinsights) format_dictrnowr4strr>rsplitrstrftimeDATETIME_FORMATrstripr* get_insights) r?rorRrSdetailsrgr5rr{remaining_times rryz0ProgressBarHandler._get_progress_bar_update_dictsr* CL lnn")&/mwvQXYbQc7mq79;M7M7Mim/4GG%!)t++4*%:+0:a%iiddl 233::3BB1E+NEBB#|<00AAKYKei&G&G&G G Gko . :"In$E$E$EEOOP_```@B6C6O]00222UY 0==?? A A Arr5cF||_|jdS)zW Set a new total for the progress bar :param total: New total N)r5r6r])r?r5s r set_new_totalz ProgressBarHandler.set_new_totals&        r traceback_errc&|j-dtj|j}nt |}|j5t||_|j ddddS#1swxYwYdS)z Set the exception traceback string and notify the progress bar handler that it's ready :param traceback_err: Traceback error Nr) __cause__rMr format_tb __traceback__rr;rr7notify)r?rrSs r set_exceptionz ProgressBarHandler.set_exceptions  " .GGI$7 8S$T$TUUMM ..M  7 @ @+>}+M+MD (  6 = = ? ? ? @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @ @s .BB  B )r+r#)r+N)FNr)__name__ __module__ __qualname__rrboolrrrr r rr@rFr rNrC tqdm_typer^rbryintr Exceptionrrrrr#r##s!$4!/!fj!'+CH~!KSTW=!hs!"0!59!!!!F"d K:K:K:K:Z*9*****"NS>B t t9 td t.6sm tGK t t t tFJAA)ATA5=c]ANRSVX[S[nAAAAB!3!4!!!! @9 @ @ @ @ @ @ @rr#))r8rrrrrtypingrrr r rXr r r mpire.commsr rmpire.exceptionrmpire.insightsr mpire.paramsrr mpire.signalrmpire.tqdm_utilsrr mpire.utilsrmpire.dashboard.dashboardrmpire.dashboard.utilsrmpire.dashboard.managerr ImportErrorrr#rrrrs((((((((########,,,,,,,,,,,,;;;;;;;;00000000//////))))))::::::::77777722222222&&&&&& "AAAAAA::::::@@@@@@@""""   """"" "'i@i@i@i@i@i@i@i@i@i@s"A55 BB