o :n0j@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z Gddde Z d8d e ed ed efd d Zd efddZdd e eded efddZd9d e ede ed efddZded efddZded efddZded efddZded efdd Zded efd!d"Zd efd#d$Zd%ed&ed efd'd(Zejded edBfd)d*Zd efd+d,Zd efd-d.Zd efd/d0Z d efd1d2Z!d efd3d4Z"d5e ed efd6d7Z#dS):zUtility functions.N)Sequence)log_infoc@seZdZdZdS) PatoolErrorzRaised when errors occur.N)__name__ __module__ __qualname____doc__r r T/home/wildlama/miniconda3/envs/lam_a2e/lib/python3.10/site-packages/patoolib/util.pyrsrutf-8cmdencodingreturncCstj|tjd|ddjS)z#Return decoded output from command.Treplace)stdoutcheckrerrors) subprocessrunPIPEr)r rr r r backtick"s  rcCs$tjdkotjduotjdS)z?Return True iff this script is run with pythonw.exe on Windows.ntNz pythonw.exe)osnamesys executablelowerendswithr r r r run_under_pythonw)s r verbositycKs|dkrdtt|}td|trtj|d<d|d<|dkr(tj|d<|d kr1tj|d <|rT|dkrJd d d |D}td|| drTd|}tj |fi|}|j S)zIRun command without error checking. @return: command return code r zrunning creationflagsinputrrstderr, css*|]\}}|dtt|VqdS)=N) shell_quotestr).0kvr r r Js(zrun..z with shell) joinmapshell_quote_ntrrrZCREATE_NO_WINDOWDEVNULLitemsgetr returncode)r r kwargsinforesr r r r2s&    rrret_okcKs4t|fi|}||vrd|d|}t||S)z+Run command and raise PatoolError on error.z Command `z ' returned non-zero exit status )rr)r r;r7retcodemsgr r r run_checkedSs r>valuecCstjdkr t|St|S)zQuote all shell metacharacters in given string value with strong (i.e. single) quotes, handling the single quote especially. r)rrr2shell_quote_unixr?r r r r)\s r)cCs|dd}d|dS)zQuote argument for Unix system.'z'\'')rrAr r r r@es  r@cCsd|vr d|dS|S)z^Quote argument for Windows system. Modeled after distutils _nt_quote_args() function. r!"r rAr r r r2ks r2programcC0t|}|rt|dg}ttd|tjSdS)zDetermine if the RAR codec is installed for 7z program. If installed, `7z i` will print something like ... Codecs: 1 D 40301 Rar1 1 D 40302 Rar2 1 D 40303 Rar3 1 D 40305 Rar5 ... iz Rar\d$F find_programrboolresearch MULTILINErDZ_7zformatsr r r p7zip_supports_rarts   rOcCrE)zDetermine if COMPRESS (.Z) archives are supported for 7z program. If installed, `7z i` will print something like ... Formats: 0 ...................... Z z taz (.tar) 1F 9D ... rFz Z\s+z\s+tazFrGrMr r r p7zip_supports_compresss  rPcCsxtjdtj}tjdkrt|t}t|t}t|t}t }tj |r:t |D] }t|tj ||}q-|S)z>Get the list of directories to search for executable programs.PATHr)renvironr5defpathrappend_to_path get_nt_7z_dirget_nt_mac_dirget_nt_winrar_dirget_peazip_addon_dirpathisdirlistdirr0)rYZ peazip_dirsubdirr r r system_search_paths     r]rY directorycCs6tj|r ||vr |S|tjs|tj7}||S)zVAdd a directory to the PATH environment variable, if it is a valid directory. )rrYrZrpathsep)rYr^r r r rTs   rTcCstj|tdS)zLook for given program.)rY)shutilwhichr])rDr r r rHsrHc Csddl}ddl}|d}d}z6|dkr*|dr*||j|d|j|jB}n||j|}z| |ddW| |WS| |wt yPYdSw)z9Return 7-Zip directory from registry, or an empty string.rNzSOFTWARE\7-Zip32bitZ64Pathr#) winregplatform architecturemachinerOpenKeyHKEY_LOCAL_MACHINEZKEY_READZKEY_WOW64_64KEY QueryValueExZCloseKeyOSError)rdreZ python_bitsZkeynamekeyr r r rUs&   rUcCsd}tj|S)z+Return the Windows program files directory.z%ProgramFiles%)rrY expandvars)Zprogvarr r r get_nt_program_dirs rncCtjtdS)z/Return Monkey Audio Compressor (MAC) directory.zMonkey's AudiorrYr0rnr r r r rVrVcCro)zReturn WinRAR directory.ZWinRARrpr r r r rWrqrWcCs4tdkrdStdkrdStdkrdSdS)z3Get platform-dependen directory for PeaZip add-ons.Windowsz C:\Program Files\PeaZip\res\bin\Linuxz/usr/lib/peazip/res/bin/Darwinz+/Applications/peazip.app/Content/MacOS/bin/r#)resystemr r r r rXs   rXalistcCs>t|dkrd|dd}|d}|d|Sd|S)zCReturn comma separated string, and last entry appended with ' or '.rr'Nr%z or )lenr0)rvheadtailr r r strlist_with_ors  rz)r )r:)$r functoolsrrJrerr`rcollections.abcrlogr Exceptionrr*rrIrintrr>r)r@r2rOrPr]rTcacherHrUrnrVrWrXrzr r r r s<   !