o wi0 @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. cs tjg|Ri|IdHSN)asyncio to_thread)argskwargsfunc`/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/fsspec/implementations/asyn_wrapper.pywrapperszasync_wrapper..wrapper) functoolswraps)r objr 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. cs4tj|i|d|_||_|jj|_|dS)NT)super__init__ asynchronoussync_fsprotocol_wrap_all_sync_methods)selfrrr __class__r r r-s   zAsyncFileSystemWrapper.__init__cCsd|jjS)Nasync_)rfsid)rr r r r4szAsyncFileSystemWrapper.fsidcCsvt|jD]3}|dr qt|j|}t|trqt|j|}t|r8t |s8t ||d}t |d||qdS)zg Wrap all synchronous methods of the underlying filesystem with asynchronous versions. _)rN) dirr startswithinspectgetattr_static isinstancepropertygetattrcallableriscoroutinefunctionrsetattr)r method_nameattrmethod 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|i|}t|dSr)rr)rrrr)r sync_fs_classr r r[szSAsyncFileSystemWrapper.wrap_class..GeneratedAsyncFileSystemWrapper.__init__)__name__ __module__ __qualname__r __classcell__r r,rr GeneratedAsyncFileSystemWrapperZsr2AsyncWrapper)r-)clsr,r2r r1r wrap_classIs z!AsyncFileSystemWrapper.wrap_class) r-r.r/__doc__rr#rr classmethodr6r0r r rr r s  rr)rr r fsspec.asynrrrr r r r s