o h@sddlmZddlmZmZmZmZmZmZm Z ddl Z ddl m Z ddl mZddlmZddlmZddlmZeeeeeeZe de jegee jZee d Zed d e jfd d ZGd ddeZdS))contextmanager) CFUNCTYPEPOINTERc_int c_longlongc_void_pcastcreate_string_bufferN) open_files)AbstractArchiveFileSystem) MemoryFile)DEFAULT_BLOCK_SIZEread_set_seek_callback NO_OPEN_CBallc #st|ttfdd}fdd}t|}t|}tr*tj}tj} nt j tj }t j tj } t j ||} t| |t| d||| t j | VWddS1s`wYdS)zRead an archive from a seekable file-like object. The `file` object must support the standard `readinto` and 'seek' methods. cs$}t|tt}|d<|S)Nr)readintorrr) archive_pcontextptrptrlengthbufbuf_pfileu/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/fsspec/implementations/libarchive.py read_func*s z custom_reader..read_funccs||S)N)seektell)rroffsetwhence)rrr seek_func3s z custom_reader..seek_funcN)r rrffi READ_CALLBACK SEEK_CALLBACKnew_apir NO_CLOSE_CB libarchiveread OPEN_CALLBACKVOID_CBCLOSE_CALLBACKnew_archive_readr read_open ArchiveRead) r format_name filter_name block_sizerr!read_cbseek_cbopen_cbclose_cbrrrr custom_reader!s"   "r6csneZdZdZdZdZdZddddeffdd Ze d d Z e fd d Z d dZ    dddZZS)LibArchiveFileSystemayCompressed archives as a file-system (read-only) Supports the following formats: tar, pax , cpio, ISO9660, zip, mtree, shar, ar, raw, xar, lha/lzh, rar Microsoft CAB, 7-Zip, WARC See the libarchive documentation for further restrictions. https://www.libarchive.org/ Keeps file object open while instance lives. It only works in seekable file-like objects. In case the filesystem does not support this kind of file object, it is recommended to cache locally. This class is pickleable, but not necessarily thread-safe (depends on the platform). See libarchive documentation for details. r'FrNc stj|fi||dkrtdt|tr8t|fd|i|p i}t|dkr4td|d|d|d}||_||_ ||_ d |_ d S) aB Parameters ---------- fo: str or file-like Contains ZIP, and must exist. If a str, will fetch file using :meth:`~fsspec.open_files`, which must return one file exactly. mode: str Currently, only 'r' accepted target_protocol: str (optional) If ``fo`` is a string, this value can be used to override the FS protocol inferred from a URL target_options: dict (optional) Kwargs passed when instantiating the target FS, if ``fo`` is a string. r9z%Only read from archive files acceptedprotocolzPath "z(" did not resolve to exactly one file: ""rN) super__init__ ValueError isinstancestrr lenof __enter__for1 dir_cache)selfrEmodetarget_protocoltarget_optionsr1kwargsfiles __class__rrr>^s    zLibArchiveFileSystem.__init__ccsJ|jdt|j|jd }|VWddS1swYdS)Nr)r1)rErr6r1)rGarcrrr _open_archives  "z"LibArchiveFileSystem._open_archivecst|dS)N/)r=_strip_protocollstrip)clspathrMrrrRsz$LibArchiveFileSystem._strip_protocolc sdddddddd|jdurdSi|_g}|E}|D]:js&js&q|jd d |tjDfd d D}jrFd nd |d<|j||j|d<qWdn1sbwY|jdd ||DdS)NpathnamesizectimerHuidgidmtime)namerWcreatedrHrYrZr[cSi|] }||dddqSr directory)r\rWtyper.0dirnamerrr  z2LibArchiveFileSystem._get_dirs..csi|] }|t|qSr)getattr)rckeyentryfieldsrrresr`rrar\cSr^r_rrbrrrrerf) rFrPisdirisfileupdate _all_dirnamessetr\append)rG list_namesrOfrrir _get_dirss@    zLibArchiveFileSystem._get_dirsrbTc Ks||}|dkr tt}|(}|D]} | j|krq| jdkr$n| | jD]} | }ntqWdn1s=wYt|||dS)Nrur)fsrUdata) rRNotImplementedErrorbytesrPrVrW get_blocksr?r ) rGrUrHr1 autocommit cache_optionsrKrwrOrjblockrrr_opens$    zLibArchiveFileSystem._open)ruNTN)__name__ __module__ __qualname____doc__ root_markerr:cachabler r>rrP classmethodrRrtr~ __classcell__rrrMrr7Hs*' ,r7) contextlibrctypesrrrrrrr r'libarchive.ffir"fsspecr fsspec.archiver fsspec.implementations.memoryr fsspec.utilsr r$ c_archive_p check_intrhasattrr% page_sizer6r7rrrrs $      &