W|hf6SrSSKJrJr SSKJr SSKrSS0rSSjrSSjr"S S 5r "S S \ 5r "S S\ 5r "SS5r "SS\ \ 5r "SS\ 5rSSjrg)zHImplements the generic progress logger class, and the ProgressBar class.)tqdm tqdm_notebook) OrderedDictNnotebookFc&US:XaSOS[S'g)NonTFr)SETTINGS)turns I/home/james-whalen/.local/lib/python3.13/site-packages/proglog/proglog.pyrr s$(DL4uHZc4[U5U:aU$USUS-$)Nz...)len)s max_lengths r troncate_stringrs#Q*$1CAkzNU,BCr cP\rSrSrSrS SjrSrS SjrSrSr S r S r S r S r g)ProgressLoggerzGeneric class for progress loggers. A progress logger contains a "state" dictionary. Parameters ---------- init_state : dict Dictionary representing the initial state. Ncz0Ul0Ul/UlSUlUbURR U5 gg)Nr)statestoredlogs log_indentupdate)self init_states r __init__ProgressLogger.__init__s;     ! JJ  j ) "r cZURRSUR-U-5 g)N )rappendr)rmessages r logProgressLogger.log&s" #/7:;r cUb@[US5nURSRUR55 SSS5 gSRUR5$!,(df  g=f)Na )openwritejoinr)rfilepathfs r dump_logsProgressLogger.dump_logs)sS  h$ $)),-%$99TYY' '%$s +A A-c g)zExecute something after the state has been updated by the given state elements. This default callback does nothing, overwrite it by subclassing. Nrkws r callbackProgressLogger.callback0 r c ^URRU5 UR"S0UD6 g)ahStore objects in the logger and trigger ``self.store_callback``. This works exactly like ``logger()``, but the later is meant for simple data objects (text, numbers) that will be sent over the network or written to a file. The ``store`` method expects rather large objects which are not necessarily serializable, and will be used eg to draw plots on the fly. Nr0)rrstore_callbackr1s r storeProgressLogger.store8s' 2 !b!r c g)zExecute something after the store has been updated by the given state elements. This default callback does nothing, overwrite it by subclassing Nr0r1s r r7ProgressLogger.store_callbackDr5r c+j# UR5Hup#UHnU"S0X$0D6 Uv M M g7f)zIterate through a list while updating the state. Examples -------- >>> for username in logger.iter(user=['tom', 'tim', 'lea']): >>> # At every loop, logger.state['user'] is updated >>> print(username) Nr0)items)rr2fielditerableits r iterProgressLogger.iterLs5 "xxzOE#{# *s13c ^URRU5 UR"S0UD6 g)Nr0)rrr3r1s r __call__ProgressLogger.__call__Zs# " r )rrrrN)__name__ __module__ __qualname____firstlineno____doc__rr#r-r3r8r7rArD__static_attributes__r0r r rrs/*<(  "  r rcn\rSrSrSrSrSSjr\S5rSr Sr S r SS jr SS jr S rS rg)ProgressBarLogger_aTGeneric class for progress loggers. A progress logger contains a "state" dictionary. Parameters ---------- init_state : dict Initial state of the logger. bars : None, list, tuple, or dict, optional Either None (will be initialized with no bar) or a list/tuple of bar names (e.g., ['main', 'sub']) which will be initialized with index -1 and no total, or a dictionary (possibly ordered) of bars, of the form `{bar_1: {title: 'bar1', index: 2, total: 23}, bar_2: {...}}`. ignored_bars : None, list of str, or 'all_others', optional Either None (newly met bars will be added) or a list of blacklisted bar names, or 'all_others' to signify that all bar names not already in `self.bars` will be ignored. logged_bars min_time_interval : int or float Time in seconds between progress bar updates. ignore_bars_under : int Nct[RX5 Uc [5nOD[U[[ 45(a)[UVs/sHnU[ USSSSS94PM sn5n[U[[ 45(a [U5nX0lX@l X RS'XPl X`l gs snf)Nr)titleindextotalr"indentbars) rrr isinstancelisttupledictset ignored_bars logged_barsrmin_time_intervalignore_bars_under)rrrWr]r^r_r`bs r rProgressBarLogger.__init__ys 1 <=D tUm , ,"!1BdDQRST!D lT5M 2 2|,L(&! 6!2!2sB5c URS$)zReturn ``self.state['bars'].``rW)r)rs r rWProgressBarLogger.barsszz&!!r cvURcgURS:XaXR;$XR;$)NF all_others)r]rWrbars r bar_is_ignored ProgressBarLogger.bar_is_ignoreds;    $   , .ii' '+++ +r cdUR(dgURS:XagXR;$)NFallT)r^rgs r bar_is_loggedProgressBarLogger.bar_is_loggeds.    &*** *r cp[US5(a [U5OSnUSL=(a X R:$)N__len__)hasattrrr`)rr?lengths r iterable_is_too_short'ProgressBarLogger.iterable_is_too_shorts2")(I">">XDd"I2H2H)HIr c 8^^^^SU;aURS5mOSmUR5ummTRT5(dTRT5(aT$UT-m[ TS5(aT"S0TS-[ T50D6 UUUU4SjnU"5$)aJIterate through a list while updating a state bar. Parameters ---------- bar_prefix : str Bar prefix. Examples -------- >>> for username in logger.iter_bar(user=['tom', 'tim', 'lea']): >>> # At every loop, logger.state['bars']['user'] is updated >>> # to {index: i, total: 3, title:'user'} >>> print (username) bar_messageNrp__totalc 3v># [R"5nSn[T5H[up[R"5nUS:XdX0- TR:a%TbT"S0TS-T"U50D6 T"S0TS-U0D6 UnUv M] TRTSU:wa T"S0TS-U0D6 T"S0TS-US-0D6 g7f)Nr __message__indexrTr0)time enumerater_rW) last_timeir@now_timerhrvr?rs r new_iterable0ProgressBarLogger.iter_bar..new_iterables IA"8,99;F 4t7M7M M".Dk 1;r?CD0C)OQ/0 (I-yy~g&!+,i+,  ,C)OQU+ ,sB6B9r0)poppopitemrirsrqr)r bar_prefixr2rrhrvr?s` @@@r iter_barProgressBarLogger.iter_bars B &&/KK X   s # #t'A'A('K'KO3 8Y ' '  4C)OS]3 4 - - ~r cg)aNExecute a custom action after the progress bars are updated. Parameters ---------- bar Name/ID of the bar to be modified. attr Attribute of the bar attribute to be modified value New value of the attribute old_value Previous value of this bar's attribute. This default callback does nothing, overwrite it by subclassing. Nr0)rrhattrvalue old_values r bars_callbackProgressBarLogger.bars_callbacks& r c 2[UR5SS9nUGHJup4SU;dMURS5upVURU5(aM9UR U5 XPR ;a[ USSSS9UR U'UR UUnURU5(aUS:H=(a XG:nUS:XdU(aURUR US'OUR USUlURS U<S U<S U<35 U=RUR- slX@R UU'URXVXG5 GMM URRU5 UR"S 0UD6 g) Nc4USRS5(+$)NrrU)endswith)kvs r ,ProgressBarLogger.__call__..sbennW6M2Mr )key__rR)rSrTrUr"rTrUrV[z] z: r0)sortedr=splitrirrWr[rmrr# bar_indentrrrr3) rr2r=rrrhrrnew_bars r rDProgressBarLogger.__call__sHrxxz'MNJCs{IIdO &&s++s ii'%)2TSW%XDIIcN IIcN40 %%c**#wGU5FGW37?? #x0*.))C.*BHHc4?@OOt6O', #t$""3e?' ( " r )r`r]rr^r_)NNNrlrr)rF)rGrHrIrJrKrrpropertyrWrirmrsrrrDrLr0r r rNrN_s[.J 36"",+J,\ *r rNcL\rSrSrSrS SjrSrSrSrSr S r g) TqdmProgressBarLoggeri aTqdm-powered progress bar for console or Notebooks. Parameters ---------- init_state : dict Initial state of the logger. bars : None, list, tuple, or dict, optional Either None (will be initialized with no bar) or a list/tuple of bar names (e.g., ['main', 'sub']) which will be initialized with index -1 and no total, or a dictionary (possibly ordered) of bars, of the form `{bar_1: {title: 'bar1', index: 2, total: 23}, bar_2: {...}}`. leave_bars : bool, optional Whether to leave the progress bars displayed upon completion. ignored_bars : None, list of str, or 'all_others', optional Either None (newly met bars will be added) or a list of blacklisted bar names, or 'all_others' to signify that all bar names not already in `self.bars` will be ignored. notebook : bool, optional True will make the bars look nice (HTML) in the Jupyter notebook. It is advised to leave to 'default' as the default can be globally set from inside a notebook with `import proglog; proglog.notebook_mode()`. print_messages : bool If True, every `logger(message='something')` will print a message in the console or notebook. min_time_interval : int or float Time in seconds between progress bar updates. ignore_bars_under : int Nc *[RUUUUUU US9 X0l[URV s/sHoS4PM sn 5UlUS:Xa [ SnX`lXplUR(a [Ul g[Ul gs sn f)N)rrWr]r^r`r_defaultr) rNr leave_barsrrW tqdm_barsr rprint_messagesrr) rrrWrr]r^rrr_r`rhs r rTqdmProgressBarLogger.__init__*s "" !%#// # %$TYY%GYcDkY%GH y  +H ,%)]]M  &Hs Bc XR;a!URUbURU5 URUnURUSUS[ [ [ US55S9URS9URU'g)z:Create a new tqdm bar, possibly replacing an existing one.NrUrSr"now)rUdescpostfixleave)rclose_tqdm_barrWrr[rstrr)rrhinfoss r new_tqdm_bar"TqdmProgressBarLogger.new_tqdm_barGs} >> !s(;(G    $ #"ii.w_Sy1A-BCD// ( sr cURUR5 UR(dSURU'gg)zClose and erase the tqdm bar.N)rcloserrgs r r$TqdmProgressBarLogger.close_tqdm_barSs1 s!!#}}"&DNN3 r c(XR;dURUcURU5 US:XaX4:aQURUSnU(aX5:aURU5 gURUR X4- 5 gURU5 URUR US-5 gUS:XaMURUR [ [U55S9 URUR S5 gg)NrTrUr{r"rr)rrrWrr set_postfixrr)rrhrrrrUs r r#TqdmProgressBarLogger.bars_callbackYs ~~ %4>>#+>+F   c " 7?! #w/en'',NN3'..u/@A!!#&s#**5195 Y  NN3  + +E 0K + L NN3  & &q )r c UR(aQSU;aJUS(a?UR(a[US5 gURR US5 gggg)Nr")rrprintrr)r1s r r3TqdmProgressBarLogger.callbackjsI   IOI}}bm$ 9 . :GO r )rrrrr) NNFNrlrTrr) rGrHrIrJrKrrrrr3rLr0r r rr s=> =:  ' *"/r rc \rSrSrSrSrSrg)RqWorkerProgressLoggerircXlSURR;a40URRS'URR5 ggN progress_data)jobmetasave)rrs r rRqWorkerProgressLogger.__init__ss; $((-- /-/DHHMM/ * HHMMO 0r c ~URURRS'URR5 gr)rrrrr1s r r3RqWorkerProgressLogger.callbackys%)- o&  r )rN)rGrHrIrJrr3rLr0r r rrrs  r rc(\rSrSrSSjrSrg)RqWorkerBarLoggeri~Nr0c ^[RX5 [RUUUUUUS9 g)N)rrWr]r^r_)rrrN)rrrrWr]r^r_s r rRqWorkerBarLogger.__init__s7 ''2"" !%#/ # r )NNr0rlr)rGrHrIrJrrLr0r r rr~s   r rc\rSrSrSrSrg)MuteProgressBarLoggericg)NTr0rgs r ri$MuteProgressBarLogger.bar_is_ignoredsr r0N)rGrHrIrJrirLr0r r rrsr rcFUS:Xa [UUUUUS9$Uc [5$U$)Nrh)rWr]r^r_r`)rr)loggerrWr]r^r_r`s r default_bar_loggerrs<$%#//    $&& r )r))NNrlrr)rKrr collectionsrr|r rrrrNrrrrrr0r r rsN$#  =DJJZjjZc/-c/L   .0A ,-  r