What I'm trying to achieve is something along the lines of spawning multiple parents and with each parent do some work and then spawn a few childs to check for other things and grab those results in the parent to do further work. I was also trying to make 2 different limits of spawn because the parent work can do more than the childs.
How would I accomplish this?
It works if I don't use limit2 but I would like to have the two limiters.
import trio
import asks
import time
import random
async def child(parent, i, sender, limit2):
async with limit2:
print('Parent {0}, Child {1}: started! Sleeping now...'.format(parent, i))
#await trio.sleep(random.randrange(0, 3))
print('Parent {0}, Child {1}: exiting!'.format(parent, i))
async with sender:
await sender.send('Parent {0}, Child {1}: exiting!'.format(parent, i))
async def parent(i, limit):
async with limit:
print('Parent {0}: started! Sleeping now...'.format(i))
#await trio.sleep(random.randrange(0, 3))
sender, receiver = trio.open_memory_channel(10)
limit2 = trio.CapacityLimiter(2)
async with trio.open_nursery() as nursery:
for j in range(10):
nursery.start_soon(child, i, j, sender, limit2)
async with receiver:
async for value in receiver:
print('Got value: {!r}'.format(value))
print('Parent {0}: exiting!'.format(i))
async def main():
limit = trio.CapacityLimiter(1)
async with trio.open_nursery() as nursery:
for i in range(1):
nursery.start_soon(parent, i, limit)
if __name__ == "__main__":
start_time = time.perf_counter()
trio.run(main)
duration = time.perf_counter() - start_time
print("Took {:.2f} seconds".format(duration))