=2gdZddlmZddlZddlZddlmZddlmZm Z ddl Z ddl m Z m Z mZddlmZmZd d d  dd Zd dd  ddZ ddZy)a/ Convert .tar.bz2 to .conda Uses ``tempfile.SpooledTemporaryFile`` to buffer ``pkg-*.tar`` and ``info-*.tar``, then compress directly into an open `ZipFile` at the end. `SpooledTemporaryFile` buffers the first 10MB of the package and its metadata in memory, but writes out to disk for larger packages. ) annotationsN)Path)CallableIterator)ZSTD_COMPRESS_LEVELZSTD_COMPRESS_THREADS conda_builder)CondaComponentstream_conda_componentc@tjttSN)levelthreads zstandardZstdCompressorrr Alib/python3.12/site-packages/conda_package_streaming/transmute.pyr ((!+@rc$|jdSNzinfo/ startswithfilenames rrr"h6I6I'6Rr compressoris_infoc|jdsJdtjj|sJtjj |dt d }t |}t|||||S)a Convert .tar.bz2 conda package to .conda-format under path. :param package: path to .tar.bz2 conda package :param path: destination path for transmuted .conda package :param compressor: A function that creates instances of ``zstandard.ZstdCompressor()`` to override defaults. :param is_info: A function that returns True if a file belongs in the ``info`` component of a `.conda` package. ``conda-package-handling`` (not this package ``conda-package-streaming``) uses a set of regular expressions to keep expected items in the info- component, while other items starting with ``info/`` wind up in the pkg- component. :return: Path to transmuted package. .tar.bz2z#can only convert .tar.bz2 to .condaN)r!r"package_stream)endswithospathisdirbasenamelenr transmute_stream)packager(r!r"stemr%s r transmuter/s}4   J 'N)NN ' 77==   77  G $%7J'7 8D+G4N   %  rc@tjttSrrrrrrrGrrc$|jdSrrrs rrrJrrc t||d}t||||5}|D]H\}}|jr"|j||j |8|j|J ddd|S#1swY|SxYw)a? Convert (TarFile, TarInfo) iterator like those produced by ``stream_conda_component`` to .conda-format under path. Allows for more creative data sources. e.g. recompress ``.conda``: .. code-block:: python transmute_stream(..., package_stream=itertools.chain( stream_conda_component("package.conda", component=CondaComponent.pkg), stream_conda_component("package.conda", component=CondaComponent.info), )) This example could move files between the ``pkg-`` and ``info-`` components depending on the ``is_info`` function. :param stem: output filename without extension :param path: destination path for transmuted .conda package :param compressor: A function that creates instances of ``zstandard.ZstdCompressor()`` to override defaults. :param is_info: A function that returns True if a file belongs in the ``info`` component of a `.conda` package. ``conda-package-handling`` (not this package ``conda-package-streaming``) uses a set of regular expressions to keep expected items in the info- component, while other items starting with ``info/`` wind up in the pkg- component. :param package_stream: Iterator of (Tarfile, TarInfo) tuples. :return: Path to transmuted package. .condar N)rr isfileaddfile extractfile) r.r(r!r"r% output_path conda_tartarmembers rr,r,AsXtvV_-K tTj' Ji)KC}}!!&#//&*AB!!&) * K  K s AA88Bc |jdsJdtjj|sJ|jdrdnd}tjj |dt | }|dk(r!t jt jg}nt jg}t||d}t|d5}tj|d5}|D]f}tt|||} | D]H\} } | jr"|j| | j!| 8|j| Jh dddddd|S#1swYxYw#1swY|SxYw) a Convert .conda package to .tar.bz2 format under path. Can recompress .tar.bz2 packages. Args: package: path to .conda or .tar.bz2 package. path: destination path for transmuted package. Returns: Path to transmuted package. )r$r3zUnknown extensionr3r$Nrbzx:bz2) component)r&r'r(r)r*r+r infopkgropentarfileiterr r4r5r6) r-r(incoming_formatfile_id componentsr7fileobjpkg_tarr=streamr9r:s rtransmute_tar_bz2rIxsH   2 3H5HH 3 77==  ")"2"28"+=+=> %(() ty12K gt k7)Kw#I0'YWXF% V==?OOFCOOF,CDOOF+ &$*L  *L)K  s% E/$A,E#E/#E, (E//E9)r!&Callable[[], zstandard.ZstdCompressor]r"Callable[[str], bool]returnr)r!rJr"rKr%z1Iterator[tuple[tarfile.TarFile, tarfile.TarInfo]])r-strrLr)__doc__ __future__rr'rApathlibrtypingrrrcreaterr r package_streamingr r r/r,rIrrrrTs# %MMF &S% %#% %\ &S4 4#4F4n) ) )r