o i'@snUddlZddlmZmZGdddeZejdddZejeeed<deefdd ZGd d d Z dS) N)Optional NamedTuplec@s.eZdZUdZeed<eed<eeed<dS)ExecutionContextz Context information about the currently executing node. Attributes: node_id: The ID of the currently executing node list_index: The index in a list being processed (for operations on batches/lists) prompt_idnode_id list_indexN)__name__ __module__ __qualname____doc__str__annotations__rintrr3/mnt/c/Users/fbmor/ComfyUI/comfy_execution/utils.pyrs rcurrent_executing_context)defaultreturncCs tdSN)rgetrrrrget_executing_contexts rc@s<eZdZdZd dededeefddZdd Zd d Z dS) CurrentNodeContexta> Context manager for setting the current executing node context. Sets the current_executing_context on enter and resets it on exit. Example: with CurrentNodeContext(node_id="123", list_index=0): # Code that should run with the current node context set process_image() NrrrcCst|||d|_d|_dS)N)rrr)rcontexttoken)selfrrrrrr__init__ s  zCurrentNodeContext.__init__cCst|j|_|Sr)rsetrr)rrrr __enter__(szCurrentNodeContext.__enter__cCs|jdur t|jdSdSr)rrreset)rexc_typeexc_valexc_tbrrr__exit__,s zCurrentNodeContext.__exit__r) rr r r r rrrrr"rrrrrs   r) contextvarstypingrrr ContextVarrr rrrrrrs