Aluode/PerceptionLabPortable
0
1o
2 ��Yi� � @ s� d dl mZ d dlmZmZmZmZmZmZm Z d dl3Z4d dl m Z d dl
mZ d dlmZ d dlmZ d dlmZ eeeeee�Ze � de jegee j�Zee d �Zed5d6e jfd d ��Z G d
d� de�Z dS )� )�contextmanager)� CFUNCTYPE�POINTER�c_int�7c_longlong�c_void_p�cast�create_string_bufferN)�8open_files)�AbstractArchiveFileSystem)�9MemoryFile)�DEFAULT_BLOCK_SIZE�read_set_seek_callback�10NO_OPEN_CB�allc # s� � t |�� t� t��� ��fdd�}�fdd�}t�|�}t|�}tr*tj}tj} nt j11� tj �}t j12�
tj �} t j13�||�� }14t|15|� t�|16d||| � t j17�|18�V W d � dS 1 s`w Y dS )z�Read an archive from a seekable file-like object.19 20 The `file` object must support the standard `readinto` and 'seek' methods.21 c s$ �� � �}t|tt��}�|d< |S )Nr )�readintor r r )� archive_p�contextZptrptr�length��bufZbuf_p�file� ��E:\DocsHouse\542 percep lab latest\PerceptionLab\PerceptionLab_Portable\python_embed\Lib\site-packages\fsspec/implementations/libarchive.py� read_func* s 22z custom_reader.<locals>.read_funcc s � � ||� � �� S )N)�seek�tell)r r �offset�whence)r r r � seek_func3 s z custom_reader.<locals>.seek_funcN)r r r �ffiZ
READ_CALLBACK�
SEEK_CALLBACK�new_apir Z NO_CLOSE_CB�23libarchive�readZ
OPEN_CALLBACKZVOID_CBZCLOSE_CALLBACKZnew_archive_readr Z read_openZ ArchiveRead) r � format_nameZ filter_name�24block_sizer r Zread_cbZseek_cbZopen_cbZclose_cbr r r r �
custom_reader! s" �25 2627"�r' c sn e Zd ZdZdZdZdZddddef� fdd� Ze d d28� �Z29e � fd d ��Z d
d� Z
ddd�Z� ZS )�LibArchiveFileSystemay Compressed archives as a file-system (read-only)30 31 Supports the following formats:32 tar, pax , cpio, ISO9660, zip, mtree, shar, ar, raw, xar, lha/lzh, rar33 Microsoft CAB, 7-Zip, WARC34 35 See the libarchive documentation for further restrictions.36 https://www.libarchive.org/37 38 Keeps file object open while instance lives. It only works in seekable39 file-like objects. In case the filesystem does not support this kind of40 file object, it is recommended to cache locally.41 42 This class is pickleable, but not necessarily thread-safe (depends on the43 platform). See libarchive documentation for details.44 � r# F�rNc s� t � j| fi |�� |dkrtd��t|t�r8t|fd|i|p i ��}t|�dkr4td|� d|� d���|d }|| _|�� | _ || _45d | _ d S )46aB 47 Parameters48 ----------49 fo: str or file-like50 Contains ZIP, and must exist. If a str, will fetch file using51 :meth:`~fsspec.open_files`, which must return one file exactly.52 mode: str53 Currently, only 'r' accepted54 target_protocol: str (optional)55 If ``fo`` is a string, this value can be used to override the56 FS protocol inferred from a URL57 target_options: dict (optional)58 Kwargs passed when instantiating the target FS, if ``fo`` is59 a string.60 r* z%Only read from archive files accepted�protocol� zPath "z(" did not resolve to exactly one file: "�"r N) �super�__init__�61ValueError�62isinstance�strr63 �lenZof� __enter__�for& � dir_cache)�selfr5 �modeZtarget_protocolZtarget_optionsr& �kwargs�files�� __class__r r r/ ^ s 64 �6566z LibArchiveFileSystem.__init__c c sJ � | j �d� t| j | jd�� }|V W d � d S 1 s w Y d S )Nr )r& )r5 r r' r& )r7 �arcr r r �
_open_archive� s67 � "�z"LibArchiveFileSystem._open_archivec s t � �|��d�S )N�/)r. �_strip_protocol�lstrip)�cls�pathr; r r r@ � s z$LibArchiveFileSystem._strip_protocolc s� dddddddd��| j d urd S i | _ g }| �� �E}|D ]:� � js&� js&q | j �d d68� | �t� j��D �� � �fd d69��D �}� jrFd nd
|d< |�� j� || j |d < q W d � n1 sbw Y | j �dd70� | �|�D �� d S )N�pathname�size�ctimer8 �uid�gid�mtime)�namerE �createdr8 rG rH rI c S � i | ] }||d dd��qS �r � directory)rJ rE �typer ��.0�dirnamer r r �71<dictcomp>� � ��z2LibArchiveFileSystem._get_dirs.<locals>.<dictcomp>c s i | ]72}|t � �| ��qS r )�getattr)rQ �key��entry�fieldsr r rS � s rN r rO rJ c S rL rM r rP r r r rS � rT ) r6 r> �isdir�isfile�updateZ
_all_dirnames�setrJ �append)r7 Z73list_namesr= �fr rW r � _get_dirs� s@ �74 7576 �� � ���z LibArchiveFileSystem._get_dirs�rbTc K s� | � |�}|dkr t�d}| �� �(}|D ] } | j|kr q| jdkr# n| �| j�D ]}77|78} nt�qW d � n1 s<w Y t| ||d�S )Nra � r )�fsrC �data)r@ �NotImplementedErrorr> rD rE Z79get_blocksr0 r ) r7 rC r8 r& Z80autocommitZ
cache_optionsr9 rd r= rX �blockr r r �_open� s$ 81 828384�� �zLibArchiveFileSystem._open)ra NTN)�__name__�85__module__� __qualname__�__doc__Z root_markerr+ Zcachabler
r/ r r> � classmethodr@ r` rg �
__classcell__r r r; r r( H s* �'86,�r( ) �87contextlibr �ctypesr r r r r r r r# Zlibarchive.ffir Zfsspecr88 Zfsspec.archiver Z fsspec.implementations.memoryr Z fsspec.utilsr
r! Z c_archive_pZ check_intr �hasattrr" Z page_sizer' r( r r r r �<module> s $89 �90&