3
\=p @ s( d dgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z y ddlZW n" ek
r ejZdd ZY nX dd Zd d Zd
ZG d d d Zd
d Zdd ZG dd dZdd Zdd Zd ddZdd Z G dd d Z dd Z e d kr$e dS ) TraceCoverageResults N) monotonicc C s t jd d S )N)syssettrace r r /usr/lib64/python3.6/trace.py _unsettraceE s r c C s t j| tj| d S )N) threadingr r )funcr r r _settraceH s
r c C s t jd tjd d S )N)r r r
r r r r r L s
z#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 jj|qS r )ospathnormpath).0dr r r
U s z$_Ignore.__init__..z )set_mods_dirs_ignore)selfmodulesdirsr r r __init__S s
z_Ignore.__init__c C s || j kr| j | S || jkr,d| j |< dS x*| jD ] }|j|d r4d| j |< dS q4W |d krnd| j |< dS x,| jD ]"}|j|tj rvd| j |< dS qvW d| j |< dS )Nr .r )r r
startswithr r sep)r filename
modulenamemodr r r r namesY s$
z
_Ignore.names)NN)__name__
__module__ __qualname__r r# r r r r r
R s
r
c C s t jj| }t jj|\}}|S )N)r r basenamesplitext)r baser extr r r _modname s r+ c C s t jj| }d}xJtjD ]@}t jj|}|j|r|t| t jkrt|t|kr|}qW |rv| t|d d }n| }t jj|\}}|jt jd}t j r|jt j d}t jj
|\}}|j dS )N r r ) r r normcaser r lenr
splitdrivereplacealtsepr( lstrip)r Z comparepathZlongestdirr) Zdriver r* r r r _fullmodname s r4 c @ s: e Zd ZdddZdd Zdd Zdd
d Zdd d
ZdS )r Nc C s || _ | j d kri | _ | j j | _|| _| jd kr8i | _| jj | _|| _| jd krZi | _| jj | _|| _|| _| jry@t| jd}tj |\}}}W d Q R X | j
| j ||| W n@ t t
tfk
r } z td| j|f tjd W Y d d }~X nX d S )NrbzSkipping counts file %r: %s)file)countscopyZcounter calledfuncscallersinfileoutfileopenpickleloadupdate __class__OSErrorEOFError
ValueErrorprintr stderr)r r7 r9 r; r: r< ferrr r r r s,
zCoverageResults.__init__c C s |j do|jdS )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 ] }|j|d|| ||< q*W x|D ] }d||< qPW x|D ] }d||< qfW d S )Nr r )r7 r9 r: get) r otherr7 r9 r: 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 ].\}}|j|i }||< | j||f ||< qW i }x$|j D ]\}}| j|r@q&|jdrX|d d }|d krt j
j t j
j |}t
|}n$|}t j
j|st j| t|}|rt|}ni }tj|}t j
j||d
}t|d }tj|j\}}W d Q R X | j|||||\} } |r&| r&td | | } | | ||f||< q&W |r|rtd
x2t|D ]&} || \} } }}td|| q`W | jry&t j | j| j | jft| jdd W n6 t k
r } ztd| t j d W Y d d } ~ X nX d S )Nzfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r, z***z -->z %s.%s -> %s.%sz.pycr z.coverr5 d z lines cov% module (path)z%5d %3d%% %s (%s)wbz"Can't save counts files because %s)r6 )!r9 rE sortedr: r7 rM itemsrL rK r r dirnameabspathr+ existsmakedirsr4 _find_executable_linenos linecachegetlinesjoinr= tokenizedetect_encodingreadlinewrite_results_fileintr< r> dumprB r rF )!r Z show_missingsummarycoverdirZcallsr r! funcnameZlastfileZ lastcfileZpfileZpmodZpfunccfileZcmodZcfuncZper_filelineno lines_hitZsumscountr3 lnotabsourceZ coverpathfpencoding_n_hitsn_linesZpercentmrH r r r
write_results s|
z CoverageResults.write_resultsc C s yt |d|d}W n6 tk
rH } ztd||f tjd d S d }~X nX d}d} | xt|dD ]r\}
} |
|kr|jd||
| d7 } |d7 }n.|
|krt| kr|jd |d7 }n
|jd |j| jd
qdW W d Q R X | |fS ) Nw)rm z3trace: Could not open %r for writing: %s - skipping)r6 r r z%5d: z>>>>>> z )r r ) r= rB rE r rF enumeratewritePRAGMA_NOCOVER
expandtabs) r r linesrj rh rm r< rH rp ro rg liner r r r` 5 s(
z"CoverageResults.write_results_file)NNNNN)TFN)N)r$ r% r& r rL r@ rr r` r r r r r s
[c C s0 i }x&t j| D ]\}}||krd||< qW |S )Nr )disZfindlinestarts)codestrslinenosrn rg r r r _find_lines_from_codeU s
r c C s8 t | |}x(| jD ] }tj|r|jt|| qW |S )N)r co_constsinspectZiscoder@ _find_lines)r| r} r~ cr r r r _ s
r c
C s i }t j}t| |dr}tj|j}x^|D ]V\}}}} }
|t jkr||t jkr||\} } | \}
}x t| |
d D ] }d||< qlW |}q*W W d Q R X |S )N)rm r )tokenINDENTr= r] generate_tokensr_ STRINGrange)r rm r Z
prev_ttyperG tokZttypeZtstrstartendrz ZslineZscolZelineZecolir r r
_find_stringsk s
r c C s y(t j| }|j }|j}W d Q R X W n6 tk
r^ } ztd| |f tjd i S d }~X nX t|| d}t | |}t
||S )Nz%Not printing coverage data for %r: %s)r6 exec) r] r= readrm rB rE r rF compiler r )r rG progrm rH r| r} r r r rY s
rY c @ s e Zd Zddddf f dddf ddZdd Zd d d
Zd d Zd
d Zdd Zdd Z dd Z
dd Z dd Z dd Z
dd ZdS ) 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 )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 s6
zTrace.__init__c C s dd l }|j}| j||| d S )Nr )__main____dict__runctx)r cmdr dictr r r run s z Trace.runc
C sL |d kr i }|d kri }| j s(t| j zt||| W d | j sFt X d S )N)r r r r r )r r globalslocalsr r r r s
z Trace.runctxc O s< d }| j stj| j z|||}W d | j s6tjd X |S )N)r r r r )r r argskwresultr r r runfunc s z
Trace.runfuncc
C s |j }|j}|rt|}nd }|j}d }|| jkrL| j| d k r| j| }nd | j|< dd tj|D }t|dkrdd tj|d D }t|dkrdd tj|d D } t| dkr| d j}|| j|< |d k rd||f }|||fS )Nc S s g | ]}t j|r|qS r )r Z
isfunction)r rG 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 framer| r r! re ZclsnameZfuncsZdictsclassesr r r file_module_function_of s,
z Trace.file_module_function_ofc C s0 |dkr,| j |}| j |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 | j |}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 }|jjdd }|rbt|}|d k rf| jj||}|sf| jrZtd||jf | j S nd S d S )Nr __file__z! --- modulename: %s, funcname: %s)
r f_globalsrM r+ r r# r rE 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}| jj|dd | j|< | jrTtdt | j dd tj j
|}td||t j ||f dd | j
S ) Nrz r r z%.2f )r z
%s(%d): %sr, )r r f_linenor7 rM r rE r r r r' rZ getliner )r r r r r rg rO bnamer r r r 2 s z Trace.localtrace_trace_and_countc C sd |dkr^|j j}|j}| jr4tdt | j dd tjj|}td||t j
||f dd | j S )Nrz z%.2fr )r z
%s(%d): %sr, ) r r r r rE r r r r' rZ r r )r r r r r rg r r r r r A s zTrace.localtrace_tracec C s< |dkr6|j j}|j}||f}| jj|dd | j|< | jS )Nrz r r )r r r r7 rM r )r r r r r rg rO r r r r N s zTrace.localtrace_countc C s t | j| j| j| j| jdS )N)r; r< r9 r: )r r7 r; r< r r )r r r r resultsV s
z
Trace.results)NN)r$ r% r& r r r r r r r r r r r r r r r r r s 0
)
c s t j } | jdddd | jdd}|jddd d
d |jd d
d dd |jddd dd |jddd dd | jd}|j }|jddd dd |jddd dd |jd d d d |jd d!d"d |jd#d$d d%d |jd&d'd d(d |jd)d*d d+d | jd,d-}|jd.d/g d0d1 |jd2d/g d3d1 | jd4d5d6d7 | jd8t jd9d7 | j }|jrd:d;jt j
f}t j j
t jf| t j j
t jf| fdd? |jD |_fd@d?|jD |_|jr|js| jdA t|j|jdB}|j|j|j|jS t|j|j|j |j gs*| jdC |j rL|jsB|jrL| jdD |jrh|j rh| jdE |j d kr~| jdF |j f|j t _ t j j!|j t j dG< t"|j|j|j |j |j|j|j|j|j#dH }yJt$|j }t%|j& |j dI}W d Q R X |j dJd d dK} |j'|| | W nP t(k
rT }
z t j)dLt j dG |
f W Y d d }
~
X n t*k
rh Y nX |j+ }|j,s|j|j|j|j d S )MNz --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 programlibzpython{0.major}.{0.minor}c s4 t jjt jj| } | jdjd } t jj| S )Nz$prefixz $exec_prefix)r r
expanduser
expandvarsr0 r )s) _exec_prefix_prefixr r parse_ignore_dir s z main..parse_ignore_dirc S s$ g | ] }|j d D ] }|j qqS ),)splitstrip)r r r" r r r r s zmain..c s& g | ] }|j tjD ] } |qqS 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 --reportz3filename is missing: required with the main optionsr )r r r r r; r< r r r )r r$ __package__
__cached__z Cannot run file %r because: %s)-argparseArgumentParser add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER
parse_argsZ
ignore_dirformatr version_infor r r\ base_prefixbase_exec_prefixZ
ignore_moduleZreportr6 errorr rr Zmissingrc rd anyr ri Z listfuncsZ
trackcallsr r argvrU r r r= r r r rB exit
SystemExitr Z no_report) parserZgrpZ_grpZoptsZrel_pathr trl r| ZglobsrH r )r r r r main\ s
(r r )N) __all__r rZ r rer r r] r r r{ r> Ztimer r r
ImportErrorr r r rw r
r+ r4 r r r r rY r r r$ r r r r 2 sF 2 /
M