U ,jg0 @sBddlZddlZddlZddlmZdddZGdddeZdS)N)AsyncFileSystemcstfdd}|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)argskwargsfuncG/tmp/pip-unpacked-wheel-ojcr9xjz/fsspec/implementations/asyn_wrapper.pywrapperszasync_wrapper..wrapper) functoolswraps)robjr r rr async_wrappersrcs@eZdZdZfddZeddZddZedd 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. cs0tj||d|_||_|jj|_|dS)NT)super__init__Z asynchronoussync_fsprotocol_wrap_all_sync_methods)selfrrr __class__r r r-s  zAsyncFileSystemWrapper.__init__cCsd|jjS)NZasync_)rfsid)rr r r r4szAsyncFileSystemWrapper.fsidcCsvt|jD]f}|drq t|j|}t|tr4q t|j|}t|r t |s t ||d}t |d||q dS)zg Wrap all synchronous methods of the underlying filesystem with asynchronous versions. _)rN) dirr startswithinspectgetattr_static isinstancepropertygetattrcallableriscoroutinefunctionrsetattr)r method_nameattrmethodZ async_methodr r r r8s    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 GeneratedAsyncFileSystemWrapperZsr.ZAsyncZWrapper)r))clsr(r.r r-r wrap_classIs z!AsyncFileSystemWrapper.wrap_class) r)r*r+__doc__rr rr classmethodr0r,r r rr r s  r)N)rr rZ fsspec.asynrrrr r r r s