a
lj @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dl m
Z
d dl m Z m
Z
mZmZmZmZ d dlmZmZmZ d dlmZ d dlmZ eedd d
Z eedd d d
Z eedddZ eee ee
dddZ edededdZ e ej!Z!eedddZ"eedddZ#eeee dd d Z$eee%e&f dd d Z'eedd d!Z(eee%e&f dd"d#Z)eedd$d%Z*dS )& N)contextmanager)NamedTemporaryFile)AnyBinaryIOIteratorListUnioncast)retrystop_after_delay
wait_fixed) get_path_uid) format_size)pathreturnc C s t jdksttdsdS d }| |krtj| rvt dkrfz t| }W n ty\ Y dS 0 |dkS t | tj
S q | tj | }} q dS )Nwin32geteuidTr F) sysplatformhasattrosr lexistsr r
OSErroraccessW_OKdirname)r ZpreviousZpath_uid r B/usr/lib/python3.9/site-packages/pip/_internal/utils/filesystem.pycheck_path_owner s r )srcdestr c C sn zt | | W nX tyh | |fD ]:}z t|}W n tyH Y q&0 |r&t d| dq& Y n0 dS )zWrap shutil.copy2() but map errors copying socket files to
SpecialFileError as expected.
See also https://bugs.python.org/issue37700.
`z
` is a socketN)shutilcopy2r is_socketSpecialFileError)r r fZis_socket_filer r r copy2_fixed- s r' c C s t t| jS N)statS_ISSOCKr lstatst_moder r r r r$ D s r$ )r kwargsr c k s t f dtj| tj| dd|T}tt|}z |V W | t| n| t| 0 W d n1 s0 Y dS )a% Return a file-like object pointing to a tmp file next to path.
The file is created securely and is ensured to be written to disk
after the context reaches its end.
kwargs will be passed to tempfile.NamedTemporaryFile to control
the way the temporary file will be opened.
Fz.tmp)deletedirprefixsuffixN)
r r r r basenamer r flushfsyncfileno)r r. r& resultr r r adjacent_tmp_fileH s
r8 T g ?)Zreraisestopwaitc C sH t j| s(t j| }|| kr"q(|} q t jdkr@t | t jS t| S )zgCheck if a directory is writable.
Uses os.access() on POSIX, tries creating files on Windows.
posix)r r isdirr namer r _test_writable_dir_win)r parentr r r test_writable_diri s
rA c s d}d t dD ]}|d fddt dD }tj| |}z t|tjtjB tjB }W n& tyr Y q t y Y dS 0 t
| t | d S qt d
d S ) NZ(accesstest_deleteme_fishfingers_custard_Z$abcdefghijklmnopqrstuvwxyz0123456789
c 3 s | ]}t V qd S r( )randomchoice).0_Zalphabetr r z)_test_writable_dir_win.. FTz3Unexpected condition testing for writable directory)
rangejoinr r openO_RDWRO_CREATO_EXCLFileExistsErrorPermissionErrorcloseunlinkr )r r3 rG r> filefdr rH r r? { s
r? )r patternr c sB g }t | D ].\ }}t||}| fdd|D q|S )zReturns a list of absolute paths of files beneath path, recursively,
with filenames which match the UNIX-style shell glob pattern.c 3 s | ]}t j |V qd S r( )r r rM )rF r& rootr r rI rJ z find_files..)r walkfnmatchfilterextend)r rX r7 rG filesmatchesr rY r
find_files s
ra c C s t j| rdS t j| S )Nr )r r islinkgetsizer- r r r file_size s rd c C s t t| S r( )r rd r- r r r format_file_size s re c C sB d}t | D ].\}}}|D ] }t j||}|t|7 }q q|S )Ng )r r[ r rM rd )r sizerZ Z_dirsr_ filenameZ file_pathr r r directory_size s rh c C s t t| S r( )r rh r- r r r format_directory_size s ri )+r\ r Zos.pathrD r" r) r
contextlibr Ztempfiler typingr r r r r r Zpip._vendor.tenacityr
r r Zpip._internal.utils.compatr
Zpip._internal.utils.miscr strboolr r' r$ r8 Z_replace_retryreplacerA r? ra intfloatrd re rh ri r r r r s4