U
��.ei � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZejdd� �Zdd� Z dd� Z
dS )� Nc c s$ t �� } z
| V W 5 t�| � X dS )z2Create a temporary directory in a context manager.N)�tempfileZmkdtemp�shutilZrmtree)Ztd� r �?/usr/lib/python3.8/site-packages/pip/_vendor/pep517/dirtools.py�tempdir
s
r c
O sD zt j| |�W S tk
r> } z|jtjkr.� W 5 d}~X Y nX dS )zWLike `mkdir`, but does not raise an exception if the
directory already exists.
N)�os�mkdir�OSError�errnoZEEXIST)�args�kwargs�excr r r �mkdir_p s
r c C s� t �� }t�|d�}t�| �D ]r\} }}|D ]0}tj�| |�}tj�|| �}|� |d d� q,|D ],}tj�| |�}tj�|| �}|�
||� qbq |S )z0Construct an in-memory zip file for a directory.�w�/� ) �io�BytesIO�zipfileZZipFiler �walk�path�join�relpathZwritestr�write)�root�bufferZzip_file�dirs�filesr Zfs_pathZrel_pathr r r �dir_to_zipfile s r ) r r �
contextlibr r r
r �contextmanagerr r r r r r r � s