U Cv f@sUddlmZddlZddlZddlZddlmZmZzddlm Z Wne k r`ej Z YnXe Z e Z eZded<ddZd d Zdd d Zdd dddZddZd ddZGdddZGdddZddZddZdS)!) annotationsN)AnyMutableMapping)SerializableLockz#MutableMapping[Any, threading.Lock] _FILE_LOCKScCs6z t|}Wn$tk r0t}t|<YnX|SN)rKeyError threadingLock)keylockr 9/tmp/pip-unpacked-wheel-h316xyqg/xarray/backends/locks.py_get_threaded_locks  rcCs ~tSr)multiprocessingr )r r r r_get_multiprocessing_lock srcCsf|dkr tS|dkrtS|dkr$tS|dkrZzddlm}Wntk rTd}YnX|St|dS)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) schedulerZDistributedLockr r r_get_lock_maker's  rz str | None)returnc Csz"ddl}ddlm}|||}Wntk r8YdSXz"ddlm}t|j|rZWdSWnttfk rtYnXz||j j krWdSWntk rYnXdS)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) daskZ dask.baserrrr isinstance__self__AttributeErrorrget)rZ collectionrrZ actual_getrr r r_get_schedulerFs$       r cCst}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 rZ lock_makerr r rget_write_lockls r!TcCs|r |S||SdS)zAcquire a lock, possibly in a non-blocking fashion. Includes backwards compatibility hacks for old versions of Python, dask and dask-distributed. Nacquire)r blockingr r rr#}sr#c@sJeZdZdZddZdddZddZd d Zd d Zd dZ ddZ dS) CombinedLockzA combination of multiple locks. Like a locked door, a CombinedLock is locked if any of its constituent locks are locked. cCstt||_dSr)tuplesetlocks)selfr(r r r__init__szCombinedLock.__init__Tcstfdd|jDS)Nc3s|]}t|dVqdS)r$Nr".0r r+r r sz'CombinedLock.acquire..)allr(r)r$r r+rr#szCombinedLock.acquirecCs|jD] }|qdSr)r(releaser)r r r rr1s zCombinedLock.releasecCs|jD] }|qdSr)r( __enter__r2r r rr3s zCombinedLock.__enter__cGs|jD]}|j|qdSr)r(__exit__)r)argsr r r rr4s zCombinedLock.__exit__cCstdd|jDS)Ncss|] }|jVqdSr)lockedr,r r rr.sz&CombinedLock.locked..)anyr(r)r r rr6szCombinedLock.lockedcCsdt|jdS)Nz CombinedLock())listr(r8r r r__repr__szCombinedLock.__repr__N)T) __name__ __module__ __qualname____doc__r*r#r1r3r4r6r;r r r rr%s r%c@s:eZdZdZdddZddZddZd d Zd d Zd S) DummyLockz;DummyLock provides the lock API without any actual locking.TcCsdSrr r0r r rr#szDummyLock.acquirecCsdSrr r8r r rr1szDummyLock.releasecCsdSrr r8r r rr3szDummyLock.__enter__cGsdSrr )r)r5r r rr4szDummyLock.__exit__cCsdS)NFr r8r r rr6szDummyLock.lockedN)T) r<r=r>r?r#r1r3r4r6r r r rr@s  r@cCsjg}|D].}t|tr$||jq|dk r||qt|}|dkrPt|S|dkr`|dStSdS)z/Combine a sequence of locks into a single lock.Nr)rr%extendr(appendlenr@)r(Z all_locksr Z num_locksr r r combine_lockss  rEcCs|dks|dkrtS|S)z'Ensure that the given object is a lock.NF)r@)r r r r ensure_locksrF)N)NN)T) __future__rrr weakreftypingrrZ dask.utilsrrr Z HDF5_LOCKZ NETCDFC_LOCKWeakValueDictionaryr__annotations__rrrr r!r#r%r@rErFr r r rs,   &