ndbdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ej dZ ej e  ddlZn!#e$rddlZejddZYnwxYwGdde ZGd d ejZejd fddZ dddZdejfddZdS)z7 Unpack conda packages without using a temporary file. ) annotationsN)Enum) Generatorz@zstandard could not be imported. Running without .conda support.ceZdZdZdZdZdS)CondaComponentpkginfoc|jSN)value)selfs Ilib/python3.11/site-packages/conda_package_streaming/package_streaming.py__str__zCondaComponent.__str__"s zN)__name__ __module__ __qualname__rr rrrrrs- C Drrc0eZdZedfd ZdZdZxZS)TarfileNoSameOwner)umaskcHtj|i|||_dS)aOpen an (uncompressed) tar archive `name'. `mode' is either 'r' to read from an existing archive, 'a' to append data to an existing file or 'w' to create a new file overwriting an existing one. `mode' defaults to 'r'. If `fileobj' is given, it is used for reading or writing data. If it can be determined, `mode' is overridden by `fileobj's mode. `fileobj' is not closed, when TarFile is closed. N)super__init__r)r rargskwargs __class__s rrzTarfileNoSameOwner.__init__'s, $)&))) rcdS)z Override chown to be a no-op, since we don't want to preserve ownership here. (tarfile.TarFile only lets us toggle all of (chown, chmod, mtime)) Nr)r tarinfo targetpath numeric_owners rchownzTarfileNoSameOwner.chown3s rc tj||jd|jz zdS#t$r}t jd|d}~wwxYw)zd Set file permissions of targetpath according to tarinfo, respecting umask. izcould not change modeN)oschmodmoderOSErrortarfile ExtractError)r rr es rr%zTarfileNoSameOwner.chmod:se  G HZ1C!D E E E E E G G G&'>??Q F Gs%) A AA )rrrUMASKrr"r% __classcell__)rs@rrr&sj$)       GGGGGGGrrFreturn>Generator[tuple[tarfile.TarFile, tarfile.TarInfo], None, None]c#K ||d5}|D]}||fV dddn #1swxYwY|r|dSdS#|r|wwxYw)z+ Yield (tar, member) from fileobj. zr|)fileobjr&N)close)r0 tarfile_openclosefdtarmembers r tar_generatorr6Es \' 5 5 5 " " "6k!!!! " " " " " " " " " " " " " " " "   MMOOOOO  7  MMOOOO s& A * A.A.AA(c(d}t|||S)a Yield members from conda's embedded info/ tarball. For .tar.bz2 packages, yield all members. Yields (tar, member) tuples. You must only use the current member to prevent tar seeks and scans. To extract to disk, it's possible to call ``tar.extractall(path)`` on the first result and then ignore the rest of this generator. ``extractall`` takes care of some directory permissions/mtime issues, compared to ``extract`` or writing out the file objects yourself. r )stream_conda_component)filenamer0 components rstream_conda_infor;Vs I !(GY ? ??rr:CondaComponent | strct|drttdt j|p|}t j| d\}}}|d|fd| D}|stdd|t|d ksJtj ||d }nJt|d rt!j|p|d }nt#dt%||duS)a  Yield members from .conda's embedded {component}- tarball. "info" or "pkg". For .tar.bz2 packages, yield all members. Yields (tar, member) tuples. You must only use the current member to prevent tar seeks and scans. To extract to disk, it's possible to call ``tar.extractall(path)`` on the first result and then ignore the rest of this generator. ``extractall`` takes care of some directory permissions/mtime issues, compared to ``extract`` or writing out the file objects yourself. z.condaNz(Cannot unpack `.conda` without zstandard.-cHg|]}|j|Sr)r9 startswith).0r component_names r z*stream_conda_component..s>   dm.F.F~.V.V    rz didn't find z component in rz.tar.bz2rb)r&zunsupported file extension)r3)strendswith zstandard RuntimeErrorzipfileZipFiler$pathbasename rpartitioninfolist LookupErrorlenZstdDecompressor stream_readeropenbz2 ValueErrorr6) r9r0r:zffile_id_component_filenamereaderrCs @rr8r8js  8}}h''7  IJJ J _W0 1 1((22==cBB A%1111    [[]]   " WU^UU8UUVV V%&&!+++++--;; GG&q) * *   X   + +7'-XD9995666 D 9 9 99r)r-r.r )r:r<r-r.)__doc__ __future__rrVr$os.pathr(rKenumrtypingrrr+rI ImportErrorwarningswarnrTarFilerrUr6r;rr8rrrrfs#"""""   OOOHMTUUUIII TGGGGGGGG@-15$@@@@@*n>P$:$:$:$:$:$:$:s AA-,A-