U
��,a�t�@s�ddgZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl m Z
ddlZdZGdd�d�Zdd �Zd
d �ZGd d�d�Zd
d�Zdd�Zddd�Zdd�ZGdd�d�Zdd�Zedkr�e�dS)�Trace�CoverageResults�N)� monotonicz#pragma NO COVERc@s eZdZddd�Zdd�ZdS)�_IgnoreNcCs:|s
t�nt|�|_|s gn dd�|D�|_ddi|_dS)NcSsg|]}tj�|��qS�)�os�path�normpath��.0�drr�*/opt/alt/python38/lib64/python3.8/trace.py�
<listcomp>Gs�z$_Ignore.__init__.<locals>.<listcomp>z<string>�)�set�_mods�_dirs�_ignore)�self�modules�dirsrrr
�__init__Es
�z_Ignore.__init__cCs�||jkr|j|S||jkr,d|j|<dS|jD]"}|�|d�r2d|j|<dSq2|dkrld|j|<dS|jD]$}|�|tj�rrd|j|<dSqrd|j|<dS)Nr�.r)rr�
startswithrr�sep)r�filename�
modulename�modr rrr
�namesKs$
z
_Ignore.names)NN)�__name__�
__module__� __qualname__rr rrrr
rDs
rcCs tj�|�}tj�|�\}}|S�N)rr�basename�splitext)r�baser�extrrr
�_modnamevs r'cCs�tj�|�}d}tjD]@}tj�|�}|�|�r|t|�tjkrt|�t|�kr|}q|rr|t|�dd�}n|}tj�|�\}}|�tjd�}tj r�|�tj d�}tj�
|�\}}|� d�S)N�rr) rr�normcase�sysr�lenr�
splitdrive�replace�altsepr$�lstrip)rZ comparepathZlongest�dirr%Zdriverr&rrr
� _fullmodname}s
r1c@s:eZdZddd�Zdd�Zdd�Zdd
d �Zdd d
�ZdS)rNc
Cs�||_|jdkri|_|j��|_||_|jdkr8i|_|j��|_||_|jdkrZi|_|j��|_||_||_|jr�z@t|jd��}t� |�\}}}W5QRX|�
|� |||��Wn@t t
tfk
r�}z td|j|ftjd�W5d}~XYnXdS)N�rbzSkipping counts file %r: %s��file)�counts�copyZcounter� calledfuncs�callers�infile�outfile�open�pickle�load�update� __class__�OSError�EOFError�
ValueError�printr*�stderr)rr5r7r9r8r:�f�errrrr
r�s2
��zCoverageResults.__init__cCs|�d�o|�d�S)N�<�>)r�endswith)rrrrr
�is_ignored_filename�sz#CoverageResults.is_ignored_filenamec Csn|j}|j}|j}|j}|j}|j}|D] }|�|d�||||<q(|D] }d||<qJ|D] }d||<q\dS�Nrr)r5r7r8�get) r�otherr5r7r8Z other_countsZother_calledfuncsZ
other_callers�keyrrr
r>�s
zCoverageResults.updateTFc"
Cs�|jr@t�td�|j}t|�D] \}}}td|||f�q"|jr�t�td�d}} t|j�D]h\\}
} } \}
}}|
|kr�t�td|
d�|
}d} |
|
kr�| |
kr�td|
�|
} td| | ||f�qfi}|jD].\}}|�|i�}||<|j||f||<q�i}|��D�]\}}|�|��r0�q|�d��rH|dd �}|dk�rpt j
� t j
� |��}t
|�}n$|}t j
�|��s�t �|�t|�}|�r�t|�}ni}t�|�}t j
�||d
�}t|d ��}t�|j�\}}W5QRX|�|||||�\} } |�r| �rtd | | �} | | ||f||<�q|�rt|�rttd
�t|�D]&} || \} } }}td|| ��qL|j�r�z6t|jd�� } t � |j|j|jf| d�W5QRXWn6t k
�r�}!ztd|!t j d�W5d}!~!XYnXdS)Nzfunctions called:z*filename: %s, modulename: %s, funcname: %szcalling relationships:r(z***z -->z %s.%s -> %s.%sz.pyc���z.coverr2�dz lines cov% module (path)z%5d %3d%% %s (%s)�wbrz"Can't save counts files because %sr3)!r7rC�sortedr8r5rL�itemsrJrIrr�dirname�abspathr'�exists�makedirsr1�_find_executable_linenos� linecache�getlines�joinr;�tokenize�detect_encoding�readline�write_results_file�intr:r<�dumpr@r*rD)"rZ show_missing�summary�coverdirZcallsrr �funcnameZlastfileZ lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno� lines_hitZsums�countr0�lnotab�sourceZ coverpath�fp�encoding�_�n_hits�n_linesZpercent�mrErFrrr
�
write_results�s�
��
� �z CoverageResults.write_resultsc
Cs�zt|d|d�}Wn>tk
rP}z td||ftjd�WY�dSd}~XYnXd}d} |��t|d�D]r\}
} |
|kr�|�d||
�| d7} |d7}n.|
|kr�t| kr�|�d �|d7}n
|�d
�|�| �d ��qjW5QRX| |fS) N�w�rlz3trace: Could not open %r for writing: %s - skippingr3)rrrrz%5d: z>>>>>> z �) r;r@rCr*rD� enumerate�write�PRAGMA_NOCOVER�
expandtabs) rr�linesrirgrlr:rFrornrf�linerrr
r_(s.��
z"CoverageResults.write_results_file)NNNNN)TFN)N)r r r!rrJr>rqr_rrrr
r�s�
\cCs,i}t�|�D]\}}||krd||<q|S)Nr)�disZfindlinestarts)�code�strs�linenosrmrfrrr
�_find_lines_from_codeHs
rcCs4t||�}|jD] }t�|�r|�t||��q|Sr")r� co_consts�inspectZiscoder>� _find_lines)r|r}r~�crrr
r�Rs
r�c Cs�i}tj}t||d��j}t�|j�}|D]R\}}}} }
|tjkrv|tjkrv|\} } | \}
}t| |
d�D] }d||<qh|}q(W5QRX|S)Nrsr)�token�INDENTr;r\�generate_tokensr^�STRING�range)rrlr Z
prev_ttyperE�tokZttypeZtstr�start�endrzZslineZscolZelineZecol�irrr
�
_find_strings^s
r�c
Cs�z(t�|��}|��}|j}W5QRXWn@tk
rh}z"td||ftjd�iWY�Sd}~XYnXt||d�}t ||�}t
||�S)Nz%Not printing coverage data for %r: %sr3�exec) r\r;�readrlr@rCr*rD�compiler�r�)rrE�progrlrFr|r}rrr
rXus ��
rXc @sveZdZd dd�Zdd �Zd d
d �Zd d
�Zde_dd�Zdd�Z dd�Z
dd�Z dd�Z dd�Z
dd �Zd d �ZdS)!rrrrNFc
Cs�||_||_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|_dSrK)r9r:r�ignorer5Zpathtobasename� donothing�trace� _calledfuncs�_callers�
_caller_cache�
start_time�_time�globaltrace_trackcallers� globaltrace�globaltrace_countfuncs�globaltrace_lt�localtrace_trace_and_count�
localtrace�localtrace_trace�localtrace_count)
rrhr��
countfuncs� countcallers�
ignoremods�
ignoredirsr9r:�timingrrr
r�s6
zTrace.__init__cCs ddl}|j}|�|||�dS)Nr)�__main__�__dict__�runctx)r�cmdr��dictrrr
�run�sz Trace.runc Csh|dkr i}|dkri}|js6t�|j�t�|j�zt|||�W5|jsbt�d�t�d�XdSr")r�� threading�settracer�r*r�)rr��globals�localsrrr
r��s
z Trace.runctxc Os�t|�dkr|^}}}nV|s&td��nHd|krZ|�d�}|^}}ddl}|jdtdd�ntdt|�d��d}|js�t�|j �z|||�}W5|js�t�d�X|S) N�z8descriptor 'runfunc' of 'Trace' object needs an argument�funcrz0Passing 'func' as keyword argument is deprecated)�
stacklevelz7runfunc expected at least 1 positional argument, got %dr)
r+� TypeError�pop�warnings�warn�DeprecationWarningr�r*r�r�)�args�kwrr�r��resultrrr
�runfunc�s.
�
� z
Trace.runfuncz ($self, func, /, *args, **kw)c
Cs�|j}|j}|rt|�}nd}|j}d}||jkrL|j|dk r�|j|}n�d|j|<dd�t�|�D�}t|�dkr�dd�t�|d�D�}t|�dkr�dd�t�|d�D�} t| �dkr�| dj}||j|<|dk r�d||f}|||fS)NcSsg|]}t�|�r|�qSr)r�Z
isfunction)r rErrr
r�s
�z1Trace.file_module_function_of.<locals>.<listcomp>rcSsg|]}t|t�r|�qSr)�
isinstancer�r
rrr
r�s
�rcSsg|]}t|d�r|�qS)� __bases__)�hasattr)r r�rrr
r�s
�z%s.%s) �f_code� co_filenamer'�co_namer��gcZ
get_referrersr+r )
r�framer|rr rdZclsnameZfuncsZdicts�classesrrr
�file_module_function_of�s,
z Trace.file_module_function_ofcCs0|dkr,|�|�}|�|j�}d|j||f<dS�N�callr)r��f_backr�)rr��why�arg� this_funcZ parent_funcrrr
r� s
z Trace.globaltrace_trackcallerscCs |dkr |�|�}d|j|<dSr�)r�r�)rr�r�r�r�rrr
r�s
z Trace.globaltrace_countfuncscCsj|dkrf|j}|j�dd�}|rbt|�}|dk rf|j�||�}|sf|jrZtd||jf�|j SndSdS)Nr��__file__z! --- modulename: %s, funcname: %s)
r�� f_globalsrLr'r�r r�rCr�r�)rr�r�r�r|rr Z ignore_itrrr
r� s �zTrace.globaltrace_ltcCs�|dkr~|jj}|j}||f}|j�|d�d|j|<|jrTtdt�|jdd�tj �
|�}td||t � ||�fdd�|j
S) Nrzrr�%.2f� �r��
%s(%d): %sr()r�r��f_linenor5rLr�rCr�rrr#rY�getliner�)rr�r�r�rrfrN�bnamerrr
r�7s
��z Trace.localtrace_trace_and_countcCsd|dkr^|jj}|j}|jr4tdt�|jdd�tj�|�}td||t �
||�fdd�|j S)Nrzr�r�r�r�r() r�r�r�r�rCr�rrr#rYr�r�)rr�r�r�rrfr�rrr
r�Fs
��zTrace.localtrace_tracecCs<|dkr6|jj}|j}||f}|j�|d�d|j|<|jS)Nrzrr)r�r�r�r5rLr�)rr�r�r�rrfrNrrr
r�Ss zTrace.localtrace_countcCs t|j|j|j|j|jd�S)N)r9r:r7r8)rr5r9r:r�r�)rrrr
�results[s
�z
Trace.results) rrrrrrNNF)NN)r r r!rr�r�r��__text_signature__r�r�r�r�r�r�r�r�rrrr
r�s&�
2
)
c s�ddl}|��}|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/d0gd1d2�|jd3d0gd4d2�|jd5d
d6d7d2�|jd8d9d:d;�|jd<|jd=d;�|��}|j�r�t� d>��t� d?����fd@dA��dBdC�|j
D�|_
�fdDdC�|jD�|_|j �r�|j �s�|�
dE�t|j |j dF�}|�|j|j|j�St|j|j|j|jg��s |�
dG�|j�rB|j�s8|j�rB|�
dH�|j�r\|j�s\|�
dI�|jdk�rr|�
dJ�t|j|j|j|j|j
|j|j |j |jdK� }z�|j�r�ddl }|j}|� |�\} }
} | j f|j �t _!dL| j |
j"|
j#|
ddM�} n^|jf|j �t _!t$j%�&|j�t j%d<t'|jdN��}
t(|
�)�|jdO�} W5QRX|jdLdddP�} |�*| | | �WnPt+k
�r�}z t �,dQt j!d|f�W5d}~XYnt-k
�r�YnX|�.�}|j/�s�|�|j|j|j�dS)RNrz --version�versionz trace 2.0)�actionr�z Main optionsz(One of these (or --report) must be givenz-cz--count�
store_truez�Count 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 --listfuncsz�Keep 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--reportz�Generate 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
--coverdirz�Directory 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-module�appendzqIgnore 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
platstdlibcs4tj�tj�|��}|�d���d��}tj�|�S)Nz$prefixz $exec_prefix)rr�
expanduser�
expandvarsr-r )�s)� _exec_prefix�_prefixrr
�parse_ignore_dir�sz main.<locals>.parse_ignore_dircSs$g|] }|�d�D] }|���qqS)�,)�split�strip)r r�r rrr
r�s �zmain.<locals>.<listcomp>cs&g|] }|�tj�D] }�|��qqSr)r�r�pathsep)r r�r�)r�rr
r�s�z -r/--report requires -f/--file)r9r: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�r9r:r�r�)r r�� __package__�
__loader__�__spec__�
__cached__r2r�)r�r r�r�z Cannot run file %r because: %s)0�argparse�ArgumentParser� add_argumentZadd_argument_groupZ add_mutually_exclusive_groupZ REMAINDER�
parse_argsZ
ignore_dir� sysconfigZget_pathZ
ignore_moduleZreportr4�errorrrqZmissingrbrc�anyr�rhZ listfuncsZ
trackcallsr�rr��module�runpyZ_get_module_detailsr�r�r*�argv�parent�loaderrrrTr;r�r�r�r@�exit�
SystemExitr�Z no_report)r��parserZgrpZ_grpZoptsr��tr�Z module_nameZmod_nameZmod_specr|ZglobsrkrFr)r�r�r�r
�mainas��
�
�
�
�
�
���
�
�
��
�
�
��
�
�
�
�� �(rr�)N)�__all__rYrr*r�r�r\r�r�r{r<�timerr�r�rwrr'r1rrr�r�rXrrr rrrr
�<module>2s8 2 0
_ |