a
lj @ s U d dl Z d dlZd dlZd dlZd dlZd dlmZmZ d dl m
Z
m Z m Z m
Z
mZmZ d dlmZmZ eeZedddZedd d
d Zdae
e ed < ee d d
ddZG dd dZda e
e ed< ee e d
ddZ G dd dZ e Z G dd dZ G dd de Z!dS ) N) ExitStackcontextmanager)AnyDictIteratorOptionalTypeVarUnion)enumrmtree_T
TempDirectory)boundz build-envzephem-wheel-cachez req-build)Z BUILD_ENVZEPHEM_WHEEL_CACHEZ REQ_BUILD_tempdir_managerreturnc c sJ t 0} t| }azd V W |an|a0 W d n1 s<0 Y d S N)r r )stackZold_tempdir_manager r @/usr/lib/python3.9/site-packages/pip/_internal/utils/temp_dir.pyglobal_tempdir_manager s
r c @ s@ e Zd ZdZddddZeeddddZeed d
d ZdS ) TempDirectoryTypeRegistryz Manages temp directory behaviorNr c C s
i | _ d S r _should_deleteselfr r r __init__* s z"TempDirectoryTypeRegistry.__init__)kindvaluer c C s || j |< dS )z[Indicate whether a TempDirectory of the given kind should be
auto-deleted.
Nr )r r r r r r
set_delete- s z$TempDirectoryTypeRegistry.set_deleter r c C s | j |dS )z^Get configured auto-delete flag for a given TempDirectory type,
default True.
T)r get)r r r r r
get_delete3 s z$TempDirectoryTypeRegistry.get_delete) __name__
__module__ __qualname____doc__r strboolr r" r r r r r ' s r _tempdir_registryc c s$ t } t a zt V W | a n| a 0 dS )zuProvides a scoped global tempdir registry that can be used to dictate
whether directories should be deleted.
N)r) r )Zold_tempdir_registryr r r tempdir_registry= s
r* c @ s e Zd ZdS )_DefaultN)r# r$ r% r r r r r+ K s r+ c s e Zd ZdZdeddfee eede f eed fdd
Z
e edd d
Z edd d Z
eed
ddZeeeddddZeedddZddddZ ZS )r
aM 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.
NZtempF)pathdeleter globally_managedc sl t |tu r$|d ur d}nd }|d u r6| |}|| _d| _|| _|| _|rhtd us^J t | d S )NF)
superr _default_create_path_deletedr- r r
enter_context)r r, r- r r. __class__r r r g s
zTempDirectory.__init__r c C s | j rJ d| j | jS )Nz"Attempted to access deleted path: )r3 r2 r r r r r, s zTempDirectory.pathc C s d| j j d| jdS )N< >)r6 r# r, r r r r __repr__ s zTempDirectory.__repr__)r r c C s | S r r r r r r __enter__ s zTempDirectory.__enter__)excr tbr c C s8 | j d ur| j }ntr$t| j}nd}|r4| d S )NT)r- r) r" r cleanup)r r< r r= r- r r r __exit__ s
zTempDirectory.__exit__r c C s, t jtjd| dd}td| |S )z s zTempDirectory.cleanup)r# r$ r% r& r0 r r' r r( r+ r propertyr, r: r r; r r? r1 r>
__classcell__r r r5 r r
R s$ !
c s\ e Zd ZdZdZd
eee dd fdd
Ze ee
e ddd Z eed
d d 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-~.=%0123456789N)originalr- r c s | d| _t j|d d S )Nz/\)r- )rstriprO r/ r )r rO r- r5 r r r s z AdjacentTempDirectory.__init__)namer c c s t dt|D ]D}t| j|d D ],}dd| ||d }||kr$|V q$qt t| jD ]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)rangelen itertools combinations_with_replacement
LEADING_CHARSjoin)clsrQ i candidatenew_namer r r _generate_names s
z%AdjacentTempDirectory._generate_namesr c C s t j| j\}}| |D ]d}t j||}zt | W n2 tyn } z|jtj krZ W Y d }~q d }~0 0 t j
|} qq t j
t j d| dd}t
d| |S )Nr@ rA rB rD )rE r, splitrO r_ rZ mkdirOSErrorerrnoZEEXISTrF rG rH rI rJ )r r rootrQ r] r, exr r r r1 s z AdjacentTempDirectory._create)N)r# r$ r% r& rY r' r r( r classmethodr r_ r1 rM r r r5 r rN s rN )"rc rW ZloggingZos.pathrE rG
contextlibr r typingr r r r r r Zpip._internal.utils.miscr
r Z getLoggerr# rI r Z
tempdir_kindsr __annotations__r r r) r* r+ r0 r
rN r r r r s2
^