관리-도구
편집 파일: build_tracker.cpython-38.pyc
U &?�f� � @ s� d dl Z d dlZd dlZd dlZd dlmZ d dlmZmZm Z m Z m Z d dl m Z d dlmZ e�e�Ze jeed d�dd ��Ze jed d �d d ��ZG dd� de�ZG dd� d�ZdS )� N)� TracebackType)�Dict� Generator�Optional�Type�Union)�InstallRequirement)� TempDirectory�NNN)�changes�returnc k s� t j}t� }i }| �� D ]>\}}z|| ||< W n tk rL |||< Y nX |||< qz d V W 5 |�� D ].\}}||kr�||= qlt|t�s�t�|||< qlX d S �N)�os�environ�object�items�KeyError� isinstance�str�AssertionError)r �target�non_existent_marker� saved_values�name� new_value�original_value� r �Z/opt/hc_python/lib/python3.8/site-packages/pip/_internal/operations/build/build_tracker.py�update_env_context_manager s r )� BuildTrackerNN�r c c st t j�d�} t�� �V}| d krL|�tdd��j} |�t| d�� t � d| � t | �� }|V W 5 Q R X W 5 Q R X d S )N�PIP_BUILD_TRACKERz build-tracker)�kind)r! z Initialized build tracking at %s) r r �get� contextlib� ExitStack� enter_contextr �pathr �logger�debugr )�root�ctx�trackerr r r �get_build_tracker( s r- c @ s e Zd ZdZdS )� TrackerIdz:Uniquely identifying string provided to the build tracker.N)�__name__� __module__� __qualname__�__doc__r r r r r. 5 s r. c @ s� e Zd ZdZedd�dd�Zd d�dd�Zeee ee ee dd �d d �Z e ed �d d�Z ee dd�dd�Zee dd�dd�Zdd�dd�Zejeeed d�dd��ZdS )r a7 Ensure that an sdist cannot request itself as a setup requirement. When an sdist is prepared, it identifies its setup requirements in the context of ``BuildTracker.track()``. If a requirement shows up recursively, this raises an exception. This stops fork bombs embedded in malicious packages.N)r* r c C s || _ i | _t�d| j � d S )NzCreated build tracker: %s)�_root�_entriesr( r) )�selfr* r r r �__init__B s zBuildTracker.__init__r c C s t �d| j� | S )NzEntered build tracker: %s)r( r) r3 )r5 r r r � __enter__G s zBuildTracker.__enter__)�exc_type�exc_val�exc_tbr c C s | � � d S r )�cleanup)r5 r8 r9 r: r r r �__exit__K s zBuildTracker.__exit__)�keyr c C s" t �|�� ��� }tj�| j|�S r )�hashlib�sha224�encode� hexdigestr r' �joinr3 )r5 r= �hashedr r r � _entry_pathS s zBuildTracker._entry_path)�reqr= r c C s� | � |�}z t|��}|�� }W 5 Q R X W n tk r> Y nX |j� d|� �}t|��|| jksft�t|ddd��}|�t |�� W 5 Q R X || j|< t � d|| j � dS )z,Add an InstallRequirement to build tracking.z is already being built: �wzutf-8)�encodingz Added %s to build tracker %rN) rD �open�read�FileNotFoundError�link� LookupErrorr4 r �writer r( r) r3 )r5 rE r= � entry_path�fp�contents�messager r r �addW s zBuildTracker.addc C s, t �| �|�� | j|= t�d|| j� dS )z1Remove an InstallRequirement from build tracking.z Removed %s from build tracker %rN)r �unlinkrD r4 r( r) r3 )r5 rE r= r r r �remover s zBuildTracker.removec C s6 t | j�� �D ]\}}| �||� qt�d| j� d S )NzRemoved build tracker: %r)�listr4 r rT r( r) r3 )r5 r= rE r r r r; { s zBuildTracker.cleanupr c c s* t |�}| �||� dV | �||� dS )z�Ensure that `key` cannot install itself as a setup requirement. :raises LookupError: If `key` was already provided in a parent invocation of the context introduced by this method.N)r. rR rT )r5 rE r= � tracker_idr r r �track� s zBuildTracker.track)r/ r0 r1 r2 r r6 r7 r r � BaseExceptionr r< r. rD r rR rT r; r$ �contextmanagerr rW r r r r r 9 s � r )r$ r> �loggingr �typesr �typingr r r r r Z pip._internal.req.req_installr � pip._internal.utils.temp_dirr � getLoggerr/ r( rY r r r- r. r r r r r �<module> s