U
    ü‚ºef  ã                   @   sz   d dl Z d dlmZ d dlmZ d dlmZ dd„ Zzd dlm	Z	 W n0 e
k
rt   d dlmZ G d	d
„ d
eƒZ	Y nX dS )é    N)Úget_running_loop©Úwraps)ÚSynchronousOnlyOperationc                    s,   ‡ fdd„}t ˆ ƒr$ˆ }d‰ ||ƒS |S dS )z•
    Decorator to mark functions as async-unsafe. Someone trying to access
    the function while in an async context will get an error message.
    c                    s   t ˆ ƒ‡ ‡fdd„ƒ}|S )Nc                     s>   z
t ƒ  W n tk
r   Y nX tj d¡s4tˆƒ‚ˆ | |ŽS )NZDJANGO_ALLOW_ASYNC_UNSAFE)r   ÚRuntimeErrorÚosÚenvironÚgetr   )ÚargsÚkwargs)ÚfuncÚmessage© ú8/tmp/pip-unpacked-wheel-lctamlir/django/utils/asyncio.pyÚinner   s    
z.async_unsafe.<locals>.decorator.<locals>.innerr   )r   r   ©r   )r   r   Ú	decorator   s    zasync_unsafe.<locals>.decoratorzKYou cannot call this from an async context - use a thread or sync_to_async.N)Úcallable)r   r   r   r   r   r   Úasync_unsafe   s    ÿr   )Úaclosing)ÚAbstractAsyncContextManagerc                   @   s(   e Zd ZdZdd„ Zdd„ Zdd„ ZdS )	r   z°
        Async context manager for safely finalizing an asynchronously
        cleaned-up resource such as an async generator, calling its
        ``aclose()`` method.
        c                 C   s
   || _ d S ©N©Úthing)Úselfr   r   r   r   Ú__init__9   s    zaclosing.__init__c                 Ã   s   | j S r   r   )r   r   r   r   Ú
__aenter__<   s    zaclosing.__aenter__c                 Ç   s   | j  ¡ I d H  d S r   )r   Úaclose)r   Úexc_infor   r   r   Ú	__aexit__?   s    zaclosing.__aexit__N)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   r   r   r   r   r   r   r   2   s   r   )r   Zasyncior   Ú	functoolsr   Zdjango.core.exceptionsr   r   Ú
contextlibr   ÚImportErrorr   r   r   r   r   Ú<module>   s   "