HeL<UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z mZddlmZmZmZmZmZddlmZmZddlmZmZmZmZmZmZmZm Z d Z!e"ed dZ#ere#JeZ$e!rFGd d Z%e%e#d Z#e#JejLedddZe'Z(de)d<ddZ*ddZ+ddZ,nddZ*ddZ,e GddZ-ddZ.GddeZ/y) zBCallback functions and support for sys.monitoring data collection.) annotationsN) dataclass)CodeType FrameType)AnyCallableSet TYPE_CHECKINGcast)short_filename short_stack) AnyCallableTArcTFileDispositionTLineNo TTraceDataTTraceFileData TracerCoreTWarnFnF monitoringc eZdZdZddZddZy)LoggingWrapperz*Wrap a namespace to log all its functions.c ||_||_yN)wrapped namespace)selfrrs 1/usr/lib/python3/dist-packages/coverage/sysmon.py__init__zLoggingWrapper.__init__:s"DL&DNcdfd }|S)Ncxtjd||tj|i|S)N.)logrgetattrr)argskwargsnamers r_wrappedz,LoggingWrapper.__getattr__.._wrapped?s@t~~&avdVF8<=2wt||T2DCFCCr )r&rr'rreturnr)rr(r)s`` r __getattr__zLoggingWrapper.__getattr__>s DOr N)rrrstrr*None)r(r-r*zCallable[..., Any])__name__ __module__ __qualname____doc__rr,r+r rrr7s8 ' r rzsys.monitoringT)fullshort_filenames frame_idszset[int] seen_threadsc tj}ttt j j }||zdzd}|tvr1tj|td|d|dtdD]-}t|d5}t|d|d ||d d d d /y #1swY:xYw) z1Write a message to our detailed debugging log(s).iw07dz New thread  z: )z /tmp/foo.outa:: T)fileflushN) osgetpidr int threadingcurrent_threadidentr6addr$r openprint)msgpidtidtslugfilenamefs rr$r$Msiik3 002889#I*3/ l "   S ! +cU!E7#km_= >  CH h$ CQugRu-ATB C C  C  C Cs B77C ct|trGdt|dd|jdzdt |j d|j dzSt|S)z)Make a customized repr for logged values.z) isinstanceridco_namer co_filenameco_firstlinenorepr)args rarg_reprrZbsl c8 $"S'"&3;;-q)*>#//:=Qs?Q?Q>RRSTU  Cyr cdfd }|S)z%Decorate a function to log its calls.cHtjdfd }|S)Nc d g}t|D]'\}}| |j|dt|)tt |ddj ddj |d|g|}|S#t$r}td|jj d|td j tj| tJtjtjjd #t $rtd YwxYwd}~wwxYw) N=rOr;(z, )z!!r<rz/oops, shutting off events with disabled tool id)zipappendrZr$rTr/join Exception __class__ tracebackformat_exceptionsys_monitoring set_eventssysr COVERAGE_ID ValueError) rr& args_reprsr(rYretexcmethodnamess rr)z0panopticon.._decorator.._wrappedps0!#J%(%5E c<$"))TF!HSM?*CDE2d8B-q(9499Z;P:QQRST --CJ "S]]334Bse<= : :3 ?@AO-999&11#..2L2LaP&OMNO s=A9A== D/AD*6DD*D'$D*&D''D**D/)rrr&rr*r) functoolswraps)rqr)rrs` r _decoratorzpanopticon.._decoratoros% __V $ % .Or )rqrr*rr+rrrus` r panopticonrwls 6r cy)zAWrite a message to our detailed debugging log(s), but not really.Nr+)rHs rr$r$sr cdd}|S)z5Decorate a function to log its calls, but not really.c|Srr+)meths rruzpanopticon.._decoratorsKr )r{rr*rr+rvs rrwrws r c0eZdZUdZded<ded<ded<y) CodeInfoz/The information we want about each code object.booltracingzTTraceFileData | None file_datazdict[int, int] | None byte_to_lineN)r/r0r1r2__annotations__r+r rr}r}s9 M$$''r r}cpi}|jD] \}}}| t||dD]}|||< "|S)z6Make a dict mapping byte code offsets to line numbers.)co_linesrange)codeb2lbstartbendlinenoboffsets rbytes_to_linesrsQ C $ &f   q1 &%G  && Jr ceZdZdZddZddZeddZeddZeddZ ddZ ddZ dd Z e rdd Zndd Zed d ddZed d  ddZed d d ddZed d d ddZed dd dZed dd dZy)! SysMonitorzHPython implementation of the raw data tracer for PEP669 implementations.c|d|_||d|_d|_|||_i|_g|_i|_i|_d|_ddi|_ d|_ d|_ y)NFstartsr) trace_arcsshould_start_contextswitch_contextmyid code_infos code_objects last_lineslocal_event_codes sysmon_onstatsstopped _activity)rtool_ids rrzSysMonitor.__init__sy   OS!CG  02-/0268 a   r ctd|jjD}t|j}dt |dd|d|dS)Nc32K|]}t|ywr)len).0vs r z&SysMonitor.__repr__..s8SV8sz)sumdatavaluesrrT)rpointsfiless r__repr__zSysMonitor.__repr__sN8TYY%5%5%788DII D" Rx7GwgVVr cd|_tJtj|jdt j tj |j}tj}|jrtj|j|j|jz||j|j||j|j||j|j ||j|j"||j$|j&ndtj|j|j||j|j||j$|j(tj+d|_y)zStart this Tracer.FNz coverage.pyT)rri use_tool_idrrspartialregister_callbackeventsrrjPY_START PY_UNWINDsysmon_py_start PY_RESUMEsysmon_py_resume_arcs PY_RETURNsysmon_py_return_arcssysmon_py_unwind_arcsLINEsysmon_line_arcssysmon_line_linesrestart_eventsr)rregisterrs rstartzSysMonitor.starts? )))""499m<$$^%E%EtyyQ&& ??  % % &"2"22  V__d&:&: ; V%%t'A'A B V%%t'A'A B V%%t'A'A B V[[$"7"7 8  % %dii A V__d&:&: ; V[[$"8"8 9%%'r cH|jsytJtj|jdd|_|jj D]#}tj |j|d%i|_tj|jy)zStop this Tracer.NrF)rrirjrrrset_local_events free_tool_id)rrs rstopzSysMonitor.stops~~ )))!!$))Q/**113 @D  + +DIItQ ? @!###DII.r c$|jy)z+The process has forked, clean up as needed.N)rrs r post_forkzSysMonitor.post_forks  r c|jS)zHas there been any activity?rrs ractivityzSysMonitor.activitys ~~r cd|_y)zReset the activity() flag.FNrrs rreset_activityzSysMonitor.reset_activity s r cy)z+Return a dictionary of statistics, or None.Nr+rs r get_statszSysMonitor.get_statssr cftjjjjSz2Get the frame of the Python code we're monitoring.inspect currentframef_backrs r callers_framezSysMonitor.callers_frames'$$&--44;; r cRtjjjSrrrs rrzSysMonitor.callers_frames'')0077 7r r@cd|_|jdxxdz cc<|jjt |}d}d}||j }|j }||j}|jj|}|Qtjj}tr |j}|j||}||j|<|j}|rP|j} | J| |j vrt#|j | <|j | }t%|} nd}d} t'||| |jt |<|j(j+||rt,j.j0} |j2ret4Jt4j7|j8|| j:| j<z| j>z||j@t |<|r7|jBr+|jE}|jF |jH|<yt,j.jJS)z-Handle sys.monitoring.events.PY_START events.TrN)rrr)&rrrgetrTrrrVshould_trace_cacherrrLOG should_tracetracesource_filenamersetrr}rrcrkrrrrirrrrrrrrrWrDISABLE) rrinstruction_offset code_info tracing_coderrLdispframe tracenamerrs rrzSysMonitor.sysmon_py_start!s 8!OO''41 $( +/  $,,L!++I  ''H**..x8D|,,.55!LLE((5948''1::L 00  ,,,DII-+.5DIIi( IIi0 $T* (0$# )DOOBtH %    $ $T *..>>)555"33 (( **+!++ & 8>)) )r cV|j}|j|j|<y)zBHandle sys.monitoring.events.PY_RESUME events for branch coverage.N)rf_linenor)rrrrs rrz SysMonitor.sysmon_py_resume_arcses$ ""$!&r Ncx|j}|jjt|}|h|j\|j j|}|?||j f}ttt|jj||j j|dy)zBHandle sys.monitoring.events.PY_RETURN events for branch coverage.N) rrrrTrrrWr r rrEpop)rrrretvalrr last_linearcs rrz SysMonitor.sysmon_py_return_arcsms ""$OO''41  Y%8%8%D++E2I$ 4#6#6"67SY 3 3488= E4(r rpcj|j}|jj|d}t|try|j j t|}|P|jC|@||j f}ttt|jj|yyyy)zBHandle sys.monitoring.events.PY_UNWIND events for branch coverage.N)rrrrS GeneratorExitrrrTrrWr r rrE)rrr exceptionrrrrs rrz SysMonitor.sysmon_py_unwind_arcs~s ""$OO''t4 i / OO''41  Y%8%8%D$ 4#6#6"67SY 3 3488=%&E r linec|jt|}|j0ttt |jj |tjjS)z;Handle sys.monitoring.events.LINE events for line coverage.) rrTrr r rrErkrr)rr line_numberrs rrzSysMonitor.sysmon_line_linessQOOBtH-    * Wy22 3 7 7 D~~%%%r c2|jt|}d}|jp|j}|jj |}|4||f}t tt|jj|||j|<|S)z=Handle sys.monitoring.events.LINE events for branch coverage.N) rrTrrrrr r rrE)rrrrrorrrs rrzSysMonitor.sysmon_line_arcssOOBtH-     *&&(E++E2I$ +.SY 3 3488=%0DOOE " r )rrAr*r.)r*r-)r*r.)r*r~)r*zdict[str, int] | None)r*r)rrrrAr* MonitorReturn)rrrrArobjectr*r)rrrrAr BaseExceptionr*r)rrrrAr*r)r/r0r1r2rrrwrrrrrrrrrrrrrrr+r rrrsrR@W \2\ / /\    8A*A*F00250 00T"))25)?E) )#) U#>>25>BO> >$>$& &   r r)rHr-r*r.)rYrr*r-)rrz str | Noner*r)rrr*zdict[int, int])0r2 __future__rrsrr?os.pathrkrBrg dataclassesrtypesrrtypingrrr r r coverage.debugr r coverage.typesrrrrrrrrrr%rirrrrr6rr$rZrwr}rrr+r rrsI"  !%7    lD1  %% %M  $N4DEN  %% %#)##$K!UL("C*DP (( (yyr