3 d @sdZddlmZddlZddlZddlZddlZddlZddlZddl m Z ddZ dd Z d d Z dd dZGdddeZGdddeZdS)zE Helpers for logging. This module needs much love to become useful. )print_functionN)mkdirpcCs"tjjdrtd|dS|SdS)zRemove .1s to the time under Windows: this is the time it take to stat files. This is needed to make results similar to timings under Unix, for tests winrg?N)sysplatform startswithmax)tr //tmp/pip-build-v9q4h5k9/joblib/joblib/logger.py _squeeze_times r cCst|}d||dfS)Nz%.1fs, %.1fmingN@)r )r r r r format_time"srcCs(t|}|dkrd|dSd|SdS)N<z%4.1fmingN@z %5.1fs)r )r r r r short_format_time's rcCsTdtjkr,ddl}|j}|jddddnd}tj|||d}|rP|jf||S)Nnumpyr@r)Z precision thresholdZ edgeitems)depthindent)rmodulesrZget_printoptionsZset_printoptionspprintpformat)objrrnpZ print_optionsoutr r r r/s  rc@s4eZdZdZd ddZddZddZdd d Zd S)Loggerz& Base class for logging messages. rcCs ||_dS)z Parameters ---------- depth: int, optional The depth of objects printed. N)r)selfrr r r __init__CszLogger.__init__cCstjd||fdS)Nz[%s]: %s)loggingwarning)rmsgr r r warnLsz Logger.warncCstjd||fdS)Nz[%s]: %s)r!debug)rr#r r r r%Osz Logger.debugrcCst|||jdS)z2Return the formatted representation of the object.)rr)rr)rrrr r r formatSsz Logger.formatN)r)r)__name__ __module__ __qualname____doc__r r$r%r&r r r r r?s  rc@s$eZdZdZd ddZd ddZdS) PrintTimez9 Print and log messages while keeping track of time. NcCs |dk r|dk rtdtj|_|j|_|dk r@tjj|d}||_|dk rttjj |tjj |rxDt ddD]6}y$t j |d||d|dWqxYqxXqxWyt j||dWn YnXy:t|d&}|jd|jd tj|jWdQRXWn YnXdS) Nz&Cannot specify both logfile and logdirz joblib.logr z.%iz.1wz Logging joblib python script z ---%s--- ) ValueErrortime last_time start_timeospathjoinlogfilerdirnameexistsrangeshutilmovecopyopenwritectime)rr5Zlogdirir r r r _s4      $zPrintTime.__init__FcCs|s$tj|j}d|t|f}n tj|j}d|||df}t|tjd|jdk ry(t|jd}t||dWdQRXWn YnXtj|_dS)zr Print the time elapsed between the last call and the current call, with an optional message. z%s: %sz%s: %.2fs, %.1f minr)fileNa) r/r0rr1printrstderrr5r<)rr#totalZ time_lapseZfull_msgfr r r __call__s  zPrintTime.__call__)NN)r@F)r'r(r)r*r rGr r r r r+[s %r+)rr)r* __future__rr/rr2r9r!rZdiskrr rrrobjectrr+r r r r s