Your IP : 216.73.216.213
B
E�W�9 � @ s� d Z ddlZddlZddlmZ ddlmZ ddlmZ ddl m
Z
m Z ddl m
Z
e e�ZyddlmZmZ W n< ek
r� e�d �d
kr�ej�d � e�d � dZY nX G d
d� de�Zdd� ZG dd� de�ZdS )z#Raw data collector for coverage.py.� N)�env)�iitems)�abs_file)�CoverageException�isolate_module)�PyTracer)�CTracer�CFileDispositionZCOVERAGE_TEST_TRACER�cz:*** COVERAGE_TEST_TRACER is 'c' but can't import CTracer!
� c @ s e Zd ZdZdS )�FileDispositionz9A simple value type for recording what to do with a file.N)�__name__�
__module__� __qualname__�__doc__� r r �E/opt/alt/python37/lib64/python3.7/site-packages/coverage/collector.pyr # s r c C s | j j}|�d�r|S dS )zKWho-Tests-What hack: Determine whether this frame begins a new who-context.�testN)�f_code�co_name�
startswith)�frameZfn_namer r r �should_start_context( s
r c @ s� e Zd ZdZg Zeddddg�Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zd d � Zd S ) � Collectora Collects trace data.
Creates a Tracer object for each thread, since they track stack
information. Each Tracer points to the same shared data, contributing
traced data points.
When the Collector is started, it creates a Tracer for the current thread,
and installs a function to create Tracers for each new thread started.
When the Collector is stopped, all active Tracers are stopped.
Threads started while the Collector is stopped will never have Tracers
associated with them.
�greenlet�eventlet�gevent�threadc C sf || _ || _|| _|| _d| _d| _| j�|�}t|�dkrHt d| ��|rT|�
� nd| _ y�| j dkrxddl }|j
| _nf| j dkr�ddl} | jj
| _nH| j dkr�ddl}
|
j
| _n,| j d ks�| j s�ddl} | | _n t d
| ��W n& tk
�r t d | j f ��Y nX tt�d d��| _| �� |�r0t| _n t�p8t| _| jtk�rVt| _d
| _ n t | _d| _ dS )a� Create a collector.
`should_trace` is a function, taking a file name, and returning a
`coverage.FileDisposition object`.
`check_include` is a function taking a file name and a frame. It returns
a boolean: True if the file should be traced, False if not.
If `timid` is true, then a slower simpler trace function will be
used. This is important for some environments where manipulation of
tracing functions make the faster more sophisticated trace function not
operate properly.
If `branch` is true, then branches will be measured. This involves
collecting data on which statements followed each other (arcs). Use
`get_arc_data` to get the arc data.
`warn` is a warning function, taking a single string message argument,
to be used if a warning needs to be issued.
`concurrency` is a list of strings indicating the concurrency libraries
in use. Valid values are "greenlet", "eventlet", "gevent", or "thread"
(the default). Of these four values, only one can be supplied. Other
values are ignored.
Nr z$Conflicting concurrency settings: %s� r r r r r z Don't understand concurrency=%sz?Couldn't trace with concurrency=%s, the module isn't installed.Z COVERAGE_WTWTF) � should_trace�
check_include�warn�branch� threading�concur_id_func�SUPPORTED_CONCURRENCIES� intersection�lenr �pop� concurrencyr Z
getcurrentZeventlet.greenthreadZ greenthreadr � ImportError�int�os�getenv�wtw�resetr � _trace_classr r Zfile_disposition_classZsupports_pluginsr ) �selfr r Ztimidr" r! r) Zthese_concurrenciesr r r r# r r r �__init__G sL
zCollector.__init__c C s dt | �| �� f S )Nz<Collector at 0x%x: %s>)�id� tracer_name)r1 r r r �__repr__� s zCollector.__repr__c C s | j jS )z0Return the class name of the tracer we're using.)r0 r
)r1 r r r r4 � s zCollector.tracer_namec C sJ i | _ i | _| j | jd< i | _tjr:ddl}|�d�| _ni | _g | _dS )z2Clear collected data, and prepare to collect more.Nr �module) �data�contexts� file_tracersr ZPYPY�__pypy__Znewdict�should_trace_cache�tracers)r1 r: r r r r/ � s
zCollector.resetc C s� | � � }| j|_| j|_| j|_| j|_| j|_t|d�rD| j|_n | jr`t d| j
| � � f ��t|d�rr| j |_ t|d�r�| j
|_
t|d�r�| j|_| jr�t|d�r�t|_t|d�r�| j|_|�� }| j�|� |S )z8Start a new Tracer object, and store it in self.tracers.r$ z@Can't support concurrency=%s with %s, only threads are supportedr9 r# r r �switch_context)r0 r7 r" Z
trace_arcsr r; r! �hasattrr$ r r) r4 r9 r# r r. r r= �startr< �append)r1 �tracer�fnr r r �
_start_tracer� s4
zCollector._start_tracerc C s& t �d� | �� }|r"||||�}|S )z0Called on new threads, installs the real tracer.N)�sys�settracerC )r1 r �event�argrB r r r �_installation_trace� s
z Collector._installation_tracec
C s� | j r| j d �� g }t�� }|r@t|dd�}|r@t|dg �}y | �� }W n" | j rh| j d �� � Y nX | j �| � xL|D ]D}|\\}}}} y||||| d� W q� tk
r� t d��Y q�X q�W | j
r�| j
� | j � dS )z#Start collecting trace information.����__self__NZtraces)�linenoz3fullcoverage must be run with the C trace function.)
� _collectors�pauserD �gettrace�getattrrC �resumer@ � TypeError� Exceptionr# rE rH )
r1 Ztraces0Zfn0Ztracer0rB �argsr rF rG rK r r r r? s.
zCollector.startc C s^ | j s
t�| j d | ks.td| | j d f ��| �� g | _| j �� | j rZ| j d �� dS )z"Stop collecting trace information.rI z0Expected current collector to be %r, but it's %rN)rL �AssertionErrorrM r< r( rP )r1 r r r �stop0 s
zCollector.stopc C sl xT| j D ]J}|�� |�� }|rtd� x(t|�� �D ]}td||| f � q6W qW | jrh| j�d� dS )z+Pause tracing, but be prepared to `resume`.z
Coverage.py tracer stats:z%20s: %sN)r< rU Z get_stats�print�sorted�keysr# rE )r1 rA Zstats�kr r r rM @ s zCollector.pausec C s: x| j D ] }|�� qW | jr.| j�| j� n| �� dS )z Resume tracing after a `pause`.N)r< r? r# rE rH rC )r1 rA r r r rP L s
zCollector.resumec C s( | j �|i �}x| jD ]
}||_qW dS )z1Who-Tests-What hack: switch to a new who-context.N)r8 �
setdefaultr< r7 )r1 Z new_contextr7 rA r r r r= U s zCollector.switch_contextc C s� dd� }| j r |�|| j�� n|�|| j�� |�|| j�� | jr�ddl}d�t �
� �}t |d��}|�| j |� W dQ R X | �
� dS )zZSave the collected data to a `CoverageData`.
Also resets the collector.
c S s t dd� t| �D ��S )z;Return a dict like d, but with keys modified by `abs_file`.c s s | ]\}}t |�|fV qd S )N)r )�.0rY �vr r r � <genexpr>e s z=Collector.save_data.<locals>.abs_file_dict.<locals>.<genexpr>)�dictr )�dr r r �
abs_file_dictc s z*Collector.save_data.<locals>.abs_file_dictr Nzcoverage_wtw_{:06}.py�w)r" Zadd_arcsr7 Z add_linesZadd_file_tracersr9 r. �pprint�formatr, �getpid�openr8 r/ )r1 Zcovdatar` rb �out_fileZwtw_outr r r � save_data] s zCollector.save_dataN)r
r r r rL �setr% r2 r5 r4 r/ rC rH r? rU rM rP r= rg r r r r r / s W.)
' r )r r, rD �coverager Zcoverage.backwardr Zcoverage.filesr Z
coverage.miscr r Zcoverage.pytracerr Zcoverage.tracerr r r* r- �stderr�write�exit�objectr r r r r r r �<module> s$