֦ir d Z ddgZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
mZ ddlZdZ G d d Zd Zd
Z G d d Zd Zd
ZddZd Z G d d Zd Ze dk( r e yy)a program/module to trace Python program or function execution
Sample use, command line:
trace.py -c -f counts --ignore-dir '$prefix' spam.py eggs
trace.py -t --ignore-dir '$prefix' spam.py eggs
trace.py --trackcalls spam.py eggs
Sample use, programmatically
import sys
# create a Trace object, telling it what to ignore, and whether to
# do tracing or line-counting or both.
tracer = trace.Trace(ignoredirs=[sys.base_prefix, sys.base_exec_prefix,],
trace=0, count=1)
# run the new command using the given tracer
tracer.run('main()')
# make a report, placing output in /tmp
r = tracer.results()
r.write_results(show_missing=True, coverdir="/tmp")
TraceCoverageResults N) monotonicz#pragma NO COVERc e Zd ZddZd Zy)_IgnoreNc |s
t n
t | | _ |sg n,|D cg c]! }t j j | # c}| _ ddi| _ y c c}w )Nz )set_modsospathnormpath_dirs_ignore)selfmodulesdirsds /usr/lib64/python3.12/trace.py__init__z_Ignore.__init__F sW ")SUs7|
#R37*937a +-''*:*:1*=37*9
#Q( *9s &A c || j v r| j | S || j v rd| j |< y| j D ]' }|j |dz sd| j |< y |d| j |< y| j D ]5 }|j |t j
z s&d| j |< y d| j |< y)Nr .r )r r
startswithr r sep)r filename
modulenamemodr s r namesz
_Ignore.namesL s %<<
+ + #'(DLL $ ::C $$S3Y/+, Z(
'(DLL $ A ""1rvv:.+, Z( $% Z NN)__name__
__module__ __qualname__r r r r r r E s ) )r r c t j j | }t j j | \ }}|S ),Return a plausible module name for the path.)r r
basenamesplitext)r
baser exts r _modnamer+ w s5 77 D !DGG$$T*MHc Or c t j j | }d}t j D ]m }t j j | }|j | s4|t | t j k( sTt | t | kD sl|}o |r| t | dz d }n| }t j j | \ }}|j t j d }t j r |j t j d }t j j | \ }}|j d S )r& r Nr ) r r
normcasesysr lenr
splitdrivereplacealtsepr( lstrip)r
comparepathlongestdirr) driver r* s r _fullmodnamer9 ~ s ''""4(KGxxgg s# ! !# &;s3x+@BFF+J3x#g,& CL1$%&''$$T*KE4 << $D yy||BIIs+GG$$T*MHc ??3 r c 4 e Zd Z ddZd Zd ZddZd dZy)
r Nc || _ | j i | _ | j j | _ || _ | j i | _ | j j | _ || _ | j i | _ | j j | _ || _ || _ | j
r] t | j
d 5 }t j | \ }}}d d d | j | j ||| y y # 1 sw Y .xY w# t t t f$ r5}t d| j
d|t j" Y d }~y d }~ww xY w)Nrb)callerszSkipping counts file : file)countscopycounter calledfuncsr= infileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr/ stderr)r rA rD rE r= rF ferrs r r zCoverageResults.__init__ s8 ;; DK{{'') & # !D ++002 << DL||((* ;;
N$++t,39;;q>0FK - DNN6;N PQ -, Xz2
N)-c;BE**N N
Ns0 D0 D$7+D0 $D-)D0 0E9+E44E9c J |j d xr |j d S )z_Return True if the filename does not refer to a file
we want to have reported.
<>)r endswith)r r s r is_ignored_filenamez#CoverageResults.is_ignored_filename s% ""3'BH,=,=c,BBr c | j }| j }| j }|j }|j }|j }|D ] }|j |d || z ||< |D ] }d||< |D ] }d||< y)z.Merge in the data from another CoverageResultsr r N)rA rD r= get) r otherrA rD r= other_countsother_calledfuncs
other_callerskeys r rJ zCoverageResults.update s && ,, || ! - -
C **S!,|C/@@F3K %C K % !C GCL !r c | j rIt t d | j }t | D ] \ }}}t d|d|d| | j rt t d dx}} t | j D ]\ \ \ }
} } \ }
}}|
|k7 rt t d|
d |
}d} |
|
k7 r| |
k7 rt d|
|
} t d | d
| d |d
| ^ i }| j D ]0 \ }}|j |i x}||< | j ||f ||< 2 i }|j
D ]U \ }}| j | r|j d r|d
d }|Ht j j t j j | }t | }n$|}t j |d t | }|r t! | }ni }t# j$ | }t j j' ||dz }t) |d 5 }t+ j, |j. \ }}d
d
d
| j1 |||| \ } } |s8| s Y d
}!~!y
d
}!~!ww xY w)af
Write the coverage results.
:param show_missing: Show lines that had no hits.
:param summary: Include coverage summary per module.
:param coverdir: If None, the results of each module are placed in its
directory, otherwise it is included in the directory
specified.
zfunctions called:z
filename: z, modulename: , funcname: zcalling relationships:r- z***z -->z r z -> z.pycNT)exist_okz.coverr< d z lines cov% module (path)z%5d %3d%% %s (%s)wbr z"Can't save counts files because %sr? ) rD rO sortedr= rA rY itemsrW rV r r
dirnameabspathr+ makedirsr9 _find_executable_linenos linecachegetlinesjoinrG tokenizedetect_encodingreadlinewrite_results_fileintrF rH dumprL r/ rP )"r show_missingsummarycoverdircallsr r funcnamelastfile lastcfilepfilepmodpfunccfilecmodcfuncper_filelineno lines_hitsumscountr7 lnotabsource coverpathfpencoding_n_hitsn_linespercentmrQ rR s" r
write_resultsz CoverageResults.write_results sL G % &$$E28-.*h"J: < 3@ << G * +#% %Hy dll+ =%%u';tUH$G%. $H "IE>i5&8'5) %IdE4GH , $ Hf-5\\(B-G GI* $ Xv,> ?If !,
'~~ /OHe''1 (#CR= ggoobggooh&?@ %h /
C$/ )( 3
1(; ''1F S*x*?@Ih% & 6 6r{{ C ! & " 5 5i6.> M !1& .- &%$ .-
S:S@szzRR
SsB 3#L;"M 89M1M ;M M M M N M??Nc t |d| }d }d } |5 t |d
D ] \ }
} |
|v r"|j
d ||
z | d
z
} |d
z
}n4|
|v r t | vr|j
d |d
z
}n|j
d
|j
| j d ddd | |fS # t $ r,}t d|d|dt j Y d}~yd}~ww xY w# 1 sw Y | |fS xY w)z'Return a coverage results file in path.wr ztrace: Could not open z for writing: z - skippingr? N)r r r r z%5d: z>>>>>> z ) rG rL rO r/ rP enumeratewritePRAGMA_NOCOVER
expandtabs) r r
linesr r r rF rR r r r lines r rq z"CoverageResults.write_results_file( s 4x8G
)% 3 Y&MM'If,="=>aKFqLGv%n.D MM),qLGMM),
dooa 01 !4 " w 1 26=DGJJ
P " w s# B2 BC*2 C';"C""C'*C6)NNNNN)TFNN)r! r" r# r rW rJ r rq r$ r r r r s& =A'+N4C $YSv r c X i }t j | D ] \ }}||vs d||< |S )z:Return dict where keys are lines in the line number table.r )disfindlinestarts)codestrslinenosr r s r _find_lines_from_coder H s8 G''- 6 GFO . Nr c t | | }| j D ]3 }t j | s|j t || 5 |S )z>! NN;q$/ 0 Nr c ` i }t j }t | | 5 }t j |j
}|D ]S \ }}}} }
|t j k( r6|t j k( r#|\ } } | \ }
}t | |
dz D ] }d||< |}U ddd |S # 1 sw Y |S xY w)zReturn a dict of possible docstring positions.
The dict maps line numbers to strings. There is an entry for
line that contains only a string or a part of a triple-quoted
string.
r r N)tokenINDENTrG rn generate_tokensrp STRINGrange)r r r
prev_ttyperQ tokttypetstrstartendr slinescolelineecolis r
_find_stringsr ^ s A J
h *a&&qzz2-0 )E4T $ -"'KE4"%KE4 "5%!) 4 ! 5 J .1
+
H
+
Hs A8B##B-c L t j | 5 }|j }|j }ddd t | d }t | }t || S # 1 sw Y .xY w# t $ r-}t d| d|t j i cY d}~S d}~ww xY w)zAReturn dict where keys are line numbers in the line number table.Nz Not printing coverage data for r> r? exec) rn rG readr rL rO r/ rP compiler r )r rQ progr rR r r s r rj rj u s
]]8
$668DzzH % 46 *D 8 ,D tT "" %
$
!)307:zz C s3 A- A!A- !A*&A- - B#6"B B# B#c ^ e Zd Z ddZd ZddZd Zd Zd Zd Z d Z
d
Z d Z d Z
d
Zy)r Nc
|| _ || _ t || | _ i | _ i | _ d| _ || _ i | _ i | _ i | _
d| _ | rt | _ |r| j | _ y|r| j | _ y|r%|r#| j | _ | j" | _ y|r#| j | _ | j& | _ y|r#| j | _ | j( | _ yd| _ y)ax
@param count true iff it should count number of times each
line is executed
@param trace true iff it should print out each line that is
being counted
@param countfuncs true iff it should just output a list of
(filename, modulename, funcname,) for functions
that were called at least once; This overrides
`count' and `trace'
@param ignoremods a list of the names of modules to ignore
@param ignoredirs a list of the names of directories to ignore
all of the (recursive) contents of
@param infile file from which to read stored counts to be
added into the results
@param outfile file in which to write the results
@param timing true iff timing information be displayed
r Nr )rE rF r ignorerA pathtobasename donothingtrace _calledfuncs_callers
_caller_cache
start_time_timeglobaltrace_trackcallers globaltraceglobaltrace_countfuncsglobaltrace_ltlocaltrace_trace_and_count
localtracelocaltrace_tracelocaltrace_count)
r r r
countfuncs countcallers
ignoremods
ignoredirsrE rF timings
r r zTrace.__init__ s ( j*5
#gDO # < t j | j t j | j t ||| | j s+t j d t j d y y # | j s+t j d t j d w w xY wr )r threadingsettracer r/ r )r r globalslocalss r r z Trace.runctx s ?bG >B6~~ t / / 0 LL)) * ) gv &>> T"""4( "4>> T"""4( "s
B 9Cc d }| j s t j | j ||i |}| j st j d |S # | j st j d w w xY wr )r r/ r r )r funcargskwresults r runfuncz
Trace.runfunc sh ~~ LL)) * #4&2&F>> T"
>> T" "s A $A>c |j }|j }|r t | }nd }|j }d }|| j v r | j | | j | }nd | j |< t j | D cg c] }t j | r| }}t | dk( rt j |d D cg c] } t | t r| }
} t |
dk( r]t j |
d D cg c] } t | d r| } } t | dk( r | d j }|| j |< ||d|}|||fS c c}w c c} w c c} w )Nr r __bases__r )f_code co_filenamer+ co_namer gc
get_referrersr
isfunctionr0
isinstancer hasattrr! )
r framer r r rx clsnamerQ funcsr dictsr classess
r file_module_function_ofz Trace.file_module_function_of s ||## !(+J J<< 4%% %!!$'3 ,,T2'+D t $ !# 0 0 6 3 61 # . .q 1 6E 3
5zQ$&$4$4U1X$> 5$>q *1d 3 $> 5u:?*,*:*:58*D ?*DQ&-a&= !*DG ?7|q(")!*"5"5
4;**40 ")84HX---3 5 ?s E'E,E1c |dk( r>| j | }| j |j }d| j ||f< yy)zkHandler for call events.
Adds information about who called who to the self._callers dict.
callr N)r f_backr )r r whyarg this_func parent_funcs r r z Trace.globaltrace_trackcallers sG
&= 44U;I 66u||DK67DMM; 2 3 r c P |dk( r!| j | }d| j |< yy)zoHandler for call events.
Adds (filename, modulename, funcname) to the self._calledfuncs dict.
r r N)r r )r r r r r s r r z Trace.globaltrace_countfuncs s0
&= 44U;I+,D i ( r c $ |dk( r|j }|j j dd }|r`t | }|R| j j || }|s3| j rt d|d|j | j S yyyy)zHandler for call events.
If the code block being entered is to be ignored, returns `None',
else returns self.localtrace.
r __file__Nz --- modulename: r` )
r f_globalsrY r+ r r r rO r r )r r r r r r r ignore_its r r zTrace.globaltrace_lt s &=< )r r f_linenorA rY r rO r r r
r' rk getliner ) r r r r r r r^ bnamer s r r z Trace.localtrace_trace_and_count' s &= ||//H^^FF"C #{{sA 6 :DKK f$// 9:DGG$$X.E $$Xv6D (eV_," 5dDb) r c |dk( r|j j }|j }| j r%t dt | j z
z d t j j | }t j || }t d||fz d |rt d|d | j S t | j S )Nr r r r r
r- r> ) r r r r rO r r r
r' rk r r )r r r r r r r
r s r r zTrace.localtrace_trace: s &= ||//H^^Ff$// 9:DGG$$X.E $$Xv6D (eV_," 5dDb) r c |dk( rR|j j }|j }||f}| j j |d dz | j |< | j
S )Nr r r )r r r rA rY r )r r r r r r r^ s r r zTrace.localtrace_countK sW &= ||//H^^FF"C #{{sA 6 :DKK r c t | j | j | j | j | j
S )N)rE rF rD r= )r rA rE rF r r )r s r resultsz
Trace.resultsS s2 t{{4;;'+||+/+<+<'+}}6 6r ) r r r r r$ r$ NNFr )r! r" r# r r r r r r r r r r r r r$ r r r r sI DEDH 0 d%
) '.R 8- . & " 6r c dd l } | j }|j ddd |j dd }|j dd d
d |j d
dd
d |j ddd
d |j ddd
d |j d }|j }|j ddd
d |j ddd
d |j d d d |j d!d"d# |j d$d%d
d& |j d'd(d
d) |j d*d+d
d, |j d-d. }|j d/d0g d12 |j d3d0g d42 |j d5d
d6d72 |j d8d9d:; |j d<| j
d=; |j
}|j r*t j d> t j d? fd@}|j D cg c]( }|j dA D ] }|j * c}}|_
|j D cg c]. }|j t j D ]
} || 0 c}}|_ |j ro|j s|j# dB t% |j |j C } | j' |j( |j* |j, S t/ |j0 |j2 |j4 |j6 g s|j# dD |j4 r)|j2 s |j0 r|j# dE |j* r |j2 s|j# dF |j8 |j# dG t; |j2 |j0 |j4 |j6 |j |j |j |j |j< H }
|j> rodd l } |j8 } | jC | \ }
}}|jD g|jF tH _% dI|jD |jL |jN |d dJ}n|j8 g|jF tH _% t jP jS |j8 tH jP d< tU jV |j8 5 }tY |j[ |j8 dK }d d d |j8 dId d dL}|
j] || |
je } |jf s2| j' |j( |j* |j, y y c c}}w c c}}w # 1 sw Y xY w# t^ $ r6}tI j` dMtH jJ d dN| Y d }~d }~wtb $ r Y w xY w)ONr z --versionversionz trace 2.0)actionr z Main optionsz(One of these (or --report) must be givenz-cz--count
store_truezCount the number of times each line is executed and write the counts to .cover for each module executed, in the module's directory. See also --coverdir, --file, --no-report below.)r helpz-tz--tracez3Print each line to sys.stdout before it is executedz-lz --listfuncszKeep track of which functions are executed at least once and write the results to sys.stdout after the program exits. Cannot be specified alongside --trace or --count.z-Tz --trackcallsz^Keep track of caller/called pairs and write the results to sys.stdout after the program exits. Modifiersz-rz--reportzGenerate a report from a counts file; does not execute any code. --file must specify the results file to read, which must have been created in a previous run with --count --file=FILEz-Rz --no-reportz^Do not generate the coverage report files. Useful if you want to accumulate over several runs.z-fz--filez+File to accumulate counts over several runs)r z-Cz
--coverdirzDirectory where the report files go. The coverage report for . will be written to file //.coverz-mz --missingz?Annotate executable lines that were not executed with ">>>>>> "z-sz --summaryz\Write a brief summary for each file to sys.stdout. Can only be used with --count or --reportz-gz--timingzQPrefix each line with the time since the program started. Only used while tracingFiltersz Can be specified multiple timesz--ignore-moduleappendzqIgnore the given module(s) and its submodules (if it is a package). Accepts comma separated list of module names.)r defaultr z --ignore-dirzWIgnore files in the given directory (multiple directories can be joined by os.pathsep).z--moduleFzTrace a module. progname?zfile to run as main program)nargsr argumentszarguments to the programstdlib
platstdlibc t j j t j j | } | j d j d } t j j | S )Nz$prefixz $exec_prefix)r r
expanduser
expandvarsr2 r )s _exec_prefix_prefixs r parse_ignore_dirz main..parse_ignore_dir sX GG rww 1 1! 4 5
IIi ) 1 1., Oww ""r ,z -r/--report requires -f/--file)rE rF zLmust specify one of --trace, --count, --report, --listfuncs, or --trackcallsz8cannot specify both --listfuncs and (--trace or --count)z3--summary can only be used with --count or --reportz3progname is missing: required with the main options)r r r r rE rF r r )r! r __package__
__loader____spec__
__cached__r )r r! r) r, zCannot run file z
because: )4argparseArgumentParser add_argumentadd_argument_group add_mutually_exclusive_group REMAINDER
parse_args
ignore_dir sysconfigget_path
ignore_modulesplitstripr pathsepreportr@ errorr r missingru rv anyr r listfuncs
trackcallsr r r modulerunpy_get_module_detailsr r r/ argvparentloaderr
rg io open_coder r r rL exit
SystemExitr no_report)r- parsergrp_grpoptsr' r r r$ r trB module_namemod_namemod_specr globsr rR r% r& s @@r mainrU Y s
$
$
&F
I{K
#
#N 6 8C T9\& '
T9\F HT= E F T>,7 8
#
#K
0C + + -DdJ|
dM,G H T8> @T<2 3 T;| T;|= > T:l+ ,
#
#I - /C&x! " ^HbG H
< / 1
#. 0
8+=+=+ - D $$X. )),7 # $(#5#5O#5a # ))+AM OD !%N /19LA (*9L + /NDO {{yy LL9 :!DIIF$$T\\4<< OO
DJJH I 4 5 ~~4:: OP ||DJJ JK }} JK
djj$** ??t7I7I ii 5A
;; --K','@'@'M $Hh ((:4>>:CH & , , ' &oo $ "
E
77CH ''//$--8CHHQK dmm, rwwy$--@ - !MM & # " E
ue$ iikG >> dllDLL$--H ONX -, H SXXa[#FGG
sC !-W&&3W,C6W> &W2+,W> 2W;7W> > Y,X88 YYr r ) __doc____all__rG rk r r/ r5 r rn r r r rH timer r r r r r+ r9 r r r r rj r rU r! r$ r r rY s <( %
&
# #0 0d 8l l \
. # T6 T6lPId ZF r