a h @sNddlZddlZddlZddlZddlmZmZdddZGdddeZdS)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. cstjg|Ri|IdHSN)asyncioZ to_thread)argskwargsfuncQ/var/www/auris/lib/python3.9/site-packages/fsspec/implementations/asyn_wrapper.pywrapperszasync_wrapper..wrapper) functoolswraps)r objr r rr async_wrapper srcsJeZdZdZdZdZd fdd ZeddZd 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. )Z asyncwrapperrFNc s^|durt}tjfd|i||dur4||_ntj|fi||_|jj|_|dS)N asynchronous)rsuper__init__sync_fsfsspec filesystemprotocol_wrap_all_sync_methods)selffsrZtarget_protocolZtarget_optionsr __class__r r r1s zAsyncFileSystemWrapper.__init__cCsd|jjS)NZasync_)rfsid)rr r r rCszAsyncFileSystemWrapper.fsidcCsdh}t|jD]n}|ds||vr(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 isinstancepropertygetattrcallableiscoroutinefunctionrsetattr)rZexcluded_methods method_nameattrmethodZ async_methodr r r rGs   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|i|}t|dSr)rr)rrrr)r sync_fs_classr r rkszSAsyncFileSystemWrapper.wrap_class..GeneratedAsyncFileSystemWrapper.__init__)__name__ __module__ __qualname__r __classcell__r r.rr GeneratedAsyncFileSystemWrapperjsr4ZAsyncZWrapper)r/)clsr.r4r r3r wrap_classYs z!AsyncFileSystemWrapper.wrap_class)NNNN) r/r0r1__doc__rZcachablerr&rr classmethodr6r2r r rr r!s  r)N) rr r#rZ fsspec.asynrrrrr r r r s