o
0i-r @ s d dgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z ddlZdZG dd dZdd Zd
d ZG d d dZd
d Zdd ZdddZdd ZG dd d Zdd Zedkrxe dS dS )TraceCoverageResults N) monotonicz#pragma NO COVERc @ s e Zd ZdddZdd ZdS )_IgnoreNc C s: |st nt || _|sg ndd |D | _ddi| _d S )Nc S s g | ]}t j|qS )ospathnormpath.0dr r ,/opt/alt/python310/lib64/python3.10/trace.py
H s z$_Ignore.__init__..z )set_mods_dirs_ignore)selfmodulesdirsr r r
__init__F s
z_Ignore.__init__c C s || j v r
| j | S || jv rd| j |< dS | jD ]}||d r*d| j |< dS q|d u r6d| j |< dS | jD ]}||tj rKd| j |< dS q9d| j |< dS )Nr .r )r r
startswithr r sep)r filename
modulenamemodr r r r
namesL s(
z
_Ignore.namesNN)__name__
__module__ __qualname__r r r r r r
r E s
r c C s" t j| }t j|\}}|S N)r r basenamesplitext)r baser extr r r
_modnamew s r( c C s t j| }d}tjD ] }t j|}||r,|t| t jkr,t|t|kr,|}q |r:| t|d d }n| }t j|\}}|t jd}t j rU|t j d}t j
|\}}| dS )N r r ) r r normcasesysr lenr
splitdrivereplacealtsepr% lstrip)r Z comparepathZlongestdirr& Zdriver r' r r r
_fullmodname~ s$
r2 c @ s> e Zd Z dddZdd Zdd Zdd
d Zdd d
ZdS )r Nc
C s || _ | j d u r i | _ | j | _|| _| jd u r i | _| j | _|| _| jd u r-i | _| j | _|| _|| _| jrz,t| jd}t |\}}}W d n1 sVw Y |
| j |||d W d S t t
tfy } ztd| j|f tjd W Y d }~d S d }~ww d S )Nrb)callerszSkipping counts file %r: %sfile)countscopyZcounter calledfuncsr4 infileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr+ stderr)r r7 r9 r: r4 r; ferrr r r
r s:
zCoverageResults.__init__c C s | do
|dS )N<>)r endswith)r r r r r
is_ignored_filename s z#CoverageResults.is_ignored_filenamec C sp | j }| j}| j}|j }|j}|j}|D ]}||d|| ||< q|D ]}d||< q&|D ]}d||< q/d S Nr r )r7 r9 r4 get) r otherr7 r9 r4 Z other_countsZother_calledfuncsZ
other_callerskeyr r r
r? s
zCoverageResults.updateTFc "
C s | j r!t td | j }t|D ]\}}}td|||f q| jrit td d }} t| jD ]4\\}
} } \}
}}|
|krOt td|
d |
}d} |
|
kr^| |
kr^td|
|
} td| | ||f q4i }| jD ]\}}||i }||< | j||f ||< qni }| D ]\}}| |rq|dr|d d }|d u rt j
t j
|}t
|}n|}t j
|st | t|}|rt|}ni }t|}t j
||d
}t|d }t|j\}}W d n1 sw Y | |||||\} } |r| rtd | | } | | ||f||< q|r=|r=td
t|D ]} || \} } }}td|| q)| jrz)t| jd} t | j| j | jf| d W d W d S 1 scw Y W d S t y }! ztd|! t j d W Y d }!~!d S d }!~!ww d S )Nzfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r) z***z -->z %s.%s -> %s.%sz.pycz.coverr3 d z lines cov% module (path)z%5d %3d%% %s (%s)wbr z"Can't save counts files because %sr5 )!r9 rD sortedr4 r7 rM itemsrK rJ r r dirnameabspathr( existsmakedirsr2 _find_executable_linenos linecachegetlinesjoinr< tokenizedetect_encodingreadlinewrite_results_fileintr; r= dumprA r+ rE )"r Z show_missingsummarycoverdirZcallsr r funcnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr1 lnotabsourceZ coverpathfpencoding_n_hitsn_linesZpercentmrF rG r r r
write_results s
( z CoverageResults.write_resultsc
C s
z t |d|d}W n ty( } ztd||f tjd W Y d }~dS d }~ww d}d} |J t|dD ]9\}
} |
|v rO|d||
| d7 } |d7 }n|
|v rat| vra|d |d7 }n|d
|| d q5W d | |fS 1 s|w Y | |fS ) Nwrm z3trace: Could not open %r for writing: %s - skippingr5 )r r r r z%5d: z>>>>>> z ) r< rA rD r+ rE enumeratewritePRAGMA_NOCOVER
expandtabs) r r linesrj rh rm r; rG rp ro rg liner r r
r` ) s<
z"CoverageResults.write_results_file)NNNNN)TFNr# )r r! r" r rK r? rr r` r r r r
r s
\c C s. i }t | D ] \}}||vrd||< q|S )Nr )disZfindlinestarts)codestrslinenosrn rg r r r
_find_lines_from_codeI s r c C s6 t | |}| jD ]}t|r|t|| q |S r# )r co_constsinspectZiscoder? _find_lines)r} r~ r cr r r
r S s
r c C s i }t j}t| |d;}t|j}|D ])\}}}} }
|t jkr<|t jkr<|\} } | \}
}t| |
d D ]}d||< q5|}qW d |S 1 sJw Y |S )Nrt r )tokenINDENTr< r] generate_tokensr_ STRINGrange)r rm r Z
prev_ttyperF tokZttypeZtstrstartendr{ ZslineZscolZelineZecolir r r
_find_strings_ s$
r c
C s z t | }| }|j}W d n1 sw Y W n ty> } ztd| |f tjd i W Y d }~S d }~ww t|| d}t | |}t
||S )Nz%Not printing coverage data for %r: %sr5 exec) r] r< readrm rA rD r+ rE compiler r )r rF progrm rG r} r~ r r r
rY v s&
rY c @ sv e Zd Z d ddZdd Zd d
d Zd d
Zdd Zdd Zdd Z dd Z
dd Z dd Z dd Z
d d ZdS ) r r r r NFc
C s || _ || _t||| _i | _i | _d| _|| _i | _i | _ i | _
d | _ | r+t | _ |r3| j
| _d S |r;| j| _d S |rI|rI| j| _| j| _d S |rU| j| _| j| _d S |ra| j| _| j| _d S d| _d S rL )r: r; r ignorer7 Zpathtobasename donothingtrace _calledfuncs_callers
_caller_cache
start_time_timeglobaltrace_trackcallers globaltraceglobaltrace_countfuncsglobaltrace_ltlocaltrace_trace_and_count
localtracelocaltrace_tracelocaltrace_count)
r ri r
countfuncs countcallers
ignoremods
ignoredirsr: r; timingr r r
r s8
zTrace.__init__c C s dd l }|j}| ||| d S )Nr )__main____dict__runctx)r cmdr dictr r r
run s z Trace.runc C s |d u ri }|d u r i }| j st| j t| j zt||| W | j s2td td d S d S | j sBtd td w w r# )r threadingsettracer r+ r )r r globalslocalsr r r
r s
z Trace.runctxc O sP d }| j s t| j z||i |}W | j s td |S | j s'td w w r# )r r+ r r )r funcargskwresultr r r
runfunc s
z
Trace.runfuncc
C s |j }|j}|r
t|}nd }|j}d }|| jv r&| j| d ur%| j| }nCd | j|< dd t|D }t|dkridd t|d D }t|dkridd t|d D } t| dkri| d j}|| j|< |d ursd||f }|||fS )Nc S s g | ] }t |r|qS r )r Z
isfunction)r rF r r r
r
z1Trace.file_module_function_of..r c S s g | ] }t |tr|qS r )
isinstancer r
r r r
r r r c S s g | ] }t |d r|qS ) __bases__)hasattr)r r r r r
r r z%s.%s) f_code co_filenamer( co_namer gcZ
get_referrersr, r )
r framer} r r re ZclsnameZfuncsZdictsZclassesr r r
file_module_function_of s.
z Trace.file_module_function_ofc C s6 |dkr| |}| |j}d| j||f< d S d S Ncallr )r f_backr )r r whyarg this_funcZ parent_funcr r r
r s
z Trace.globaltrace_trackcallersc C s& |dkr| |}d| j|< d S d S r )r r )r r r r r r r r
r s
z Trace.globaltrace_countfuncsc C sr |dkr7|j }|jdd }|r5t|}|d ur1| j||}|s3| jr.td||jf | j S d S d S d S d S )Nr __file__z! --- modulename: %s, funcname: %s)
r f_globalsrM r( r r r rD r r )r r r r r} r r Z ignore_itr r r
r s$ zTrace.globaltrace_ltc C s |dkr?|j j}|j}||f}| j|dd | j|< | jr*tdt | j dd tj
|}td||t ||f dd | j
S ) Nr{ r r %.2f r
%s(%d): %sr) )r r f_linenor7 rM r rD r r r r$ rZ getliner )r r r r r rg rO bnamer r r
r ( s
z Trace.localtrace_trace_and_countc C sd |dkr/|j j}|j}| jrtdt | j dd tj|}td||t
||f dd | j S )Nr{ r r r r r) ) r r r r rD r r r r$ rZ r r )r r r r r rg r r r r
r 7 s
zTrace.localtrace_tracec C s< |dkr|j j}|j}||f}| j|dd | j|< | jS )Nr{ r r )r r r r7 rM r )r r r r r rg rO r r r
r D s zTrace.localtrace_countc C s t | j| j| j| j| jdS )N)r: r; r9 r4 )r r7 r: r; r r )r r r r
resultsL s
z
Trace.results) 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 s
2
)
c s dd l } | }|jdddd |dd}|jdd d
d d |jd
dd
dd |jddd
dd |jddd
dd |d}| }|jddd
dd |jddd
d d |jd d d d |jd!d"d#d |jd$d%d
d&d |jd'd(d
d)d |jd*d+d
d,d |d-d.}|jd/d0g d1d2 |jd3d0g d4d2 |jd5d
d6d7d2 |jd8d9d:d; |jd<| jd=d; | }|jrt d>t d? fd@dAdBdC |j
D |_
fdDdC|jD |_|j r|j s|
dE t|j |j dF}||j|j|jS t|j|j|j|jgs
|
dG |jr |js|jr |
dH |jr+|js+|
dI |jd u r6|
dJ t|j|j|j|j|j
|j|j |j |jdK }zk|jrudd l }|j}| |\} }
} | j g|j t _!dL| j |
j"|
j#|
d dM} n:|jg|j t _!t$j%&|jt j%d< t'(|j}
t)|
* |jdN} W d n 1 sw Y |jdLd d dO} |+| | | W n( t,y } zt -dPt j!d |f W Y d }~nd }~w t.y Y nw |/ }|j0s||j|j|j d S d S )QNr 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.Z 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 tracingZFiltersz 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 programZstdlibZ
platstdlibc s4 t jt j| } | dd } t j| S )Nz$prefixz $exec_prefix)r r
expanduser
expandvarsr. r )s) _exec_prefix_prefixr r
parse_ignore_dir s z main..parse_ignore_dirc S s$ g | ]}| d D ]}| q qS ),)splitstrip)r r r r r r
r s
zmain..c s& g | ]}| tjD ]} |q
qS r )r r pathsep)r r r )r r r
r s
z -r/--report requires -f/--file)r: r; 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 r: r; r r )r r __package__
__loader____spec__
__cached__r )r r r r z Cannot run file %r because: %s)1argparseArgumentParser add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER
parse_argsZ
ignore_dir sysconfigZget_pathZ
ignore_moduleZreportr6 errorr rr Zmissingrc rd anyr ri Z listfuncsZ
trackcallsr r r modulerunpyZ_get_module_detailsr r r+ argvparentloaderr r rU io open_coder r r rA exit
SystemExitr Z no_report)r parserZgrpZ_grpZoptsr tr Z module_nameZmod_nameZmod_specr} Zglobsrl rG r )r r r r
mainR s
$r r r# ) __all__r rZ r r+ r r r] r r r| r= timer r r rx r r( r2 r r r r rY r r r r r r r
s@ 2 0
O