U
    @vgs                     @   s   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl	m
Z
 d dlmZ erd dlmZ d dlmZmZmZmZmZmZ d dlmZ d d	lmZ eeZejd
d Zejdd ZG dd de Z!dS )    )absolute_importN)contextlib2)TempDirectory)MYPY_CHECK_RUNNING)TracebackType)DictIteratorOptionalSetTypeUnion)InstallRequirement)Linkc               
   k   s   t j}t }i }|  D ]>\}}z|| ||< W n tk
rL   |||< Y nX |||< qz
d V  W 5 | D ].\}}||kr||= qlt|tst|||< qlX d S N)osenvironobjectitemsKeyError
isinstancestrAssertionError)changestargetnon_existent_markersaved_valuesname	new_valueoriginal_value r   A/tmp/pip-unpacked-wheel-gw11q0wt/pip/_internal/req/req_tracker.pyupdate_env_context_manager   s     

r!   c               
   c   st   t jd} t V}| d krL|tddj} |t| d t	
d|  t| }|V  W 5 Q R X W 5 Q R X d S )NPIP_REQ_TRACKERzreq-tracker)kind)r"   z Initialized build tracking at %s)r   r   getr   	ExitStackenter_contextr   pathr!   loggerdebugRequirementTracker)rootctxtrackerr   r   r    get_requirement_tracker5   s    

r.   c                   @   sR   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	e
jdd ZdS )r*   c                 C   s    || _ t | _td| j  d S )NzCreated build tracker: %s)_rootset_entriesr(   r)   )selfr+   r   r   r    __init__G   s    zRequirementTracker.__init__c                 C   s   t d| j | S )NzEntered build tracker: %s)r(   r)   r/   )r2   r   r   r    	__enter__M   s    zRequirementTracker.__enter__c                 C   s   |    d S r   )cleanup)r2   exc_typeexc_valexc_tbr   r   r    __exit__R   s    zRequirementTracker.__exit__c                 C   s$   t |j  }tj| j|S r   )	hashlibsha224url_without_fragmentencode	hexdigestr   r'   joinr/   )r2   linkhashedr   r   r    _entry_path[   s    zRequirementTracker._entry_pathc              
   C   s   |  |j}z t|}| }W 5 Q R X W n0 tk
r\ } z|jtjkrL W 5 d}~X Y nX d|j|f }t||| jkst	t|d}|
t| W 5 Q R X | j| td|| j dS )z5Add an InstallRequirement to build tracking.
        Nz%s is already being built: %swzAdded %s to build tracker %r)rB   r@   openreadIOErrorerrnoENOENTLookupErrorr1   r   writer   addr(   r)   r/   )r2   req
entry_pathfpcontentsemessager   r   r    rK   `   s    
zRequirementTracker.addc                 C   s2   t | |j | j| td|| j dS )z:Remove an InstallRequirement from build tracking.
        z Removed %s from build tracker %rN)	r   unlinkrB   r@   r1   remover(   r)   r/   r2   rL   r   r   r    rS      s    zRequirementTracker.removec                 C   s,   t | jD ]}| | q
td| j d S )NzRemoved build tracker: %r)r0   r1   rS   r(   r)   r/   rT   r   r   r    r5      s    zRequirementTracker.cleanupc                 c   s   |  | d V  | | d S r   )rK   rS   rT   r   r   r    track   s    
zRequirementTracker.trackN)__name__
__module____qualname__r3   r4   r9   rB   rK   rS   r5   
contextlibcontextmanagerrU   r   r   r   r    r*   E   s   	r*   )"
__future__r   rY   rG   r:   loggingr   pip._vendorr   pip._internal.utils.temp_dirr   pip._internal.utils.typingr   typesr   typingr   r   r	   r
   r   r   pip._internal.req.req_installr   pip._internal.models.linkr   	getLoggerrV   r(   rZ   r!   r.   r   r*   r   r   r   r    <module>   s&    


