U
e5d�t � @ s� d dgZ ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z ddlZdZG dd� d�Zdd � Zd
d � ZG d d� d�Zd
d� Zdd� Zddd�Zdd� ZG dd � d �Zdd� Zedkr�e� dS )�Trace�CoverageResults� N)� monotonicz#pragma NO COVERc @ s e Zd Zddd�Zdd� ZdS )�_IgnoreNc C s: |s
t � nt |�| _|s g n dd� |D �| _ddi| _d S )Nc S s g | ]}t j�|��qS � )�os�path�normpath��.0�dr r � /usr/lib64/python3.8/trace.py�
H s �z$_Ignore.__init__..z� )�set�_mods�_dirs�_ignore)�self�modules�dirsr r r
�__init__F 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�
modulename�modr 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 �basename�splitext)r �baser �extr r r
�_modnamew 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�
|�\}}|� d�S )N� r r ) r r �normcase�sysr �lenr �
splitdrive�replace�altsepr$ �lstrip)r Z comparepathZlongest�dirr% Zdriver r& r r r
� _fullmodname~ s
r1 c @ s: e Zd Zddd�Zdd� Zdd� Zdd
d �Zdd d
�ZdS )r Nc
C s� || _ | j d kri | _ | j �� | _|| _| jd kr8i | _| j�� | _|| _| jd krZi | _| j�� | _|| _|| _| jr�z@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 )N�rbzSkipping counts file %r: %s��file)�counts�copyZcounter� calledfuncs�callers�infile�outfile�open�pickle�load�update� __class__�OSError�EOFError�
ValueError�printr* �stderr)r r5 r7 r9 r8 r: �f�errr r r
r � s2
��zCoverageResults.__init__c C s |� d�o|�d�S )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_callers�keyr r r
r> � s
zCoverageResults.updateTFc "
C s� | j r@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|d d � }|d k�rpt j
� t j
� |��}t
|�}n$|}t j
�|��s�t �|� t|�}|�r�t|�}ni }t�|�}t j
�||d
�}t|d ��}t�|j�\}}W 5 Q R X | �|||||�\} } |�r| �rtd | | �} | | ||f||< �q|�rt|�rttd
� t|�D ]&} || \} } }}td|| � �qL| j�r�z6t| 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.pyc�����z.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 �dirname�abspathr' �exists�makedirsr1 �_find_executable_linenos� linecache�getlines�joinr; �tokenize�detect_encoding�readline�write_results_file�intr: r< �dumpr@ r* rD )"r Z show_missing�summary�coverdirZcallsr r �funcnameZlastfileZ lastcfileZpfileZpmodZpfunc�cfileZcmodZcfuncZper_file�lineno� lines_hitZsums�countr0 �lnotab�sourceZ coverpath�fp�encoding�_�n_hits�n_linesZpercent�mrE 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|d�D ]r\}
} |
|kr�|�d||
� | d7 } |d7 }n.|
|kr�t| kr�|�d � |d7 }n
|�d
� |�| �d �� qjW 5 Q R X | |fS ) N�w�rl z3trace: Could not open %r for writing: %s - skippingr3 )r r r r z%5d: z>>>>>> z � ) r; r@ rC r* rD � enumerate�write�PRAGMA_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)�code�strs�linenosrm rf r r r
�_find_lines_from_codeI s
r c C s4 t | |�}| jD ] }t�|�r|�t||�� q|S r" )r � co_consts�inspectZiscoder> � _find_lines)r| r} r~ �cr r r
r� S s
r� c C s� i }t j}t| |d��j}t�|j�}|D ]R\}}}} }
|t jkrv|t jkrv|\} } | \}
}t| |
d �D ] }d||< qh|}q(W 5 Q R X |S )Nrs r )�token�INDENTr; r\ �generate_tokensr^ �STRING�range)r rl r Z
prev_ttyperE �tokZttypeZtstr�start�endrz ZslineZscolZelineZecol�ir 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 v s ��
rX c @ sv e Zd Zd dd�Zdd � 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� 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)
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 sbt�d � t�d � X d S r" )r� � threading�settracer� r* r� )r r� �globals�localsr r r
r� � s
z Trace.runctxc O s� t | �dkr| ^}}} nV| s&td��nHd|krZ|�d�}| ^}} dd l}|jdtdd� ntdt | �d ��d }|js�t�|j � z|| |�}W 5 |js�t�d � X |S ) N� z8descriptor 'runfunc' of 'Trace' object needs an argument�funcr z0Passing '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�kwr 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| }n�d | j|< dd� t�|�D �}t|�dkr�dd� t�|d �D �}t|�dkr�dd� t�|d �D �} t| �dkr�| d j}|| j|< |d k r�d||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..r c S s g | ]}t |t�r|�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 ZclsnameZfuncsZdicts�classesr r r
�file_module_function_of� s,
z Trace.file_module_function_ofc C s0 |dkr,| � |�}| � |j�}d| j||f< d S �N�callr )r� �f_backr� )r r� �why�arg� 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 }|j�dd �}|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�|d�d | 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� 8 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� G s
��zTrace.localtrace_tracec C s< |dkr6|j j}|j}||f}| j�|d�d | j|< | jS )Nrz r r )r� r� r� r5 rL r� )r r� r� r� r rf rN r r r
r� T s zTrace.localtrace_countc C s t | j| j| j| j| jd�S )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;� |�� }|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�S t|j|j|j|jg��s |�
dG� |j�rB|j�s8|j�rB|�
dH� |j�r\|j�s\|�
dI� |jd k�rr|�
dJ� t|j|j|j|j|j
|j|j |j |jdK� }z�|j�r�dd l }|j}|� |�\} }
} | j f|j �t _!dL| j |
j"|
j#|
d dM�} n^|jf|j �t _!t$j%�&|j�t j%d< t'�(|j��}
t)|
�*� |jdN�} W 5 Q R X |jdLd d dO�} |�+| | | � W nP t,k
�r� } z t �-dPt j!d |f � W 5 d }~X Y n t.k
�r� Y nX |�/� }|j0�s�|�|j|j|j� d S )QNr z --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 .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 . 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-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
platstdlibc s4 t j�t j�| ��} | �d���d� �} 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 )�,)�split�strip)r r� r r r r
r � s �zmain..c s& g | ] }|� tj�D ] }� |��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__r� )r� r r� r� z Cannot run file %r because: %s)1�argparse�ArgumentParser� 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� �module�runpyZ_get_module_detailsr� r� r* �argv�parent�loaderr r rT �io� open_coder� r� r� r@ �exit�
SystemExitr� Z no_report)r� �parserZgrpZ_grpZoptsr� �tr� Z module_nameZmod_nameZmod_specr| Zglobsrk rF r )r� r� r� r
�mainb s� �
�
�
�
�
�
���
�
�
��
�
�
��
�
�
�
�� �(r r� )N) �__all__r� 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
�2 s: 2 0
_