web-dev-qa-db-ja.com

Python 3.4の `async for`

Python 3.5 async forステートメントをPython 3.4コードに変換する方法はありますか？

PEP 0492 はasync forと言っています

async for TARGET in ITER:
    BLOCK
else:
    BLOCK2

と同等です

iter = (ITER)
iter = type(iter).__aiter__(iter)
running = True
while running:
    try:
        TARGET = await type(iter).__anext__(iter)
    except StopAsyncIteration:
        running = False
    else:
        BLOCK
else:
    BLOCK2

しかし、__aiter__はPython 3.4に存在しません

pythonasynchronousasync-awaitpython-3.4python-3.5
19
Marco Sulla

いいえ、ありません。async/await（__aiter__など）はpython3.5。py3.4で最も近いものは asyncio.gather （すべてのタスクを一度に/並行して実行し、すべてが完了するまで待つことができます）または結果を asyncio.Queue （async forと同じように順次）にプッシュできます。編集：async forの最後の例を参照してください。質問で説明されているように、代替。

これは、asyncio.gatherのala python docsの例です。

import asyncio

@asyncio.coroutine
def task(id):
    print("task: {}".format(id))
    yield from asyncio.sleep(random.uniform(1, 3))
    return id

tasks = [
    task("A"),
    task("B"),
    task("C")
]
loop = asyncio.get_event_loop()
results = loop.run_until_complete(asyncio.gather(*tasks))
loop.close()
print(results)

出力：

task: B
task: A
task: C
['A', 'B', 'C']

これがasyncio.Queue用のものです：

import asyncio

@asyncio.coroutine
def produce(queue, n):
    for x in range(n):
        print('producing {}/{}'.format(x, n))
        # todo: do something more useful than sleeping :)
        yield from asyncio.sleep(random.random())
        yield from queue.put(str(x))


@asyncio.coroutine
def consume(queue):
    while True:
        item = yield from queue.get()
        print('consuming {}...'.format(item))
        # todo: do something more useful than sleeping :)
        yield from asyncio.sleep(random.random())
        queue.task_done()


@asyncio.coroutine
def run(n):
    queue = asyncio.Queue()
    # schedule the consumer
    consumer = asyncio.ensure_future(consume(queue))
    # run the producer and wait for completion
    yield from produce(queue, n)
    # wait until the consumer has processed all items
    yield from queue.join()
    # the consumer is still awaiting for an item, cancel it
    consumer.cancel()


loop = asyncio.get_event_loop()
loop.run_until_complete(run(10))
loop.close()

編集：質問で説明されているasync for代替：

import asyncio
import random

class StopAsyncIteration(Exception):
    """"""

class MyCounter:
    def __init__(self, count):
        self.count = count

    def __aiter__(self):
        return self

    @asyncio.coroutine
    def __anext__(self):
        if not self.count:
            raise StopAsyncIteration

        return (yield from self.do_something())

    @asyncio.coroutine
    def do_something(self):
        yield from asyncio.sleep(random.uniform(0, 1))
        self.count -= 1
        return self.count

@asyncio.coroutine
def getNumbers():
    i = MyCounter(10).__aiter__()
    while True:
        try:
            row = yield from i.__anext__()
        except StopAsyncIteration:
            break
        else:
            print(row)

loop = asyncio.get_event_loop()
loop.run_until_complete(getNumbers())
loop.close()

これは、__aiter__と__anext__の両方を削除し、do_somethingメソッド自体で停止例外を発生させるか、完了時にセンチネル結果を返すことで簡略化できることに注意してください（通常、None、""、-1などの無効な値）

10
nitely