d\<UddlmZddlZddlZddlZddlmZ ddl m Z m Z ddl mZddlmZddlmZdd lmZdd lmZd eiZd ed <dZ ddlmZmZddlmZeed<dZn#e $rej!dYnwxYwdZ"dZ#ddZ$ddZ%ddZ& ddZ'ddZ(dZ)dS) ) annotationsN)glob)ConversionErrorInvalidArchiveError)AbstractBaseFormat) CondaTarBZ2filter_info_files) get_executor)rm_rf.tar.bz2z#dict[str, type[AbstractBaseFormat]]SUPPORTED_EXTENSIONSF)ZSTD_COMPRESS_LEVELZSTD_COMPRESS_THREADS)CondaFormat_v2.condaTz4Install zstandard Python bindings for .conda supportc ^ggc}tjD]~\}}|D]S}tjtj|T|fd|Dfd|Dz}|S)Nc3K|]B}tjtj|VCdSN)_ospathrelpathjoin).0_dpprefixs :lib/python3.11/site-packages/conda_package_handling/api.py z!_collect_paths..'sETTA))#(--A*>*>GGTTTTTTcLg|]tfdD S)c3\K|]&}|tjzV'dSr) startswithrsep)rfrs rr z,_collect_paths...)s5)Y)Y!,,rCG|*D*D)Y)Y)Y)Y)Y)Yr!)any)rr file_pathss @r z"_collect_paths..(sLc)Y)Y)Y)Yj)Y)Y)Y&Y&Y r!)rwalkappendrrrextend)r dir_pathsdn filenamesr& file_listrr(s` @@r_collect_pathsr1"sIz Xf--UUB  N NA   ch..sx}}R/C/CVLL M M M MTTTTTQSTTTTTTTI r!cd}tD]/}||r|dt| }0|r!|stj|}|Sr)rendswithlenrrbasename)in_fileabspathdirnameexts rget_default_extracted_folderr:.slG#++   C  +kS k*G-w-(##G,, Nr!c n|rtj|r|rtdtj|sPtjtj|ptj|}nNtj|ptj|t|d}tj |stj |t D]2}| |r||||dS3td|t!t)Nzdest_dir and prefix both provided as abs paths. If providing both, prefix can be abspath, but dest dir must be relative (relative to prefix)Fr7) componentsGDidn't recognize extension for file '{}'. Supported extensions are: {})rrisabs ValueErrornormpathrgetcwdr8r:isdirmakedirsrvalues supportedextractformatlistkeys)fndest_dirr=rrHs rrGrG8s  8>>( # #    x~~h'' Zx((v7Mx)X)XYYH8==  *ch&&r** (U ; ; ;   8>>( # # X&--//      B    NN2xJN ? ? ? EE  U \ \D-224455     r!c |stj}tj|r>tj|}tj|}|t |}ndt|trO t|5}| }dddn #1swxYwYd|D}n#xYwd}t D]}| |rv |j||||fi|}n#t$rU} tj||} tj| rt%| | d} ~ wwxYwt'd|t+t|S)Nc6g|]}|S)strip)rrs rr)zcreate..fs 111q111r!r>)rrBrr?r8r5r1 isinstancestropen readlinesrrErFcreate BaseExceptionrisfile_rm_rfr@rHrIrJ) rr0out_fn out_folderkwr&dataoutrHerr abs_out_fns rrUrUWs "Z\\  x~~f+X%%f-- ""6**"6** Is # # i %A{{}} % % % % % % % % % % % % % % %11D111II   C&--//     F # #  #fmFIvzPPRPP     X]]:v>> 8??:..':&&&    U \ \1668899     JsIC#*C ? C# CC#CC##C&D// F9AF  Fc ddl}ddlt|d}ddlm} |s|dd|dt d fSt tj |||z} d} tj | r|rD|r3tj | rtj | |d kr@ttfd } d } tj|jj||| | }n tj|jj||} || | |}|\}}}|s|rt t%||} nj#t&$r]}tj| rt+| t-|t.st |} Yd}~nd}~wwxYw|| | fS)NrFr<r)(validate_converted_files_match_streamingz Input file z% doesn't have a supported extension (z), skipping itrc2S)N)levelthreads)ZstdCompressor) zstandardzstd_compress_levelzstd_compress_threadssr compressorz_convert..compressors& //-7L0r!c.t|gdgkS)N.)rr )filenames ris_infoz_convert..is_infos((C@@@BFFr!)rjrn)!conda_package_streaming.transmutergr:validaterarrRrrrlexistsunlinkrr _functoolspartial transmutetransmute_tar_bz2rrVrWrXrQ Exception)rKout_extrZforcerhrir[conda_package_streamingr5rarYerrorsrjrnruresultr missing_filesmismatching_sizesergs `` @r_convertrse-,,,+B>>>HBBBBBB   Brr''' )  z8g+=>> ? ?F F 8  F # #,u,  SX%%f--  Jv    h  "*&9#$,(=%         G G G#*'1;% II#*'1CRI  IKKK==fbIIF28 /A}/ P 1 P_]#*,, g  E ##GE'//*g"F"FGGHHH   ( # #GE'//(G"D"DEEFFFL y ! !X&x^Z^^[]^^ "*,,y%"@"@   B #) R v  s3AEEEctD].}||r||}n/t d||S)zWFor the new pkg format, we return the size and hashes of the inner pkg part of the filez Don't know what to do with file )rrErFget_pkg_detailsr@)r6rHdetailss rrrsp&--//GG   G $ $ ,,W55G E EGEEFFF Nr!)T)NNNr)FNN)Nr)* __future__r functoolsrsosrwarnings _warningsrr exceptionsrr interfacertarballr _CondaTarBZ2utilsr r rr rXr__annotations__libarchive_enabled conda_fmtrrr_CondaFormat_v2 ImportErrorwarnTHREADSAFE_EXTRACTr1r:rGrUrrurrOr!rrs"""""""=<<<<<<<))))))000000$$$$$$000000""""""=G&&&&Z  GGGGT(s A""A:9A: