U mdW@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 F/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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) 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/r0r 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) r9r:rr;printrstderrr?rF)r$r,totalZ time_lapseZfull_msgfr r r __call__s zPrintTime.__call__)NN)rJF)r1r2r3r4r&rQr r r r r5as %r5)rr)r4 __future__rr9rr<rCr(rZdiskrrrrrobjectr"r5r r r r s    "