从作为异步任务运行的函数获取值

Question

我正在尝试以下代码：

 import asyncio

@asyncio.coroutine
def func_normal():
        print("A")
        yield from asyncio.sleep(5)
        print("B")
        return 'saad'

@asyncio.coroutine
def func_infinite():
    i = 0
    while i<10:
        print("--"+str(i))
        i = i+1
    return('saad2')

loop = asyncio.get_event_loop()

tasks = [
    asyncio.async(func_normal()),
    asyncio.async(func_infinite())]

loop.run_until_complete(asyncio.wait(tasks))
loop.close()

我不知道如何从这些函数中获取变量值。我不能这样做：

 asyncio.async(a = func_infinite())

因为这会使它成为关键字参数。我该怎么做才能做到这一点？

原文由 Saad Aleem 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

协程按原样工作。只需使用 loop.run_until_complete() 的返回值并 调用 asyncio.gather() 来收集多个结果：

 #!/usr/bin/env python3
import asyncio

@asyncio.coroutine
def func_normal():
    print('A')
    yield from asyncio.sleep(5)
    print('B')
    return 'saad'

@asyncio.coroutine
def func_infinite():
    for i in range(10):
        print("--%d" % i)
    return 'saad2'

loop = asyncio.get_event_loop()
tasks = func_normal(), func_infinite()
a, b = loop.run_until_complete(asyncio.gather(*tasks))
print("func_normal()={a}, func_infinite()={b}".format(**vars()))
loop.close()

输出

--0
--1
--2
--3
--4
--5
--6
--7
--8
--9
A
B
func_normal()=saad, func_infinite()=saad2

原文由 jfs 发布，翻译遵循 CC BY-SA 3.0 许可协议

Answer 2

loop.run_until_complete 返回你传入的函数返回的值。因此，它将返回 asyncio.wait 的输出：

 import asyncio

@asyncio.coroutine
def func_normal():
    print("A")
    yield from asyncio.sleep(5)
    print("B")
    return 'saad'

@asyncio.coroutine
def func_infinite():
    i = 0
    while i<10:
        print("--"+str(i))
        i = i+1
    return('saad2')

loop = asyncio.get_event_loop()

tasks = [
    asyncio.async(func_normal()),
    asyncio.async(func_infinite())]

done, _ = loop.run_until_complete(asyncio.wait(tasks))
for fut in done:
    print("return value is {}".format(fut.result()))
loop.close()

输出：

 A
--0
--1
--2
--3
--4
--5
--6
--7
--8
--9
B
return value is saad2
return value is saad

您还可以直接从 tasks 数组访问结果：

 print(tasks[0].result())
print(tasks[1].result())

原文由 dano 发布，翻译遵循 CC BY-SA 3.0 许可协议