U ڲg@sTddlZddlZddlZddlZddlmZmZmZGdddZGdddZ dS)N)AnyDictUnioncsPeZdZdZddddZddZeeddfd d Zedd d d Z Z S)_CVarzStorage utility for Local.N)returncCstd|_dS)Nz asgiref.local) contextvarsZ ContextVar_dataselfr 1/tmp/pip-unpacked-wheel-hay1n6u8/asgiref/local.py__init__ sz_CVar.__init__cCsB|ji}z ||WStk r<t|d|YnXdSNz object has no attribute )rgetKeyErrorAttributeErrorr keystorage_objectr r r __getattr__s   z_CVar.__getattr__)rvaluercs:|dkrt||S|ji}|||<|j|dS)Nr)super __setattr__rrset)r rrr __class__r r rs  z_CVar.__setattr__)rrcCs>|ji}||kr(||=|j|nt|d|dSr)rrrrrr r r __delattr__s  z_CVar.__delattr__) __name__ __module__ __qualname____doc__r rstrrrr __classcell__r r rr rs rcsPeZdZdZdeddddZejddZd d Z fd d Z d dZ Z S)LocalaLocal storage for async tasks. This is a namespace object (similar to `threading.local`) where data is also local to the current async task (if there is one). In async threads, local means in the same sense as the `contextvars` module - i.e. a value set in an async frame will be visible: - to other async code `await`-ed from this frame. - to tasks spawned using `asyncio` utilities (`create_task`, `wait_for`, `gather` and probably others). - to code scheduled in a sync thread using `sync_to_async` In "sync" threads (a thread with no async event loop running), the data is thread-local, but additionally shared with async code executed via the `async_to_sync` utility, which schedules async code in a new thread and copies context across to that thread. If `thread_critical` is True, then the local will only be visible per-thread, behaving exactly like `threading.local` if the thread is sync, and as `contextvars` if the thread is async. This allows genuinely thread-sensitive code (such as DB handles) to be kept stricly to their initial thread and disable the sharing across `sync_to_async` and `async_to_sync` wrapped calls. Unlike plain `contextvars` objects, this utility is threadsafe. FN)thread_criticalrcCs0||_t|_||r$t|_nt|_dSN)_thread_critical threadingRLock _thread_locklocal_storager)r r$r r r r Ds   zLocal.__init__c csp|jrRz tWntk r.|jVYqlXt|jdsFt|j_|jjVn|j|jVW5QRXdS)Ncvar) r&asyncioZget_running_loop RuntimeErrorr+hasattrrr,r)r r r r _lock_storageQs    zLocal._lock_storagec Cs*|}t||W5QRSQRXdSr%)r0getattrr rstorager r r rts zLocal.__getattr__c s:|dkrt||S|}t|||W5QRXdS)N)_localr+r&r))rrr0setattr)r rrr3rr r rxs zLocal.__setattr__c Cs"|}t||W5QRXdSr%)r0delattrr2r r r r~s zLocal.__delattr__)F) rrrr boolr contextlibcontextmanagerr0rrrr"r r rr r#(s  " r#) r-r8rr'typingrrrrr#r r r r s