o US`cZ @slddlmZddlZddlmZddlmZddlmZddlmZ ej ddd Gd d d ej d ZdS)) defaultdictN)asynccontextmanager)_core)_util)EventF) DefaultDictSet)eqhashc@sPeZdZdZejddddZejeddZejdddZ e de fdd Z d S) SequenceraA convenience class for forcing code in different tasks to run in an explicit linear order. Instances of this class implement a ``__call__`` method which returns an async context manager. The idea is that you pass a sequence number to ``__call__`` to say where this block of code should go in the linear sequence. Block 0 starts immediately, and then block N doesn't start until block N-1 has finished. Example: An extremely elaborate way to print the numbers 0-5, in order:: async def worker1(seq): async with seq(0): print(0) async with seq(4): print(4) async def worker2(seq): async with seq(2): print(2) async with seq(5): print(5) async def worker3(seq): async with seq(1): print(1) async with seq(3): print(3) async def main(): seq = trio.testing.Sequencer() async with trio.open_nursery() as nursery: nursery.start_soon(worker1, seq) nursery.start_soon(worker2, seq) nursery.start_soon(worker3, seq) cCsttS)N)rrr r B/usr/local/lib/python3.10/dist-packages/trio/testing/_sequencer.py8szSequencer.F)factoryinit)defaultrpositionc Cs||jvr td||jrtd|j||dkrMz |j|IdHWntjyEd|_|j D]}| q:tdw|jrMtdzdVW|j|d dS|j|d w)Nz%Attempted to re-use sequence point {}zsequence broken!rTz+Sequencer wait cancelled -- sequence broken) _claimed RuntimeErrorformat_brokenadd_sequence_pointswaitrZ Cancelledvaluesset)selfreventr r r__call__=s(   *zSequencer.__call__N) __name__ __module__ __qualname____doc__attribrrrrrintr r r r rr s'r ) metaclass) collectionsrr%async_generatorrrrrtypingrr sFinalr r r r rs