o g@sddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z m Z mZmZmZmZddlmZmZmZddlmZddlmZded efd d Zd ed ed dfddZded efddZedede d ee fddZedededdZ e ej!Z!ded efddZ"ded efddZ#deded eefddZ$ded ee%e&ffd d!Z'ded efd"d#Z(ded ee%e&ffd$d%Z)ded efd&d'Z*dS)(N)contextmanager)NamedTemporaryFile)AnyBinaryIOIteratorListUnioncast)retrystop_after_delay wait_fixed) get_path_uid) format_sizepathreturncCstjdks ttds dStj|sJd}||krNtj|rAtdkr:z t|}W|dkSt y9YdSwt |tj S|tj |}}||ksdS)Nwin32geteuidTrF) sysplatformhasattrosrisabslexistsrr OSErroraccessW_OKdirname)rpreviouspath_uidr@/usr/lib/python3/dist-packages/pip/_internal/utils/filesystem.pycheck_path_owners"    r!srcdestc Cshz t||WdSty3||fD]}zt|}Wn ty%Yqw|r1td|dqw)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#fis_socket_filerrr copy2_fixed-s    r+cCstt|jSN)statS_ISSOCKrlstatst_moderrrr r'Dsr'kwargsc kstddtj|tj|dd|+}tt|}z|VW|t| n |t| wWddS1sCwYdS)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)deletedirprefixsuffixNr) rrrrbasenamer rflushfsyncfileno)rr2r)resultrrr adjacent_tmp_fileHs$    "r<Tg?)reraisestopwaitcCsRtj|stj|}||krn|}tj|rtjdkr%t|tjSt|S)zgCheck if a directory is writable. Uses os.access() on POSIX, tries creating files on Windows. posix)rrisdirrnamerr_test_writable_dir_win)rparentrrr test_writable_diris   rFc sd}dtdD]H}|dfddtdD}tj||}zt|tjtjBtjB}Wnty9Yqt yCYdSwt |t |d St d ) N(accesstest_deleteme_fishfingers_custard_$abcdefghijklmnopqrstuvwxyz0123456789 c3s|]}tVqdSr,)randomchoice).0_alphabetrr sz)_test_writable_dir_win..FTz3Unexpected condition testing for writable directory) rangejoinrropenO_RDWRO_CREATO_EXCLFileExistsErrorPermissionErrorcloseunlinkr)rr7rNrCfilefdrrOr rD{s       rDpatterncsBg}t|D]\}}t||}|fdd|Dq|S)zReturns a list of absolute paths of files beneath path, recursively, with filenames which match the UNIX-style shell glob pattern.c3s|] }tj|VqdSr,)rrrT)rMr)rootrr rQszfind_files..)rwalkfnmatchfilterextend)rr_r;rNfilesmatchesrr`r find_filess  rhcCstj|rdStj|S)Nr)rrislinkgetsizer1rrr file_sizes  rkcC tt|Sr,)rrkr1rrr format_file_size rmcCsBd}t|D]\}}}|D]}tj||}|t|7}qq|S)Ng)rrbrrTrk)rsizera_dirsrffilename file_pathrrr directory_sizesrscCrlr,)rrsr1rrr format_directory_sizernrt)+rcros.pathrKr%r-r contextlibrtempfilertypingrrrrrr pip._vendor.tenacityr r r pip._internal.utils.compatr pip._internal.utils.miscrstrboolr!r+r'r<_replace_retryreplacerFrDrhintfloatrkrmrsrtrrrr s6