B
F! @ 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 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/python37/lib64/python3.7/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 kr| j | S || jkr,d| j |< dS x*| jD ] }||d r4d| j |< dS q4W |d krnd| j |< dS x,| jD ]"}||tj rvd| j |< dS qvW d| j |< dS )Nr .r )r r
startswithr r sep)r filename
modulenamemodr r r r namesL s$
z
_Ignore.names)NN)__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}xJtjD ]@}t j|}||r|t| t jkrt|t|kr|}qW |rv| 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 r/ 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 | _|| _|| _| jry@t| jd}t |\}}}W d Q R X |
| ||| W n@ t t
tfk
r } z td| j|f tjd W d d }~X Y nX d S )NrbzSkipping counts file %r: %s)file)countscopyZcounter calledfuncscallersinfileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr( stderr)r r2 r4 r6 r5 r7 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 sz | j }| j}| j}|j }|j}|j}x$|D ] }||d|| ||< q*W x|D ] }d||< qPW x|D ] }d||< qfW d S )Nr r )r2 r4 r5 get) r otherr2 r4 r5 Z other_countsZother_calledfuncsZ
other_callerskeyr r r r; s
zCoverageResults.updateTFc !
C s | j rDt td | j }x(t|D ] \}}}td|||f q$W | jrt td d }} xvt| jD ]h\\}
} } \}
}}|
|krt td|
d |
}d} |
|
kr| |
krtd|
|
} td| | ||f qlW i }x8| jD ].\}}||i }||< | j||f ||< qW i }x$| D ]\}}| |r@q&|drX|d d }|d krt j
t j
|}t
|}n$|}t j
|st | t|}|rt|}ni }t|}t j
||d
}t|d }t|j\}}W d Q R X | |||||\} } |r&| r&td | | } | | ||f||< q&W |r|rtd
x2t|D ]&} || \} } }}td|| q`W | jry&t | j| j | jft| jdd W n6 t k
r } ztd| t j d W d 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.coverr0 d z lines cov% module (path)z%5d %3d%% %s (%s)wbr z"Can't save counts files because %s)r1 )!r4 r@ sortedr5 r2 rH itemsrG rF r r dirnameabspathr% existsmakedirsr/ _find_executable_linenos linecachegetlinesjoinr8 tokenizedetect_encodingreadlinewrite_results_fileintr7 r9 dumpr= r( rA )!r Z show_missingsummarycoverdirZcallsr r funcnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr. lnotabsourceZ coverpathfpencoding_n_hitsn_linespercentmrC r r r
write_results s|
z CoverageResults.write_resultsc
C s yt |d|d}W n8 tk
rJ } ztd||f tjd dS d }~X Y nX d}d} | xt|dD ]r\}
} |
|kr|d||
| d7 } |d7 }n.|
|krt| kr|d |d7 }n
|d
|| d qfW W d Q R X | |fS ) Nw)rh z3trace: Could not open %r for writing: %s - skipping)r1 )r r r r z%5d: z>>>>>> z ) r8 r= r@ r( rA enumeratewritePRAGMA_NOCOVER
expandtabs) r r linesre rc rh r7 rC rk rj rb liner r r r[ ( s(
z"CoverageResults.write_results_file)NNNNN)TFN)N)r r r r rG r; rn r[ r r r r r s
[c C s0 i }x&t | D ]\}}||krd||< qW |S )Nr )disZfindlinestarts)codestrslinenosri rb r r r _find_lines_from_codeH s
r{ c C s8 t | |}x(| jD ] }t|r|t|| qW |S )N)r{ co_constsinspectZiscoder; _find_lines)rx ry rz cr r r r~ R s
r~ c C s i }t j}t| |dr}t|j}x^|D ]V\}}}} }
|t jkr||t jkr||\} } | \}
}x t| |
d D ] }d||< qlW |}q*W W d Q R X |S )N)rh r )tokenINDENTr8 rX generate_tokensrZ STRINGrange)r rh r Z
prev_ttyperB tokZttypeZtstrstartendrv ZslineZscolZelineZecolir r r
_find_strings^ s
r c
C s y(t | }| }|j}W d Q R X W n8 tk
r` } ztd| |f tjd i S d }~X Y nX t|| d}t | |}t
||S )Nz%Not printing coverage data for %r: %s)r1 exec) rX r8 readrh r= r@ r( rA compiler r~ )r rB progrh rC rx ry r r r rT u s
rT c @ sp e Zd Zd 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 | _ | rTt | _ |rb| j
| _nZ|rp| j| _nL|r|r| j| _| j| _n2|r| j| _| j| _n |r| j| _| j| _nd| _d S )Nr r )r6 r7 r ignorer2 Zpathtobasename donothingtrace _calledfuncs_callers
_caller_cache
start_time_timeglobaltrace_trackcallers globaltraceglobaltrace_countfuncsglobaltrace_ltlocaltrace_trace_and_count
localtracelocaltrace_tracelocaltrace_count)
r rd r
countfuncs countcallers
ignoremods
ignoredirsr6 r7 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 d | j sbtd td X d S )N)r threadingsettracer r( r )r r globalslocalsr r r r s
z Trace.runctxc O s t | dkr| ^}}} n>| s&tdn0d|krB|d}| ^}} ntdt | d d }|jslt|j z|| |}W d |jstd X |S )N z8descriptor 'runfunc' of 'Trace' object needs an argumentfuncz7runfunc expected at least 1 positional argument, got %dr )r) TypeErrorpopr r( r r )argskwr r resultr r r runfunc s"
z
Trace.runfuncc
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
rB r r r r
s z1Trace.file_module_function_of..r c S s g | ]}t |tr|qS r )
isinstancer )r
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 framerx r r r` 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 )Nr 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_globalsrH r% r r r r@ r r )r r r r rx 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 ) Nrv r r z%.2f )r z
%s(%d): %sr& )r r f_linenor2 rH r r@ r r r r! rU getliner )r r r r r rb rJ bnamer r r r 3 s z Trace.localtrace_trace_and_countc C sd |dkr^|j j}|j}| jr4tdt | j dd tj|}td||t
||f dd | j S )Nrv z%.2fr )r z
%s(%d): %sr& ) r r r r r@ r r r r! rU r r )r r r r r rb r r r r r B s zTrace.localtrace_tracec C s< |dkr6|j j}|j}||f}| j|dd | j|< | jS )Nrv r r )r r r r2 rH r )r r r r r rb rJ r r r r O s zTrace.localtrace_countc C s t | j| j| j| j| jdS )N)r6 r7 r4 r5 )r r2 r6 r7 r r )r r r r resultsW s
z
Trace.results) r r r r r r NNF)NN)r r r r r r r r r r r r r r r r r r r r s
0
)
c st 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 |jd5d6d7d8 |jd9| jd:d8 | }|jrvt d;t d< fd=d>d?d@ |j
D |_
fdAd@|jD |_|j r|j s|
dB t|j |j dC}||j|j|jS t|j|j|j|jgs|
dD |jr0|js&|jr0|
dE |jrJ|jsJ|
dF |jd kr`|
dG |jf|jt_t j |jtj d< t |j|j|j|j|j
|j|j |j |j dH }yJt!|j}t"|# |jdI}W d Q R X |jdJd d dK} |$|| | W nP t%k
r6 }
z t&dLtjd |
f W d d }
~
X Y n t'k
rJ Y nX |( }|j)sp||j|j|j d S )MNr 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).r ?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 ] }| qqS ),)splitstrip)r
r r r r r r
s zmain..c s& g | ] }| tjD ] } |qqS r )r r pathsep)r
r r )r r r r
s z -r/--report requires -f/--file)r6 r7 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 --reportz3filename is missing: required with the main options)r r r r r6 r7 r r r )r r __package__
__cached__z Cannot run file %r because: %s)*argparseArgumentParser add_argumentadd_argument_group add_mutually_exclusive_group REMAINDER
parse_argsZ
ignore_dir sysconfigZget_pathZ
ignore_moduleZreportr1 errorr rn missingr^ r_ anyr rd Z listfuncsZ
trackcallsr r r( argvr r rP r r r8 r r r r= exit
SystemExitr Z no_report) r parserZgrpZ_grpZoptsr trg rx ZglobsrC r )r r r r main] s
(r r )N) __all__rU r rer( r r rX r} r rw r9 Ztimer r r rs r r% r/ r r{ r~ r rT r r r r r r r 2 s: 2 /
[