U -eL+@sdZdddgZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z m Z ddl Z eddd d d ZGd d d ejZGdddejZedZde_eeZeZeeddde ee eejfeddddZGdddejZ dS)uo Logging utilities for dill. The 'logger' object is dill's top-level logger. The 'adapter' object wraps the logger and implements a 'trace()' method that generates a detailed tree-style trace for the pickling call at log level INFO. The 'trace()' function sets and resets dill's logger log level, enabling and disabling the pickling trace. The trace shows a tree structure depicting the depth of each object serialized *with dill save functions*, but not the ones that use save functions from 'pickle._Pickler.dispatch'. If the information is available, it also displays the size in bytes that the object contributed to the pickle stream (including its child objects). Sample trace output: >>> import dill, dill.tests >>> dill.detect.trace(True) >>> dill.dump_session(main=dill.tests) ┬ M1: ├┬ F2: │└ # F2 [32 B] ├┬ D2: │├┬ T4: ││└ # T4 [35 B] │├┬ D2: ││├┬ T4: │││└ # T4 [50 B] ││├┬ D2: │││└ # D2 [84 B] ││└ # D2 [413 B] │└ # D2 [763 B] └ # M1 [813 B] adapterloggertraceN)partial)TextIOUnion|+`)│u├┬└c@s@eZdZdZddZddZddZdd Zd d Zd d Z dS) TraceAdaptera Tracks object tree depth and calculates pickled object size. A single instance of this wraps the module's logger, as the logging API doesn't allow setting it directly with a custom Logger subclass. The added 'trace()' method receives a pickle instance as the first argument and creates extra values to be added in the LogRecord from it, then calls 'info()'. Usage of logger with 'trace()' method: >>> from dill.logger import adapter as logger #NOTE: not dill.logger.logger >>> ... >>> def save_atype(pickler, obj): >>> logger.trace(pickler, "Message with %s and %r etc. placeholders", 'text', obj) >>> ... cCs ||_dSN)r)selfrrL/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/dill/logger.py__init__tszTraceAdapter.__init__cCs&td|d}|||j|dS)Nz%(prefix)s%(message)s%(suffix)shandler)TraceFormatter setFormatterr addHandler)rr formatterrrrrvs  zTraceAdapter.addHandlercCs|j|dSr)r removeHandler)rrrrrrzszTraceAdapter.removeHandlercCs||fSrr)rmsgkwargsrrrprocess|szTraceAdapter.processcCs8tjj|ddsdS|tjr.d|_g|_nd|_dS)NF)child)dillZ_dillZis_dill isEnabledForloggingINFO _trace_depth _size_stack)rpicklerrrr trace_setups  zTraceAdapter.trace_setupc Os(t|ds tj|f||dS|jdkr.dS|di}|d}d}zH|j}|jj }z||7}Wn t k r|t |7}YnXWnt t fk rYnX|dk r|s|j |n||j 8}||d<|r|jd8_|j|d<||d<|j|f|||s$|jd7_dS)Nr$extra#sizerdepth)hasattrrinfor$get startswith_filetellframer current_frameAttributeErrorlen TypeErrorr%appendpop) rr&rargsrr(Z pushed_objr*framerrrrs:      zTraceAdapter.traceN) __name__ __module__ __qualname____doc__rrrrr'rrrrrrbs rcs2eZdZdZddfdd ZfddZZS)rz Generates message prefix and suffix from record. This Formatter adds prefix and suffix strings to the log message in trace mode (an also provides empty string defaults for normal logs). Nrcstj||z|jj}|dkr$tWntk rBt}YnXzt|j }Wnt k rnd|_ YnX|tdj k|_ dS)NFzutf-8) superrstreamencodingr4localegetpreferredencodingcodecslookupname LookupErroris_utf8)rrr9rrA __class__rrrs zTraceFormatter.__init__csddd}t|dddkr||jdr:|jddd}n"|jdkrJd }n|jd dd }|jsp|td }|d |d<t|drtt |j d d}|j |d?}d||rd|dndf|d<t | |t|S)N)prefixsuffixr+rr)rr r r u├┬- rLr* z [%d %sB]ZKMGTPirM)getattrrr/r+rH translate ASCII_MAPr,intmathlogr*varsupdater?format)rrecordfieldsrLpowerr*rIrrr[s       zTraceFormatter.format)r;r<r=r>rr[ __classcell__rrrIrrsrr Fa)mode)argrareturncCs0t|tst||dSt|r$tjntjdS)u}print a trace through the stack when pickling; useful for debugging With a single boolean argument, enable or disable the tracing. Example usage: >>> import dill >>> dill.detect.trace(True) >>> dill.dump_session() Alternatively, ``trace()`` can be used as a context manager. With no arguments, it just takes care of restoring the tracing state on exit. Either a file handle, or a file name and (optionally) a file mode may be specitfied to redirect the tracing output in the ``with`` block context. A log function is yielded by the manager so the user can write extra information to the file. Example usage: >>> from dill import detect >>> D = {'a': 42, 'b': {'x': None}} >>> with detect.trace(): >>> dumps(D) ┬ D2: ├┬ D2: │└ # D2 [8 B] └ # D2 [22 B] >>> squared = lambda x: x**2 >>> with detect.trace('output.txt', mode='w') as log: >>> log("> D = %r", D) >>> dumps(D) >>> log("> squared = %r", squared) >>> dumps(squared) Arguments: arg: a boolean value, or an optional file-like or path-like object for the context manager mode: mode string for ``open()`` if a file name is passed as the first argument )fileraN) isinstancebool TraceManagerrsetLevelr"r#WARNING)rbrarrrrs'  c@s(eZdZdZddZddZddZdS) rgzDcontext manager version of trace(); can redirect the trace to a filecCs&||_||_|dk |_t|d|_dS)Nwrite)rdraredirectr,file_is_stream)rrdrarrrrs zTraceManager.__init__cCsh|jrLt|jr$t|j|_nt|j|j |_t tt |jt |_t tjt jSr)rkstderr_handlerflushrlr" StreamHandlerrdr FileHandlerrarrrgetEffectiveLevel old_levelrhr#r-)rrrr __enter__ s    zTraceManager.__enter__cGs<t|j|jr8t|jtt|js8|j dSr) rrhrrrkrrrrmrlclose)rexc_inforrr__exit__s    zTraceManager.__exit__N)r;r<r=r>rrsrvrrrrrgs rg)N)!r>__all__rD contextlibrBr"rWos functoolsrtypingrrr str maketransrU LoggerAdapterr Formatterr getLoggerr propagater_StderrHandlerrmrrfPathLikerAbstractContextManagerrgrrrrs($  (H)  (+