U
��.e� � @ s� d dl mZ d dlZd dlZd dlZd dlZd dlZd dlm Z d dl
m Z e r\d dl m
Z
e�e�ZG dd� de�ZG dd � d e�ZdS )
� )�absolute_importN)�rmtree)�MYPY_CHECK_RUNNING)�Optionalc sV e Zd ZdZd� fdd� Zedd� �Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z � Z S )�
TempDirectoryaM Helper class that owns and cleans up a temporary directory.
This class can be used as a context manager or as an OO representation of a
temporary directory.
Attributes:
path
Location to the created temporary directory
delete
Whether the directory should be deleted when exiting
(when used as a contextmanager)
Methods:
cleanup()
Deletes the temporary directory
When used as a context manager, if the delete attribute is True, on
exiting the context the temporary directory is deleted.
N�tempc sP t t| ��� |d kr"|d kr"d}|d kr4| �|�}|| _d| _|| _|| _d S )NTF)�superr �__init__�_create�_path�_deleted�delete�kind)�self�pathr
r �� __class__� �@/usr/lib/python3.8/site-packages/pip/_internal/utils/temp_dir.pyr + s
zTempDirectory.__init__c C s | j rtd�| j���| jS )Nz$Attempted to access deleted path: {})r �AssertionError�formatr �r r r r r @ s
�zTempDirectory.pathc C s d� | jj| j�S )Nz <{} {!r}>)r r �__name__r r r r r �__repr__H s zTempDirectory.__repr__c C s | S �Nr r r r r � __enter__K s zTempDirectory.__enter__c C s | j r| �� d S r )r
�cleanup)r �exc�value�tbr r r �__exit__N s zTempDirectory.__exit__c C s. t j�tjd�|�d��}t�d�|�� |S )zECreate a temporary directory and store its path in self.path
�pip-{}-��prefix� Created temporary directory: {})�osr �realpath�tempfile�mkdtempr �logger�debug)r r r r r r r
R s
�zTempDirectory._createc C s" d| _ tj�| j�r t| j� dS )z?Remove the temporary directory created and reset state
TN)r r% r �existsr r r r r r r _ s zTempDirectory.cleanup)NNr )
r �
__module__� __qualname__�__doc__r �propertyr r r r r
r �
__classcell__r r r r r s �
r c s: e Zd ZdZdZd
� fdd� Zedd� �Zdd � Z� Z S ) �AdjacentTempDirectorya� Helper class that creates a temporary directory adjacent to a real one.
Attributes:
original
The original directory to create a temp directory for.
path
After calling create() or entering, contains the full
path to the temporary directory.
delete
Whether the directory should be deleted when exiting
(when used as a contextmanager)
z-~.=%0123456789Nc s" |� d�| _tt| �j|d� d S )Nz/\)r
)�rstrip�originalr r1 r )r r3 r
r r r r | s z AdjacentTempDirectory.__init__c c s� t dt|��D ]D}t�| j|d �D ],}dd�|� ||d� }||kr$|V q$qt t| j��D ]8}t�| j|�D ]$}dd�|� | }||krt|V qtqbdS )a Generates a series of temporary names.
The algorithm replaces the leading characters in the name
with ones that are valid filesystem characters, but are not
valid package names (for both Python and pip definitions of
package).
� �~� N)�range�len� itertools� combinations_with_replacement�
LEADING_CHARS�join)�cls�name�i� candidate�new_namer r r �_generate_names� s �
�z%AdjacentTempDirectory._generate_namesc C s� t j�| j�\}}| �|�D ]b}t j�||�}zt �|� W n0 tk
rl } z|jtj kr\� W 5 d }~X Y q X t j�
|�} q�q t j�
t j d�
|�d��}t�d�
|�� |S )Nr! r" r$ )r% r �splitr3 rB r< �mkdir�OSError�errnoZEEXISTr& r' r( r r) r* )r r �rootr> r@ r Zexr r r r
� s �z AdjacentTempDirectory._create)N)
r r, r- r. r; r � classmethodrB r
r0 r r r r r1 g s
r1 )Z
__future__r rF r9 ZloggingZos.pathr% r' Zpip._internal.utils.miscr Zpip._internal.utils.typingr �typingr Z getLoggerr r) �objectr r1 r r r r � s
Q