U
,at @ s d dgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z
ddlZdZG dd dZdd Zd
d ZG d d dZd
d Zdd ZdddZdd ZG dd d Zdd Zedkre dS )TraceCoverageResults N) monotonicz#pragma NO COVERc @ s e Zd ZdddZdd ZdS )_IgnoreNc C s: |s
t nt || _|s g n dd |D | _ddi| _d S )Nc S s g | ]}t j|qS )ospathnormpath.0dr r */opt/alt/python38/lib64/python3.8/trace.py
<listcomp>G s z$_Ignore.__init__.<locals>.<listcomp>z<string> )set_mods_dirs_ignore)selfmodulesdirsr r r
__init__E s
z_Ignore.__init__c C s || j kr| j | S || jkr,d| j |< dS | jD ]"}||d r2d| j |< dS q2|d krld| j |< dS | jD ]$}||tj rrd| j |< dS qrd| j |< dS )Nr .r )r r
startswithr r sep)r filename
modulenamemodr r r r
namesK s$
z
_Ignore.names)NN)__name__
__module__ __qualname__r r r r r r
r D s
r c C s t j| }t j|\}}|S N)r r basenamesplitext)r baser extr r r
_modnamev s r' c C s t j| }d}tjD ]@}t j|}||r|t| t jkrt|t|kr|}q|rr| t|d d }n| }t j|\}}|t jd}t j r|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
r1 c @ s: e Zd ZdddZdd Zdd Zdd
d Zdd d
ZdS )r Nc
C s || _ | j d kri | _ | j | _|| _| jd kr8i | _| j | _|| _| jd krZi | _| j | _|| _|| _| jrz@t| jd}t |\}}}W 5 Q R X |
| ||| W n@ t t
tfk
r } z td| j|f tjd W 5 d }~X Y nX d S )NrbzSkipping counts file %r: %sfile)countscopyZcounter calledfuncscallersinfileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr* stderr)r r5 r7 r9 r8 r: ferrr r r
r s2
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 sn | j }| j}| j}|j }|j}|j}|D ] }||d|| ||< q(|D ] }d||< qJ|D ] }d||< q\d S Nr r )r5 r7 r8 get) r otherr5 r7 r8 Z other_countsZother_calledfuncsZ
other_callerskeyr r r
r> s
zCoverageResults.updateTFc "
C s | j r@t td | j }t|D ] \}}}td|||f q"| jrt td d }} t| jD ]h\\}
} } \}
}}|
|krt td|
d |
}d} |
|
kr| |
krtd|
|
} td| | ||f qfi }| jD ].\}}||i }||< | j||f ||< qi }| D ]\}}| |r0q|drH|d d }|d krpt j
t j
|}t
|}n$|}t j
|st | t|}|rt|}ni }t|}t j
||d
}t|d }t|j\}}W 5 Q R X | |||||\} } |r| rtd | | } | | ||f||< q|rt|rttd
t|D ]&} || \} } }}td|| qL| jrz6t| jd } t | j| j | jf| d W 5 Q R X W n6 t k
r }! ztd|! t j d W 5 d }!~!X Y nX d S )Nzfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r( z***z -->z %s.%s -> %s.%sz.pycz.coverr2 d z lines cov% module (path)z%5d %3d%% %s (%s)wbr z"Can't save counts files because %sr3 )!r7 rC sortedr8 r5 rL itemsrJ rI r r dirnameabspathr' existsmakedirsr1 _find_executable_linenos linecachegetlinesjoinr; tokenizedetect_encodingreadlinewrite_results_fileintr: r< dumpr@ r* rD )"r Z show_missingsummarycoverdirZcallsr r funcnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr0 lnotabsourceZ coverpathfpencoding_n_hitsn_linesZpercentmrE rF r r r
write_results s
z CoverageResults.write_resultsc
C s zt |d|d}W n> tk
rP } z td||f tjd W Y dS d }~X Y nX d}d} | t|dD ]r\}
} |
|kr|d||
| d7 } |d7 }n.|
|krt| kr|d |d7 }n
|d
|| d qjW 5 Q R X | |fS ) Nwrl z3trace: Could not open %r for writing: %s - skippingr3 )r r r r z%5d: z>>>>>> z ) r; r@ rC r* rD enumeratewritePRAGMA_NOCOVER
expandtabs) r r linesri rg rl r: rF ro rn rf liner r r
r_ ( s.
z"CoverageResults.write_results_file)NNNNN)TFN)N)r r r! r rJ r> rq r_ r r r r
r s
\c C s, i }t | D ]\}}||krd||< q|S )Nr )disZfindlinestarts)codestrslinenosrm rf r r r
_find_lines_from_codeH s
r c C s4 t | |}| jD ] }t|r|t|| q|S r" )r co_constsinspectZiscoder> _find_lines)r| r} r~ cr r r
r R s
r c C s i }t j}t| |dj}t|j}|D ]R\}}}} }
|t jkrv|t jkrv|\} } | \}
}t| |
d D ] }d||< qh|}q(W 5 Q R X |S )Nrs r )tokenINDENTr; r\ generate_tokensr^ STRINGrange)r rl r Z
prev_ttyperE tokZttypeZtstrstartendrz ZslineZscolZelineZecolir r r
_find_strings^ s
r c
C s z(t | }| }|j}W 5 Q R X W n@ tk
rh } z"td| |f tjd i W Y S d }~X Y nX t|| d}t | |}t
||S )Nz%Not printing coverage data for %r: %sr3 exec) r\ r; readrl r@ rC r* rD compiler r )r rE progrl rF r| r} r r r
rX u s
rX c @ sv e Zd Zd ddZdd Zd d
d Zd d
Zde_dd Zdd Z dd Z
dd Z dd Z dd Z
dd Zd d ZdS )!r r r r NFc
C s || _ || _t||| _i | _i | _d| _|| _i | _i | _ i | _
d | _ | rTt | _ |rb| j
| _nZ|rp| j| _nL|r|r| j| _| j| _n2|r| j| _| j| _n |r| j| _| j| _nd| _d S rK )r9 r: r ignorer5 Zpathtobasename donothingtrace _calledfuncs_callers
_caller_cache
start_time_timeglobaltrace_trackcallers globaltraceglobaltrace_countfuncsglobaltrace_ltlocaltrace_trace_and_count
localtracelocaltrace_tracelocaltrace_count)
r rh r
countfuncs countcallers
ignoremods
ignoredirsr9 r: timingr r r
r s6
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 sh |d kr i }|d kri }| j s6t| j t| j zt||| W 5 | j sbtd td X d S r" )r threadingsettracer r* r )r r globalslocalsr r r
r s
z Trace.runctxc O s t | dkr| ^}}} nV| s&tdnHd|krZ|d}| ^}} dd l}|jdtdd ntdt | d d }|jst|j z|| |}W 5 |jstd X |S ) N z8descriptor 'runfunc' of 'Trace' object needs an argumentfuncr z0Passing 'func' as keyword argument is deprecated)
stacklevelz7runfunc expected at least 1 positional argument, got %dr )
r+ TypeErrorpopwarningswarnDeprecationWarningr r* r r )argskwr r r resultr r r
runfunc s.
z
Trace.runfuncz ($self, func, /, *args, **kw)c
C s |j }|j}|rt|}nd }|j}d }|| jkrL| j| d k r| j| }nd | j|< dd t|D }t|dkrdd t|d D }t|dkrdd t|d D } t| dkr| d j}|| j|< |d k rd||f }|||fS )Nc S s g | ]}t |r|qS r )r Z
isfunction)r rE r r r
r s
z1Trace.file_module_function_of.<locals>.<listcomp>r c S s g | ]}t |tr|qS r )
isinstancer r
r r r
r s
r c S s g | ]}t |d r|qS ) __bases__)hasattr)r r r r r
r s
z%s.%s) f_code co_filenamer' co_namer gcZ
get_referrersr+ r )
r framer| r r rd ZclsnameZfuncsZdictsclassesr r r
file_module_function_of s,
z Trace.file_module_function_ofc C s0 |dkr,| |}| |j}d| j||f< 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 r )r r )r r r r r r r r
r s
z Trace.globaltrace_countfuncsc C sj |dkrf|j }|jdd }|rbt|}|d k rf| j||}|sf| jrZtd||jf | j S nd S d S )Nr __file__z! --- modulename: %s, funcname: %s)
r f_globalsrL r' r r r rC 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|< | jrTtdt | j dd tj
|}td||t ||f dd | j
S ) Nrz r r %.2f r
%s(%d): %sr( )r r f_linenor5 rL r rC r r r r# rY getliner )r r r r r rf rN bnamer r r
r 7 s
z Trace.localtrace_trace_and_countc C sd |dkr^|j j}|j}| jr4tdt | j dd tj|}td||t
||f dd | j S )Nrz r r r r r( ) r r r r rC r r r r# rY r r )r r r r r rf r r r r
r F s
zTrace.localtrace_tracec C s< |dkr6|j j}|j}||f}| j|dd | j|< | jS )Nrz r r )r r r r5 rL r )r r r r r rf rN r r r
r S s zTrace.localtrace_countc C s t | j| j| j| j| jdS )N)r9 r: r7 r8 )r r5 r9 r: r r )r r r r
results[ s
z
Trace.results) r r r r r r NNF)NN)r r r! r r r r __text_signature__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 |jrB|js8|jrB|
dH |jr\|js\|
dI |jd krr|
dJ t|j|j|j|j|j
|j|j |j |jdK }z|jrdd l }|j}| |\} }
} | j f|j t _!dL| j |
j"|
j#|
d dM} n^|jf|j t _!t$j%&|jt j%d< t'|jdN}
t(|
) |jdO} W 5 Q R X |jdLd d dP} |*| | | W nP t+k
r } z t ,dQt j!d |f W 5 d }~X Y n t-k
r Y nX |. }|j/s||j|j|j d S )RNr 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 <module>.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 <package>.<module> will be written to file <dir>/<package>/<module>.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.<locals>.parse_ignore_dirc S s$ g | ] }| d D ] }| qqS ),)splitstrip)r r r r r r
r s zmain.<locals>.<listcomp>c s& g | ] }| tjD ] } |qqS r )r r pathsep)r r r )r r r
r s z -r/--report requires -f/--file)r9 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 r9 r: r r )r r __package__
__loader____spec__
__cached__r2 r )r r r r z Cannot run file %r because: %s)0argparseArgumentParser add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER
parse_argsZ
ignore_dir sysconfigZget_pathZ
ignore_moduleZreportr4 errorr rq Zmissingrb rc anyr rh Z listfuncsZ
trackcallsr r r modulerunpyZ_get_module_detailsr r r* argvparentloaderr r rT r; r r r r@ exit
SystemExitr Z no_report)r parserZgrpZ_grpZoptsr tr Z module_nameZmod_nameZmod_specr| Zglobsrk rF r )r r r r
maina s
(r r )N)__all__rY r r* r r r\ r r r{ r< timer r r rw r r' r1 r r r r rX r r r r r r r
<module>2 s8 2 0
_