d4UddlmZddlZddlZddlZddlmZddlmZ ddl m Z n#e $r ej Z YnwxYwe Z e ZejZded<dZd Zdd Zddd ZdZddZGddZGddZdZdZdS)) annotationsN)MutableMapping)Any)SerializableLockz#MutableMapping[Any, threading.Lock] _FILE_LOCKSc~ t|}n-#t$r tjx}t|<YnwxYw|SN)rKeyError threadingLock)keylocks 5lib/python3.11/site-packages/xarray/backends/locks.py_get_threaded_lockrsP33 333"+."2"22{33 Ks '::c*~tjSr )multiprocessingr )r s r_get_multiprocessing_lockr!s   ! !!c|tS|dkrtS|dkrtS|dkr ddlm}n#t$rd}YnwxYw|St |)zReturns an appropriate function for creating resource locks. Parameters ---------- scheduler : str or None Dask scheduler being used. See Also -------- dask.utils.get_scheduler_lock Nthreadedr distributedr)r )rrdask.distributedr ImportErrorr ) schedulerDistributedLocks r_get_lock_makerr(s!! j !! ' ' '(( m # # # @ @ @ @ @ @ @ # # #"OOO #y!!!s2 AAreturn str | Nonec ddl}ddlm}|||}n#t$rYdSwxYw ddlm}t |j|rdSn#ttf$rYnwxYw ||j j urdSn#t$rYnwxYwdS)zDetermine the dask scheduler that is being used. None is returned if no dask scheduler is active. See Also -------- dask.base.get_scheduler rN) get_scheduler)Clientrrr) dask dask.baser rrr! isinstance__self__AttributeErrorrget)r' collectionr"r actual_getr!s r_get_schedulerr*Gs  ++++++"]3 33 tt ++++++ j)6 2 2 ! = !  (       -1 1 1$$ 2       :s/ ''A AA!A22 A?>A?cRt}t|}||S)aGet a scheduler appropriate lock for writing to the given resource. Parameters ---------- key : str Name of the resource for which to acquire a lock. Typically a filename. Returns ------- Lock object that can be used like a threading.Lock object. )r*r)r r lock_makers rget_write_lockr-ms*  I ++J :c??rTcX|r|S||S)zAcquire a lock, possibly in a non-blocking fashion. Includes backwards compatibility hacks for old versions of Python, dask and dask-distributed. acquire)rblockings rr0r0~s.  &||~~||H%%%rc>eZdZdZdZd dZdZdZdZdZ d Z d S) CombinedLockzA combination of multiple locks. Like a locked door, a CombinedLock is locked if any of its constituent locks are locked. cHtt||_dSr )tuplesetlocks)selfr7s r__init__zCombinedLock.__init__s3u::&& rTcDtfd|jDS)Nc3:K|]}t|VdS))r1Nr/).0rr1s r z'CombinedLock.acquire..s0KK74(333KKKKKKr)allr7r8r1s `rr0zCombinedLock.acquires(KKKK KKKKKKrcB|jD]}|dSr )r7releaser8rs rrAzCombinedLock.releases,J  D LLNNNN  rcB|jD]}|dSr )r7 __enter__rBs rrDzCombinedLock.__enter__s0J  D NN      rc.|jD] }|j| dSr )r7__exit__)r8argsrs rrFzCombinedLock.__exit__s/J ! !D DM4  ! !rc>td|jDS)Nc3$K|] }|jV dSr )locked)r<rs rr=z&CombinedLock.locked..s$6644;666666r)anyr7r8s rrJzCombinedLock.lockeds!664:666666rc2dt|jdS)Nz CombinedLock())listr7rLs r__repr__zCombinedLock.__repr__s4tDJ//4444rNT) __name__ __module__ __qualname____doc__r9r0rArDrFrJrPrrr3r3s '''LLLL!!!77755555rr3c2eZdZdZd dZdZdZdZdZdS) DummyLockz;DummyLock provides the lock API without any actual locking.TcdSr rVr?s rr0zDummyLock.acquire rcdSr rVrLs rrAzDummyLock.releaserZrcdSr rVrLs rrDzDummyLock.__enter__rZrcdSr rV)r8rGs rrFzDummyLock.__exit__rZrcdS)NFrVrLs rrJzDummyLock.lockedsurNrQ) rRrSrTrUr0rArDrFrJrVrrrXrXsjEE             rrXcg}|D]I}t|tr||j2|||Jt |}|dkrt|S|dkr|dSt S)z/Combine a sequence of locks into a single lock.Nr)r$r3extendr7appendlenrX)r7 all_locksr num_lockss r combine_locksrfsI## dL ) ) #   TZ ( ( ( (     T " " "II1}}I&&& a|{{rc.||durtS|S)z'Ensure that the given object is a lock.NF)rX)rs r ensure_lockrhs |tu}}{{ Krr )NN)rrrQ) __future__rrr weakrefcollections.abcrtypingr dask.utilsrrr HDF5_LOCK NETCDFC_LOCKWeakValueDictionaryr__annotations__rrrr*r-r0r3rXrfrhrVrrrrs"""""""******&+++++++&&& ~&     !! 4O73N3P3P PPPP""""""">#####L"&&&&$55555555@&$s ( 77