s( d Z g dZddlZddlZddlZddlZddlZddlZddlm Z ddl
m Z m Z ddl
Z
e ddddd Z G d
d ej Z G d d
ej Z ej d Zde_ ee Z ej Ze e dddde e e eej f deddfdZ G d dej Z 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]
)adapterloggertrace N)partial)TextIOUnion|+`) │u ├ ┬ └c 6 e Zd ZdZd Zd Zd Zd Zd Zd 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)
>>> ...
c || _ d S N)r )selfr s ]/builddir/build/BUILD/cloudlinux-venv-1.0.10/venv/lib/python3.11/site-packages/dill/logger.py__init__zTraceAdapter.__init__t s
c t d| }| | | j | d S )Nz %(prefix)s%(message)s%(suffix)shandler)TraceFormatter setFormatterr
addHandler)r r formatters r r zTraceAdapter.addHandlerv sE "#DgVVV Y''' w'''''r c : | j | d S r )r
removeHandler)r r s r r zTraceAdapter.removeHandlerz s !!'*****r c
||fS r )r msgkwargss r processzTraceAdapter.process| s F{r c t j |d sd S | t j rd|_ g |_ d S d |_ d S )NF)child )dill_dillis_dill isEnabledForloggingINFO _trace_depth _size_stack)r picklers r trace_setupzTraceAdapter.trace_setup s` z!!'!77 F W\ * * (#$G "$G #'G r c t |d st j |g|R i | d S |j d S | di }| d }d } |j }|j j } || z
}n"# t $ r |t | z
}Y nw xY wn# t t f$ r Y nw xY w|>|s|j
| n!||j
z }||d< |r|xj dz c_ |j |d< ||d< | j |g|R i | |s|xj dz
c_ d S d S )Nr. extra#sizer' depth)hasattrr infor. get
startswith_filetellframer
current_frameAttributeErrorlen TypeErrorr/ appendpop) r r0 r" argsr# r3
pushed_objr5 frames r r zTraceAdapter.trace s w// K -d - - -f - - - F ' F
7B''^^C((
=%%''DN0E
#
$!
#
#
#E
"
# * D
%#**40000+//111 $f
& A % -g w #''''''' & A % & &s6 %C B C B<9C ;B<<C CCN)
__name__
__module__ __qualname____doc__r r r r$ r1 r r! r r r r b sx " ( ( (+ + + ( ( ( & & & & &r r c 2 e Zd ZdZdd fd
Z fdZ xZS )r z
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).
Nr c V t j |i | |j j }|t n## t $ r t j }Y nw xY w t j | j }|t j d j k | _
d S # t $ r d| _
Y d S w xY w)Nzutf-8F) superr streamencodingr? localegetpreferredencodingcodecslookupnameis_utf8 LookupError)r r rD r# rO __class__s r r zTraceFormatter.__init__ s $)&))) 5 ~.H $$ 5 5 5 244HHH 5 E }X..3H %
g(>(>(C CDLLL ! ! ! DLLLL !s! 2 AAB B('B(c d ddd}t |dd dk ru|j d r|j dz
dz dz }n |j dk rd }n|j d
z
dz d z }| j s | t d z }|d
z |d< t |d rNt t j
|j d
dz }|j |dz z }d||r d| dz ndfz |d< t |
| t | S )N )prefixsuffixr6 r r4 r' r r r
u ├┬- rZ r5
z [%d %sB]KMGTPir[ )getattrr" r: r6 rU translate ASCII_MAPr7 intmathlogr5 varsupdaterM format)r recordfieldsrZ powerr5 rW s r rj zTraceFormatter.format sL "-- 67A & & * *z$$S))
= ,*E1E9 "" ,*E1H<<
; )))44s: %|F8 66 " " [a0011R7E;%(*D *dE4YGENS4H4HWY-Z ZF8 V F###ww~~f%%%r )rG rH rI rJ r rj
__classcell__)rW s @r r r sq '+
E
E
E
E
E
E
E & & & & & & & & &r r r( Fa)modeargrp returnc t | t st | | S t | r t
j n t
j dS )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
)filerp N)
isinstancebool TraceManagerr setLevelr, r- WARNING)rq rp s r r r sO N c4 140000
OOCr s " "H )
(
(
MM#c#cJJ K K PF& F& F& F& F&7( F& F& F&P'& '& '& '& '&W& '& '& '&R
6 " "
,v
'')) > " " ")>C )> )> )>uT63 34 )>c )>TX )> )> )> )>V% % % % %:4 % % % % %r