a İaC@s$UddlZddlZddlZddlmZmZmZzddlmZWne yVej ZYn0zddl m Z Wne y~dZ Yn0eZ eZeZeeej fed<ddZdd Zeeee d Zdd d Zdeed ddZddZdddZGdddZGdddZddZddZdS) N)AnyMutableMappingOptional)SerializableLock)Lock _FILE_LOCKScCs4z t|}Wn"ty.t}t|<Yn0|SN)rKeyError threadingr)keylockr 4lib/python3.9/site-packages/xarray/backends/locks.py_get_threaded_locks   rcCs ~tSr)multiprocessingr)r r r r_get_multiprocessing_lock#sr)Nthreadedr distributedcCst|S)zReturns an appropriate function for creating resource locks. Parameters ---------- scheduler : str or None Dask scheduler being used. See Also -------- dask.utils.get_scheduler_lock ) _LOCK_MAKERS) schedulerr r r_get_lock_maker2s r)returnc Csz"ddl}ddlm}|||}Wnty6YdS0z"ddlm}t|j|rXWdSWnttfypYn0z||j j urWdSWntyYn0dS)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.baser ImportErrordask.distributedr isinstance__self__AttributeErrorrget)r Z collectionrrZ actual_getrr r r_get_schedulerAs$         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_lockgs r"TcCs8|r |Stdur*t|tr*|jddS||SdS)zAcquire a lock, possibly in a non-blocking fashion. Includes backwards compatibility hacks for old versions of Python, dask and dask-distributed. Nr)timeout)acquireDistributedLockr)r blockingr r rr$xs  r$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&N)r$.0r r-r r z'CombinedLock.acquire..)allr*r+r&r r-rr$szCombinedLock.acquirecCs|jD] }|qdSr)r*releaser+r r r rr4s zCombinedLock.releasecCs|jD] }|qdSr)r* __enter__r5r r rr6s zCombinedLock.__enter__cGs|jD]}|j|qdSr)r*__exit__)r+argsr r r rr7s zCombinedLock.__exit__cCstdd|jDS)Ncss|] }|jVqdSr)lockedr.r r rr0r1z&CombinedLock.locked..)anyr*r+r r rr9szCombinedLock.lockedcCsdt|jdS)Nz CombinedLock())listr*r;r r r__repr__szCombinedLock.__repr__N)T) __name__ __module__ __qualname____doc__r,r$r4r6r7r9r>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 r3r r rr$szDummyLock.acquirecCsdSrr r;r r rr4szDummyLock.releasecCsdSrr r;r r rr6szDummyLock.__enter__cGsdSrr )r+r8r r rr7szDummyLock.__exit__cCsdS)NFr r;r r rr9szDummyLock.lockedN)T) r?r@rArBr$r4r6r7r9r r r rrCs  rCcCsjg}|D].}t|tr$||jq|dur||qt|}|dkrPt|S|dkr`|dStSdS)z/Combine a sequence of locks into a single lock.Nr)rr'extendr*appendlenrC)r*Z all_locksr Z num_locksr r r combine_lockss  rHcCs|dus|durtS|S)z'Ensure that the given object is a lock.NF)rC)r r r r ensure_locksrI)N)NN)T)rr weakreftypingrrrZ dask.utilsrrrrr%Z HDF5_LOCKZ NETCDFC_LOCKWeakValueDictionaryr__annotations__rrrrstrr!r"r$r'rCrHrIr r r rs:      &