e`.ddlmZmZmZddlZddlZddlZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlmZddlmZddlmZddlmZmZddlmZdd Z dd ZGddeZGddeZGddeZGddeZGddeZ dZ!GddeZ"ej#dkre j$j%Z&dSddZ&dS) )print_functiondivisionabsolute_importN)closing)BytesIO) ThreadPool)Queueon_win)CondaPackExceptionc\|dkrddlm}|S|dkrtd|S)Nr) cpu_countr z+n-threads must be >= 1, or -1 for all cores)multiprocessingrr ) n_threadsrs 2lib/python3.11/site-packages/conda_pack/formats.py_parse_n_threadsrsGB------y{{1}} !NOOO FTct|}|dkrt||||S|dvr"|dkrd}d}nAd}d}t||| }n*|d vr"|dkrd }d}nd}d}t||| }nd}d}t ||||| S) Nzip) zip_symlinkszip_64)ztar.gztgzparcelr zw:gzFwT) compresslevelr)ztar.bz2tbz2zw:bz2) close_filemoder)r ZipArchiveParallelGzipFileWriterParallelBZ2FileWriter TarArchive) fileobjarcrootformatcompress_levelrrrr rs rarchiver)s!++I '7!'))) ),,, >>DJJDJ,WN7@BBBGG & & & >>DJJDJ+G>6?AAAGG gw:~ ? ? ??rc4eZdZd dZdZdZdZdZdZd S) ParallelFileWriter r c||_||_||_d|_||g|_d|_t||_ t||_ tj |j|_d|j_|jdS)Nr)maxsize)targetT)r%rrsize _init_state _write_headerbuffers buffer_lengthrpoolr compress_queue threadingThread _consumer_consumer_threaddaemonstart)selfr%rrs r__init__zParallelFileWriter.__init__Cs *"    y)) #I666 ) 0 G G G'+$ ##%%%%%rc|jSN)r0r=s rtellzParallelFileWriter.tellXs yrct|tst|}t|}|dkr|||xj|z c_|xj|z c_|j||j|j kr-|j |jg|_d|_|SNr) isinstancebytes memoryviewlen_per_buffer_opr0r4r3append _block_sizer6put)r=datans rwritezParallelFileWriter.write[s$&& $d##D II q55    % % % IINII   ! #   L   % % %!D$444#'' 555! %&"rc4t|j5|j|jt |jjdD]0}|D]+}t|r|j |,1 ddddS#1swxYwYdSr@) rr5imap _compressiterr6getrHr%rO)r=r3bufs rr9zParallelFileWriter._consumerjs TY   0 09>>ND)<)@$$G$GII 0 0"00C3xx0 **3///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 0sA*B  BBcg}|}|D]*}|||+||||Sr@)_new_compressorrJcompress_flush_compressor)r=in_bufsout_bufs compressorrMs rrRzParallelFileWriter._compressrso))++  7 7D OOJ//55 6 6 6 6..z::;;;rcN|jdS|jr|j|j|jd|j||jd|_d|_d|_dSr@) r%r3r6rLr:join _write_footerflushr5rAs rclosezParallelFileWriter.closezs <  F < 2   # #DL 1 1 1 %%% ""$$$  #  rN)r,r ) __name__ __module__ __qualname__r>rBrOr9rRrarrr+r+Bss&&&&*   000rr+c<eZdZdZdZdZdZdZdZdZ dZ d S) r"ic>tjddz|_dS)Nrzlibcrc32crcrAs rr1z"ParallelGzipFileWriter._init_states:c??Z/rcztj|jtjtj tjdSrD)rj compressobjrDEFLATED MAX_WBITS DEF_MEM_LEVELrAs rrWz&ParallelGzipFileWriter._new_compressors1 2DM!%1CQHH HrcJtj||jdz|_dSNrhrir=buffers rrIz%ParallelGzipFileWriter._per_buffer_ops :fdh//*<rc`|jtjd|dS)Nzddl}||jSrD)bz2 BZ2Compressorr)r=rs rrWz%ParallelBZ2FileWriter._new_compressors"   !3444rcdSr@rerts rrIz$ParallelBZ2FileWriter._per_buffer_op rcdSr@rerAs rr2z#ParallelBZ2FileWriter._write_headerrrcdSr@rerAs rr_z#ParallelBZ2FileWriter._write_footerrrc*|Sr@)r`rs rrYz'ParallelBZ2FileWriter._flush_compressors!!!rN) rbrcrdr1rWrIr2r_rYrerrr#r#sn<<<555         """""rr#ceZdZdZdZdS) ArchiveBasec|tj|j|}|||dSr@)ospathr^r&_addr=sourcer/s raddzArchiveBase.adds3dlF33 &&!!!!!rc~tj|j|}||||dSr@)rrr^r& _add_bytes)r=r sourcebytesr/s r add_byteszArchiveBase.add_bytess5dlF33  V44444rN)rbrcrdrrrerrrrs2"""55555rrc2eZdZ d dZdZdZdZdZd S) r$FrrcL||_||_||_||_||_dSr@)r%r&rr r)r=r%r&rr rs rr>zTarArchive.__init__s,  $ *rc|jdkr d|jini}tjd|jt |jd||_|S)Nrr)r% dereferencer re)r rtarfileopenr%r r))r=kwargss r __enter__zTarArchive.__enter__s]:>)s:J:J/4#566PR|.DL06)-..'-..  rc||j|jr|jdSdSr@)r)rarr%)r=argss r__exit__zTarArchive.__exit__sD  ? ! L    ! !rc@|j||ddS)NF) recursive)r)rrs rrzTarArchive._adds% 599999rc|j||}t||_|j|t |dSr@)r) gettarinforHr0addfilerr=rrr/infos rrzTarArchive._add_bytessN|&&vv66 $$  T7;#7#788888rN)Frrrbrcrdr>rrrrrerrr$r$sj49)*++++!!! :::99999rr$a The following conda package file is a symbolic link that does not match an existing file within the same package: {0} It is likely this link points to a file brought into the environment by a dependency. Unfortunately, conda-pack does not support this practice for zip files unless the --zip-symlinks option is engaged. Please see "conda-pack --help" for more information about this option, or use a tar-based archive format instead.c.eZdZd dZdZdZdZdZdS) r!FTc>||_||_||_||_dSr@)r%r&rr)r=r%r&rrs rr>zZipArchive.__init__s#  ( rchtj|jd|jtj|_|S)Nr) allowZip64 compression)zipfileZipFiler%r ZIP_DEFLATEDr)rAs rrzZipArchive.__enter__ s5t|S26+3:3GIII  rc|jt|tjrt ddS)Nz;Large Zip File: ZIP64 extensions required but were disabled)r)rarEr LargeZipFiler )r=typery tracebacks rrzZipArchive.__exit__sL  eW1 2 2 %$$%% % % %rc  tj|}tj|j}n#t t f$rd}YnwxYw|rb|jrtj |}d|_ |jdzdz|_ tj |r|xj dzc_ |j|tj|dStj |rtj|dD]\}}}tj |tj ||} |D]Y} |jtj || tj | | Z|s|s|j|| dS |j||dS#t $r} | jt*jkr|t/| d|krNtj |dt/| dz }d||}t4|} t7| d} ~ wwxYw|j||dS) NFT) followlinksr z{}: {})rlstatstatS_ISLNKst_modeOSErrorAttributeErrorrrZipInfo create_system external_attrrisdirr)writestrreadlinkwalkr^relpathrOerrnoENOENTrHbasenamer'_dangling_link_errorr ) r=rr/stis_linkrrootdirsfilesroot2filemsgs rrzZipArchive._adds &!!Bl2:..GG(   GGG  ! /  v..%&"&(j6&9b%@"7==((/&&$.&& %%dBK,?,?@@@@@7==((-/WV-N-N-N<<)dE " VRW__T65R5R S S#(IIC L..rw||D#/F/F/1w||E3/G/GIIII#>*,)9)9&3v;;,q.:Q)R)R)1)H)H"6"="=f"E"EC"4S"9"99  L  vv . . . . .s(-0AAG33 J+=B)J&&J+c\t||}|j||dSr@)zipinfo_from_filer)rrs rrzZipArchive._add_bytesCs/ 00 dK00000rN)FTrrerrr!r!se  %%%(/(/(/T11111rr!)rcntj|}tj|j}t j|j}|dd}||}tjtj |d}|dtj tj fvr*|dd}|dtj tj fv*|r|dz }tj ||}|jdzdz|_|rd|_|xjdzc_n |j|_|S)Nrrr /rr)rrS_ISDIRrr~ localtimest_mtimernormpath splitdrivesepaltseprrr file_sizest_size)filenamearcnamerrmtime date_timezinfos rrrLs WX   RZ((r{++!A#J ?G'""27#5#5g#>#>q#ABBajRVRY///abbkGajRVRY///   sNG33!zF2r9  )EO   4 '    jEO r)r )rFTr r@)' __future__rrrrrrrwsysrr7r~rrj contextlibriormultiprocessing.poolrcompatr r corer rr)objectr+r"r#rr$rr! version_infor from_filerrerrrsl@@@@@@@@@@   ++++++!!!!!!!!$$$$$$FK#$ ? ? ? ?FMMMMMMMM`"3"3"3"3"3/"3"3"3J""""".""",55555&55599999999B %@1@1@1@1@1@1@1@1Fv1r