U
|
Z ±d# ã @ sv d dl Z d dlZd dlZd dlZd dlZddlmZ ddlmZ ddlm Z e
|
d¡Z e j dd d
|
G d d d Z dS ) é Né )Ú name_asyncgené )Ú_run)Ú_coreztrio.async_generator_errorsFT)ÚeqÚslotsc @ sR e Zd ZejejdZejedZ ejddZ
|
dd Z dd Z dd Z d
|
d Zd S ) ÚAsyncGenerators)ÚfactoryF)Úinitc sF fdd}fdd fdd}t ¡ _t j||d d S )Nc sB t tjdr j | ¡ n$d| jjd< jjd k r> j | ¡ d S )NZtaskTú@trio_foreign_asyncgen) Úhasattrr ZGLOBAL_RUN_CONTEXTÚaliveÚaddÚag_frameÚf_localsÚ
|
prev_hooksÚ firstiter)Úagen©Úself© úLd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\trio/_core/_asyncgens.pyr " s
|
z0AsyncGenerators.install_hooks.<locals>.firstiterc sF z j j| |d| dd W n tk
|
r@ j | ¡ Y nX d S )Nzclose asyncgen z (abandoned))Úname)Zspawn_system_taskÚ _finalize_oneÚ RuntimeErrorÚtrailing_needs_finalizer )r Ú agen_name)Úrunnerr r r Úfinalize_in_trio_context0 s
|
ü
|
z?AsyncGenerators.install_hooks.<locals>.finalize_in_trio_contextc sÀ t | }z| jj d¡ }W n tk
|
r4 d}Y nX |rfj | |¡ tjd|dt d| d nVj
|
j d k rj
|
| ¡ n<| ¡ }z| d ¡ W n tk
|
rª Y nX td|dd S ) Nr TzAsync generator z§ was garbage collected before it had been exhausted. Surround its use in 'async with aclosing(...):' to ensure that it gets cleaned up as soon as you're done using it.r )Ú
|
stacklevelÚsourcezNon-Trio async generator z~ awaited something during finalization; install a finalization hook to support this, or wrap it in 'async with aclosing(...):')r r r ÚgetÚAttributeErrorÚ entry_queueÚ run_sync_soonÚwarningsÚwarnÚResourceWarningr Ú finalizerÚacloseÚsendÚ StopIterationr )r r Zis_oursÚcloser©r r r r r r) ? s6
|
ÿ
|
ù
|
ÿz0AsyncGenerators.install_hooks.<locals>.finalizer)r r) )ÚsysÚget_asyncgen_hooksr Úset_asyncgen_hooks)r r r r) r r. r Ú install_hooks! s
|
2
|
z AsyncGenerators.install_hooksc à s¨ t ¡ |jkstt|jdks$tt| j| _|j |j
|
|j¡ t dd ¡I d H | j | j ¡ | j ¡ | jr¤| j}t | _|D ]}| |t|¡I d H qqnd S )Nr c S s t jjS ©N)r ZAbortZFAILED)Ú_r r r Ú<lambda>
ó z4AsyncGenerators.finalize_remaining.<locals>.<lambda>)r Z current_taskZ init_taskÚAssertionErrorÚlenZtasksÚsetr r$ r% Z
|
rescheduleZwait_task_rescheduledÚupdater Úclearr r )r r Úbatchr r r r Úfinalize_remainingt s ÿ
|
|
"z"AsyncGenerators.finalize_remainingc C s t j| j d S r3 )r/ r1 r r r r r Úclose° s zAsyncGenerators.closec à sX z2t jdd }| ¡ | ¡ I d H W 5 Q R X W n tk
|
rR t d|¡ Y nX d S )NT)ZshieldzÅException ignored during finalization of async generator %r -- surround your use of the generator in 'async with aclosing(...):' to raise exceptions like this in the context where they're generated)r Z CancelScopeÚcancelr* Ú BaseExceptionÚASYNCGEN_LOGGERÚ exception)r r r Z cancel_scoper r r r ³ s üz AsyncGenerators._finalize_oneN)Ú__name__Ú
|
__module__Ú __qualname__ÚattrÚibÚweakrefÚWeakSetr r9 r r r2 r= r> r r r r r r s S<r )rF Úloggingr/ r& rH Z_utilr Ú r r Ú getLoggerrA Úsr r r r r Ú<module> s
|
|
