U h @sFddlZddlZddlZddlmZmZdddZGdddeZdS)N)AsyncFileSystem running_asynccstfdd}|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. cstjf||IdHSN)asyncioZ to_thread)argskwargsfuncU/var/www/html/venv/lib/python3.8/site-packages/fsspec/implementations/asyn_wrapper.pywrapperszasync_wrapper..wrapper) functoolswraps)r objr r rr async_wrappersrcsNeZdZdZdZdZddfdd Zedd Zd 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. rFN) asynchronouscs@|dkrt}tj|d|i|||_|jj|_|dS)Nr)rsuper__init__sync_fsprotocol_wrap_all_sync_methods)selffsrrr __class__r r r0s  zAsyncFileSystemWrapper.__init__cCsd|jjS)NZasync_)rfsid)rr r r r8szAsyncFileSystemWrapper.fsidcCsdh}t|jD]n}|ds||kr(qt|j|}t|trBqt|j|}t|rt |st ||d}t |d||qdS)zg Wrap all synchronous methods of the underlying filesystem with asynchronous versions. open_)rN) dirr startswithinspectgetattr_static isinstancepropertygetattrcallableriscoroutinefunctionrsetattr)rZexcluded_methods method_nameattrmethodZ async_methodr r r r<s   z-AsyncFileSystemWrapper._wrap_all_sync_methodscs(Gfddd|}djd|_|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. cseZdZfddZZS)zJAsyncFileSystemWrapper.wrap_class..GeneratedAsyncFileSystemWrappercs||}t|dSr)rr)rrrr)r sync_fs_classr r r`s zSAsyncFileSystemWrapper.wrap_class..GeneratedAsyncFileSystemWrapper.__init__)__name__ __module__ __qualname__r __classcell__r r,rr GeneratedAsyncFileSystemWrapper_sr2ZAsyncZWrapper)r-)clsr,r2r r1r wrap_classNs z!AsyncFileSystemWrapper.wrap_class) r-r.r/__doc__rZcachablerr$rr classmethodr4r0r r rr r s  r)N)rr r!Z fsspec.asynrrrrr r r r s