wi,ddlZddlZddlmZmZddlmZddlmZ ddl mZ ddl m Z mZmZddlmZmZddlmZ ddlmZdZn#e$rdZd ZYnwxYwdd lmZdd lmZdd lmZd Zee dfZ ej!e"Z#GddZ$Gdde$eZ%erGdde$eZ&nGdde$Z&Gdde$eZ'Gdde$eZ(de e)dee$fdZ*GddZ+GddZ,dS) N)redirect_stderrredirect_stdout)StringIO)Lock)OptionalTupleType)TqdmExperimentalWarningtqdm)r TF)mp_dill)DisableKeyboardInterruptSignal)create_sync_managerstdTqdmPositionRegisterceZdZdZdZededdfdZddeddffd Z d eddfd Z dd e eddfd Z ede ddfdZxZS) TqdmMpirez7 Abstract class for tqdm classes that are used in mpireFmainreturnNc||_dS)z Marks this progress bar as the main progress bar :param main: Whether this progress bar is the main progress bar N)main_progress_bar)clsrs h/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/mpire/tqdm_utils.pyset_main_progress_barzTqdmMpire.set_main_progress_bar!s!%nct||j|jkr|dSdS)z Update the progress bar. Forces a final refresh when the progress bar is finished. :param n: Number of steps to update the progress bar with N)superupdatertotal final_refresh)selfr __class__s rrzTqdmMpire.update*sJ q 6TZ       rr c<||_|dS)z Update the total number of steps of the progress bar. Forces a refresh to show the new total. :param total: Total number of steps N)r refresh)r"r s r update_totalzTqdmMpire.update_total4s   rhighest_progress_bar_positioncV||dS)a* Final refresh of the progress bar. This function is called when the progress bar is finished. It should perform a final refresh of the progress bar and close it. :param highest_progress_bar_position: Highest progress bar position in case of multiple progress bars N)r%closer"r's rr!zTqdmMpire.final_refresh=s"  roptionsctt5tt5|di|dddn #1swxYwYddddS#1swxYwYdS)z Check whether the options passed to the tqdm class are valid. This function should raise an exception when the options are invalid. :param options: Options passed to the tqdm class N)rrrrr+s r check_optionszTqdmMpire.check_optionsGsXZZ ( (  /(***E*E   CNN'NNN                                 s4A% A  A% A A%A A%%A),A))rN)__name__ __module__ __qualname____doc__r classmethodboolrintrr&rr!dictr/ __classcell__r#s@rrrsAA%%$%%%[%!!!D!!!!!!#$8C=TXDT[rrc0eZdZdZddeeddfdZdS) TqdmMpireStdz2 A tqdm class that shows a standard progress bar. Nr'rc|d|_|jr$|$|jd|dzzdSdSdS)a Final refresh of the progress bar. This function is called when the progress bar is finished. It should perform a final refresh. When we're using a standard progress bar and this is the main progress bar, we add as many newlines as the highest progress bar position, such that new output is added after the progress bars. :param highest_progress_bar_position: Highest progress bar position in case of multiple progress bars TN r)r%disablerfpwriter*s rr!zTqdmMpireStd.final_refreshVsa   ! F&C&O GMM$"?!"CD E E E E E F F&O&Orr0)r1r2r3r4rr7r!r-rrr<r<SsO<< F F8C= FTX F F F F F Frr<cDeZdZdZededdfdZdfd ZxZS) TqdmMpireRichz. A tqdm class that shows a rich progress bar. r+rNc|}d|vrddi|d<n d|dd<tj5tjdt|di|ddddS#1swxYwYdS)a Check whether the options passed to the tqdm class are valid. This function should raise an exception when the options are invalid. For rich progress bars we disable the progress bar, because we don't want to show the progress bar in the terminal. For some reason, redirecting stdout/stderr makes the rich progress bar not work properly afterwards. :param options: Options passed to the tqdm class r+r?TignoreNr-)copywarningscatch_warnings simplefilterr r.s rr/zTqdmMpireRich.check_optionsjsllnnG''&/%6 ""04 "9-(**  %h0GHHHg                  s#A//A36A3cltj|i||jdS)z Display the progress bar and force a refresh of the widget. The refresh is needed to show the final update. N)rdisplay_progr%r"argskwargsr#s rrKzTqdmMpireRich.display~s: EGGOT ,V , , , J   rrN) r1r2r3r4r5r8r/rKr9r:s@rrCrCgso<<        & ! ! ! ! ! ! ! ! ! !rrCceZdZddZdS)rCrNc td)NzErich is not installed. Please install rich to use rich progress bars.) ImportErrorr"rNrOs r__init__zTqdmMpireRich.__init__seff frrP)r1r2r3rUr-rrrCrCs. g g g g g grcXeZdZdZd fd Zdeddffd ZededdfdZ xZ S) TqdmMpireNotebookz4 A tqdm class that shows a GUI widget in notebooks. rNcl|jstdddtj|i|dS)z In case we're running tqdm in a notebook we need to apply a dirty hack to get progress bars working. Solution adapted from https://github.com/tqdm/tqdm/issues/485#issuecomment-473338308  T)endflushN)rprintrrUrMs rrUzTqdmMpireNotebook.__init__sG % + #2T * * * *$)&)))))rr ct||jjd_t|S)z Update the total number of steps of the progress bar. Forces a refresh to show the new total. In a notebook we also need to update the max value of the progress bar widget. :param total: Total number of steps r) containerchildrenmaxrr&)r"r r#s rr&zTqdmMpireNotebook.update_totals/*/"&ww##E***rr+cH|}d|d<|di|dS)aa Check whether the options passed to the tqdm class are valid. This function should raise an exception when the options are invalid. For notebook progress bars we set display to false, because redirecting stdout/stderr doesn't work for notebook widgets. :param options: Options passed to the tqdm class FrKNr-)rFr.s rr/zTqdmMpireNotebook.check_optionss0,,.."  grrP) r1r2r3r4rUr7r&r5r8r/r9r:s@rrWrWs>>****** +# +$ + + + + + + D T   [     rrWc,eZdZdZdfd ZddZxZS)TqdmMpireDashboardOnlyz A tqdm class that gives no output, but will still update the internal progress-bar attributes that the dashboard relies on. rNc\t|d<tj|i|dS)z> Set the file to a StringIO object so that no output is given fileN)rrrUrMs rrUzTqdmMpireDashboardOnly.__init__s2!v$)&)))))rcdS)z Don't display anything Nr-rTs rrKzTqdmMpireDashboardOnly.displays rrP)r1r2r3r4rUrKr9r:s@rrdrds[ ******         rrdprogress_bar_stylerc|t}|dkrtS|dkrtS|dkrtS|dkrtSt d|d)a8 Get the tqdm class to use based on the progress bar style :param progress_bar_style: The progress bar style to use. Can be one of ``None``, ``std``, or ``notebook`` :return: A tuple containing the tqdm class to use and a boolean indicating whether the progress bar is a notebook widget Nrrichnotebook dashboardzInvalid progress bar style: z2. Use either None (=default), "std", or "notebook")PROGRESS_BAR_DEFAULT_STYLEr<rCrWrd ValueError)rhs rget_tqdmros!7U"" v % % z ) )  { * *%%M8JMMMNN NrcReZdZdZdeddfdZdedefdZdeefdZ d d Z dS) rz Class that keeps track of all the registered progress bar positions. Needed to properly display multiple tqdm progress bars use_dillrNcd|rtjn t|_d|_dS)zP :param use_dill: Whether dill is used as serialization library N)r rmp_Locklockhighest_position)r"rqs rrUzTqdmPositionRegister.__init__s,'/=GLNNNGII $rpositionc|j5|jdu}|j ||jkr||_dddn #1swxYwY|S)z Register new progress bar position. Returns True when it's the first one to register :param position: Progress bar position :return: Whether this progress bar is the first one to register Nrtru)r"rv first_ones rregister_progress_bar_positionz3TqdmPositionRegister.register_progress_bar_positionsY 1 1-5I$,4;P0P0P(0% 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 s #7;;cR|j5|jcdddS#1swxYwYdS)zu Obtain the highest registered progress bar position :return: Highest progress bar position Nrxr"s r!get_highest_progress_bar_positionz6TqdmPositionRegister.get_highest_progress_bar_positionsv Y ) )( ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s   cT|j5d|_ddddS#1swxYwYdS)z= Reset the registered progress bar positions Nrxr|s rreset_progress_bar_positionsz1TqdmPositionRegister.reset_progress_bar_positionssxY ) )$(D ! ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) ) )s !!rP) r1r2r3r4r6rUr7rzrr}rr-rrrrs %%$%%%% s t    )8C=))))))))))rceZdZdZdZdZdZededefdZ ed dZ ede fdZ ede ddfd Z dS) TqdmManagerzbTqdm manager wrapper for syncing multiple progress bars, independent of process start method used.Nrqrc|jdStdt5t ||_|jdt|j|j |_|j||_ dddn #1swxYwYdS)z Sets up and starts the tqdm manager :param use_dill: Whether dill is used as serialization library :return: Whether the manager was started NFzStarting TQDM managerrT) LOCKloggerdebugr rMANAGERregisterrstartrPOSITION_REGISTER)rrqs r start_managerzTqdmManager.start_managers  8 5 ,---, - - O O-h77CK K !79M N N N K     {''))CH$'K$D$DX$N$NC !  O O O O O O O O O O O O O O OtsB C  C C cb|jd|_d|_d|_dS)z( Stops the tqdm manager N)rshutdownrrrs r stop_managerzTqdmManager.stop_manager's4   $rc|j|jfS)z Obtains the connection details of the tqdm manager. These details are needed to be passed on to child process when the start method is either forkserver or spawn. :return: TQDM lock and position register rrrs rget_connection_detailsz"TqdmManager.get_connection_details1sx...rtqdm_connection_detailsc$|\|_|_dS)z| Sets the tqdm connection details. :param tqdm_connection_details: TQDM lock and position register Nr)rrs rset_connection_detailsz"TqdmManager.set_connection_details;s+B'#'''rrP)r1r2r3r4rrrr5r6rrTqdmConnectionDetailsrrr-rrrrsllG DTd[.%%%[%/'<///[/B=RBW[BBB[BBBrr)-loggingrG contextlibrriormultiprocessingrrsmultiprocessing.synchronizeLockTypetypingrrr r r tqdm_std tqdm.notebook tqdm_notebook tqdm.rich tqdm_richRICH_AVAILABLErS mpire.contextr mpire.signalr mpire.utilsrrmr getLoggerr1rrr<rCrWrdstrrorrr-rrrs77777777++++++888888((((((((((:::::::://////++++++NNINNN"!!!!!777777++++++"h(>>?  8 $ $44444444nFFFFF9hFFF&#g!!!!! 9!!!!>ggggg ggg $$$$$ =$$$N      Y    N#N4 ?NNNN.*)*)*)*)*)*)*)*)Z;B;B;B;B;B;B;B;B;B;BsA AA