B
    `d                 @   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 }xJ|  D ]>\}}y|| ||< W n tk
rN   |||< Y nX |||< qW z
d V  W d x:| D ].\}}||kr||= qrt|tst|||< qrW X 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-hu9qqap1/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 d Q R X W d 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.   )r1   r   r   r   	__enter__M   s    zRequirementTracker.__enter__c             C   s   |    d S )N)cleanup)r1   exc_typeexc_valexc_tbr   r   r   __exit__R   s    zRequirementTracker.__exit__c             C   s$   t |j  }tj| j|S )N)	hashlibsha224url_without_fragmentencode	hexdigestr   r&   joinr.   )r1   linkhashedr   r   r   _entry_path[   s    zRequirementTracker._entry_pathc          
   C   s   |  |j}y t|}| }W dQ R X W n0 tk
r\ } z|jtjkrL W dd}~X Y nX d|j|}t||| j	kst
t|d}|t| W dQ R X | j	| td|| j dS )z5Add an InstallRequirement to build tracking.
        Nz{} is already being built: {}wzAdded %s to build tracker %r)rA   r?   openreadIOErrorerrnoENOENTformatLookupErrorr0   r   writer   addr'   r(   r.   )r1   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   unlinkrA   r?   r0   remover'   r(   r.   )r1   rL   r   r   r   rS      s    zRequirementTracker.removec             C   s0   xt | jD ]}| | qW td| j d S )NzRemoved build tracker: %r)r/   r0   rS   r'   r(   r.   )r1   rL   r   r   r   r4      s    zRequirementTracker.cleanupc             c   s   |  | d V  | | d S )N)rK   rS   )r1   rL   r   r   r   track   s    
zRequirementTracker.trackN)__name__
__module____qualname__r2   r3   r8   rA   rK   rS   r4   
contextlibcontextmanagerrT   r   r   r   r   r)   E   s   	 r)   )"
__future__r   rX   rF   r9   loggingr   pip._vendorr   pip._internal.utils.temp_dirr   pip._internal.utils.typingr   typesr   typingr   r   r	   r
   r   r   Zpip._internal.req.req_installr   pip._internal.models.linkr   	getLoggerrU   r'   rY   r    r-   r   r)   r   r   r   r   <module>   s"    
