o
    h
                     @   sF   d dl Z d dlZd dlZd dlmZmZ dddZG dd deZdS )    N)AsyncFileSystemrunning_asyncc                    s   t   fdd}|S )aF  
    Wraps a synchronous function to make it awaitable.

    Parameters
    ----------
    func : callable
        The synchronous function to wrap.
    obj : object, optional
        The instance to bind the function to, if applicable.

    Returns
    -------
    coroutine
        An awaitable version of the function.
    c                     s    t j g| R i |I d H S N)asyncio	to_thread)argskwargsfunc w/var/www/html/construction_image-detection-poc/venv/lib/python3.10/site-packages/fsspec/implementations/asyn_wrapper.pywrapper   s   zasync_wrapper.<locals>.wrapper)	functoolswraps)r
   objr   r   r	   r   async_wrapper   s   r   c                       sN   e Zd ZdZdZdZdd fdd
Zedd	 Zd
d Z	e
dd Z  ZS )AsyncFileSystemWrapperaX  
    A wrapper class to convert a synchronous filesystem into an asynchronous one.

    This class takes an existing synchronous filesystem implementation and wraps all
    its methods to provide an asynchronous interface.

    Parameters
    ----------
    sync_fs : AbstractFileSystem
        The synchronous filesystem instance to wrap.
    r   FN)asynchronousc                   s@   |d u rt  }t j|d|i| || _| jj| _|   d S )Nr   )r   super__init__sync_fsprotocol_wrap_all_sync_methods)selffsr   r   r   	__class__r   r   r   0   s   
zAsyncFileSystemWrapper.__init__c                 C   s   d| j j S )Nasync_)r   fsid)r   r   r   r   r   8   s   zAsyncFileSystemWrapper.fsidc                 C   s   dh}t | jD ]7}|ds||v rqt| j|}t|tr!qt| j|}t|r?t	
|s?t|| d}t| d| | qdS )zg
        Wrap all synchronous methods of the underlying filesystem with asynchronous versions.
        open_)r   N)dirr   
startswithinspectgetattr_static
isinstancepropertygetattrcallabler   iscoroutinefunctionr   setattr)r   excluded_methodsmethod_nameattrmethodasync_methodr   r   r   r   <   s   
z-AsyncFileSystemWrapper._wrap_all_sync_methodsc                    s(   G  fddd| }d j  d|_ |S )a  
        Create a new class that can be used to instantiate an AsyncFileSystemWrapper
        with lazy instantiation of the underlying synchronous filesystem.

        Parameters
        ----------
        sync_fs_class : type
            The class of the synchronous filesystem to wrap.

        Returns
        -------
        type
            A new class that wraps the provided synchronous filesystem class.
        c                       s   e Zd Z fddZ  ZS )zJAsyncFileSystemWrapper.wrap_class.<locals>.GeneratedAsyncFileSystemWrapperc                    s   |i |}t  | d S r   )r   r   )r   r   r   r   )r   sync_fs_classr   r   r   `   s   zSAsyncFileSystemWrapper.wrap_class.<locals>.GeneratedAsyncFileSystemWrapper.__init__)__name__
__module____qualname__r   __classcell__r   r0   r   r   GeneratedAsyncFileSystemWrapper_   s    r6   AsyncWrapper)r1   )clsr0   r6   r   r5   r   
wrap_classN   s   z!AsyncFileSystemWrapper.wrap_class)r1   r2   r3   __doc__r   cachabler   r&   r   r   classmethodr:   r4   r   r   r   r   r       s    
r   r   )r   r   r#   fsspec.asynr   r   r   r   r   r   r   r   <module>   s    
