o +if6@sdZddlmZmZddlmZddlZddiZdddZdd d ZGd d d Z Gddde Z Gddde Z GdddZ Gddde e Z Gddde Z     dddZdS)zHImplements the generic progress logger class, and the ProgressBar class.)tqdm tqdm_notebook) OrderedDictNnotebookFoncCs|dkrdndtd<dS)NrTFr)SETTINGS)turnr B/var/www/html/venv/lib/python3.10/site-packages/proglog/proglog.pyr scCs t||kr|S|d|dS)Nz...)len)s max_lengthr r r troncate_strings rc@sTeZdZdZdddZddZdddZd d Zd d Zd dZ ddZ ddZ dS)ProgressLoggerzGeneric class for progress loggers. A progress logger contains a "state" dictionary. Parameters ---------- init_state : dict Dictionary representing the initial state. NcCs4i|_i|_g|_d|_|dur|j|dSdS)Nr)statestoredlogs log_indentupdate)self init_stater r r __init__szProgressLogger.__init__cCs|jd|j|dS)N )rappendr)rmessager r r log&szProgressLogger.logcCsV|dur%t|d}|d|jWddS1swYdSd|jS)Na )openwritejoinr)rfilepathfr r r dump_logs)s  " zProgressLogger.dump_logscKdS)zExecute something after the state has been updated by the given state elements. This default callback does nothing, overwrite it by subclassing. Nr rkwr r r callback0zProgressLogger.callbackcKs |j||jdi|dS)aStore 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. Nr )rrstore_callbackr&r r r store8s zProgressLogger.storecKr%)zExecute something after the store has been updated by the given state elements. This default callback does nothing, overwrite it by subclassing Nr r&r r r r*Dr)zProgressLogger.store_callbackcks:|D]\}}|D]}|di||i|Vq qdS)aIterate 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) Nr )items)rr'fielditerableitr r r iterLs zProgressLogger.itercKs |j||jdi|dS)Nr )rrr(r&r r r __call__Zs zProgressLogger.__call__N) __name__ __module__ __qualname____doc__rrr$r(r+r*r0r1r r r r rs    rc@sjeZdZdZdZ      dddZedd Zd d Zd d Z ddZ dddZ dddZ ddZ dS)ProgressBarLoggeraGeneric 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 NallrcCsxt|||durt}nt|ttfrtdd|D}t|ttfr)t|}||_||_||j d<||_ ||_ dS)Nc Ss"g|] }|t|dddddfqS)Nr)titleindextotalrindent)dict).0br r r sz.ProgressBarLogger.__init__..bars) rrr isinstancelisttupleset ignored_bars logged_barsrmin_time_intervalignore_bars_under)rrrCrHrIrJrKr r r rys   zProgressBarLogger.__init__cCs |jdS)zReturn ``self.state['bars'].``rC)r)rr r r rCs zProgressBarLogger.barscCs,|jdurdS|jdkr||jvS||jvS)NF all_others)rHrCrbarr r r bar_is_ignoreds    z ProgressBarLogger.bar_is_ignoredcCs"|jsdS|jdkr dS||jvS)NFr9T)rIrMr r r bar_is_loggeds   zProgressBarLogger.bar_is_loggedcCs(t|dr t|nd}|duo||jkS)N__len__)hasattrr rK)rr.lengthr r r iterable_is_too_shortsz'ProgressBarLogger.iterable_is_too_shortc sd|vr |dnd|\srS|tdr4didtifdd}|S)aIterate 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_messageNrQ__totalc3st}d}tD]4\}}t}|dks||jkr<dur/did|idid|i|}|Vq jd|krTdid|idid|didS)Nr __message__indexr<r )time enumeraterJrC) last_timeir/now_timerNrVr.rr r new_iterablesz0ProgressBarLogger.iter_bar..new_iterabler )poppopitemrOrTrRr )r bar_prefixr'rar r`r iter_bars   zProgressBarLogger.iter_barcCr%)aExecute 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. Nr )rrNattrvalue old_valuer r r bars_callbackszProgressBarLogger.bars_callbackc Ks&t|ddd}|D]v\}}d|vr|d\}}||r!q ||||jvr6t|dddd|j|<|j||}||rs|dkoI||k}|dksP|rY|j|j|d <n|j|d |_| d |||f|j|j 7_||j||<| ||||q |j ||jd i|dS) NcSs|dd S)Nrr=)endswith)kvr r r sz,ProgressBarLogger.__call__..)key__r:)r;r<r=rr<r=r>z [%s] %s: %sr )sortedr,splitrOrbrCr?rPrr bar_indentrirrr() rr'r,rmrgrNrfrhnew_barr r r r1s,       zProgressBarLogger.__call__)NNNr9rr)rUr2)r3r4r5r6rqrpropertyrCrOrPrTrerir1r r r r r7_s$    . r7c@sLeZdZdZ         ddd Zd d Zd d ZddZddZdS)TqdmProgressBarLoggera^Tqdm-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 NFr9defaultTrc Csjtj|||||| |d||_tdd|jD|_|dkr"td}||_||_|jr0t |_ dSt |_ dS)N)rrCrHrIrKrJcSsg|]}|dfqSr2r )r@rNr r r rB@sz2TqdmProgressBarLogger.__init__..rur) r7r leave_barsrrC tqdm_barsrrprint_messagesrr) rrrCrvrHrIrrxrJrKr r r r*s   zTqdmProgressBarLogger.__init__cCsd||jvr|j|dur|||j|}|j|d|dttt|dd|jd|j|<dS)z:Create a new tqdm bar, possibly replacing an existing one.Nr=r;rnow)r=descpostfixleave)rwclose_tqdm_barrCrr?rstrrv)rrNinfosr r r new_tqdm_barGs  z"TqdmProgressBarLogger.new_tqdm_barcCs&|j||jsd|j|<dSdS)zClose and erase the tqdm bar.N)rwcloserrMr r r r~Ssz$TqdmProgressBarLogger.close_tqdm_barcCs||jvs |j|dur|||dkrJ||kr9|j|d}|r-||kr-||dS|j|||dS|||j||ddS|dkre|j|jtt|d|j|ddSdS)Nr<r=rZrryr)rwrrCr~r set_postfixrr)rrNrfrgrhr=r r r riYs   z#TqdmProgressBarLogger.bars_callbackcKsL|jr d|vr"|dr$|jrt|ddS|j|ddSdSdSdS)Nr)rxrprintrr r&r r r r(js  zTqdmProgressBarLogger.callback) NNFNr9ruTrr) r3r4r5r6rrr~rir(r r r r rt s    rtc@seZdZddZddZdS)RqWorkerProgressLoggercCs0||_d|jjvri|jjd<|jdSdSN progress_data)jobmetasave)rrr r r rss   zRqWorkerProgressLogger.__init__cKs|j|jjd<|jdSr)rrrrr&r r r r(yszRqWorkerProgressLogger.callbackN)r3r4r5rr(r r r r rrs rc@s eZdZ     dddZdS)RqWorkerBarLoggerNr r9rcCs&t||tj||||||ddS)N)rrCrHrIrJ)rrr7)rrrrCrHrIrJr r r rs  zRqWorkerBarLogger.__init__)NNr r9r)r3r4r5rr r r r r~src@seZdZddZdS)MuteProgressBarLoggercCr%)NTr rMr r r rOsz$MuteProgressBarLogger.bar_is_ignoredN)r3r4r5rOr r r r rs rr9cCs,|dkr t|||||dS|durtS|S)NrN)rCrHrIrJrK)rtr)loggerrCrHrIrJrKr r r default_bar_loggersr)r)r )NNr9rr)r6rr collectionsrr[rrrrr7rtrrrrr r r r s(   M.f