U Nv fW@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"tjdrtd|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 startswithmaxtr 1/tmp/pip-unpacked-wheel-dylwa62s/joblib/logger.py _squeeze_times rcCst|}d||dfS)Nz%.1fs, %.1fminN@rr r r r format_time"srcCs(t|}|dkrd|dSd|SdS)N<z%4.1fminrz %5.1fsrr r r r short_format_time's rcCsTdtjkr,ddl}|}|jddddnd}tj|||d}|rP|jf||S)Nnumpyr@r)Z precision thresholdZ edgeitems)depthindent)rmodulesrZget_printoptionsZset_printoptionspprintpformat)objrrnpZ print_optionsoutr r r r/s  rc@s<eZdZdZdddZddZdd Zd d Zdd dZdS)Loggerz& Base class for logging messages. rNcCs||_|r|nd|_dS)z Parameters ---------- depth: int, optional The depth of objects printed. name: str, optional The namespace to log to. If None, defaults to joblib. ZjoblibN)r_name)selfrnamer r r __init__Cs zLogger.__init__cCst|jd||fdSNz[%s]: %s)logging getLoggerr"warningr#msgr r r warnOsz Logger.warncCstd||fdSr&)r'infor*r r r r-Rsz Logger.infocCst|jd||fdSr&)r'r(r"debugr*r r r r.Usz Logger.debugrcCst|||jdS)z2Return the formatted representation of the object.)rr)rr)r#rrr r r formatYsz Logger.format)rN)r) __name__ __module__ __qualname____doc__r%r,r-r.r/r r r r r!?s  r!c@s$eZdZdZd ddZd ddZdS) PrintTimez9 Print and log messages while keeping track of time. Nc Cs|dk r|dk rtdt|_|j|_|dk r@tj|d}||_|dk rttj |tj |rt ddD]6}z$t |d||d|dWqvYqvXqvzt ||dWn YnXz:t|d&}|d|d t|jW5QRXWn YnXdS) Nz&Cannot specify both logfile and logdirz joblib.logr z.%iz.1wz Logging joblib python script z ---%s--- ) ValueErrortime last_time start_timeospathjoinlogfilerdirnameexistsrangeshutilmovecopyopenwritectime)r#r>Zlogdirir r r r%es6     $zPrintTime.__init__Fc Cs|s$t|j}d|t|f}n t|j}d|||df}t|tjd|jdk rz(t|jd}t||dW5QRXWn YnXt|_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) r8r9rr:printrstderrr>rE)r#r+totalZ time_lapseZfull_msgfr r r __call__s zPrintTime.__call__)NN)rIF)r0r1r2r3r%rPr r r r r4as %r4)rr)r3 __future__rr8rr;rBr'rZdiskrrrrrobjectr!r4r r r r s    "