o ^i@s\ddlZddlZddlZddlZddlmZmZddlmZd ddZ GdddeeZ dS) N)AsyncFileSystem running_async)ChainedFileSystemcstfdd}|S)a 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. semaphore : asyncio.Semaphore, optional A semaphore to limit concurrent calls. Returns ------- coroutine An awaitable version of the function. c szr.4IdHtjg|Ri|IdHWdIdHS1IdHs)wYtjg|Ri|IdHSN)asyncio to_thread)argskwargsfunc semaphore^/var/www/html/RAG/RAG_venv/lib/python3.10/site-packages/fsspec/implementations/asyn_wrapper.pywrappers ,zasync_wrapper..wrapper) functoolswraps)r objr rrr r async_wrapper srcsVeZdZdZdZdZ      d 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. ) asyncwrapperrFNc sd|durt}tjdd|i||dur||_n tj|fi||_|jj|_||_|dS)N asynchronousr) rsuper__init__sync_fsfsspec filesystemprotocolr _wrap_all_sync_methods)selffsrtarget_protocoltarget_optionsr max_concurrent_tasksr  __class__rrr8s   zAsyncFileSystemWrapper.__init__cCsd|jjS)Nasync_)rfsid)rrrrr'MszAsyncFileSystemWrapper.fsidcCsdh}t|jD]9}|ds||vrqt|j|}t|tr!qt|j|}t|rAt |sAt |||j d}t |d||qdS)zg Wrap all synchronous methods of the underlying filesystem with asynchronous versions. open_)rr N) dirr startswithinspectgetattr_static isinstancepropertygetattrcallableiscoroutinefunctionrr setattr)rexcluded_methods method_nameattrmethod async_methodrrrrQs  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)rr r r)r% sync_fs_classrrruszSAsyncFileSystemWrapper.wrap_class..GeneratedAsyncFileSystemWrapper.__init__)__name__ __module__ __qualname__r __classcell__rr9r$rGeneratedAsyncFileSystemWrappertsr?AsyncWrapper)r:)clsr9r?rr>r wrap_classcs z!AsyncFileSystemWrapper.wrap_class)NNNNNN) r:r;r<__doc__rcachablerr/r'r classmethodrCr=rrr$rr(s   r)NN) rrr,r fsspec.asynrrchainedrrrrrrrs