3

asyncio.coroutineをラップして完了までにかかった時間を返すデコレータ関数を作成しようとしました。以下のレシピには、期待どおりに機能するコードが含まれています。私の唯一の問題は、@functools.wraps. 元のコルーチンの名前を保持するには? のソースを確認しましたasyncio.

import asyncio
import functools
import random
import time

MULTIPLIER = 5

def time_resulted(coro):
    @functools.wraps(coro)
    @asyncio.coroutine
    def wrapper(*args, **kargs):
        time_before = time.time()
        result = yield from coro(*args, **kargs)
        if result is not None:
            raise TypeError('time resulted coroutine can '
                'only return None')
        return time_before, time.time()
    print('= wrapper.__name__: {!r} ='.format(wrapper.__name__))
    return wrapper

@time_resulted
@asyncio.coroutine
def random_sleep():
    sleep_time = random.random() * MULTIPLIER
    print('{} -> {}'.format(time.time(), sleep_time))
    yield from asyncio.sleep(sleep_time)

if __name__ == '__main__':
    loop = asyncio.get_event_loop()
    tasks = [asyncio.Task(random_sleep()) for i in range(5)]
    loop.run_until_complete(asyncio.wait(tasks))
    loop.close()
    for task in tasks:
        print(task, task.result()[1] - task.result()[0])
    print('= random_sleep.__name__: {!r} ='.format(
        random_sleep.__name__))
    print('= random_sleep().__name__: {!r} ='.format(
        random_sleep().__name__))

結果：

= wrapper.__name__: 'random_sleep' =
1397226479.00875 -> 4.261069174838891
1397226479.00875 -> 0.6596335046471768
1397226479.00875 -> 3.83421163259601
1397226479.00875 -> 2.5514027672929713
1397226479.00875 -> 4.497471439365472
Task(<wrapper>)<result=(1397226479.00875, 1397226483.274884)> 4.266134023666382
Task(<wrapper>)<result=(1397226479.00875, 1397226479.6697)> 0.6609499454498291
Task(<wrapper>)<result=(1397226479.00875, 1397226482.844265)> 3.835515022277832
Task(<wrapper>)<result=(1397226479.00875, 1397226481.562422)> 2.5536720752716064
Task(<wrapper>)<result=(1397226479.00875, 1397226483.51523)> 4.506479978561401
= random_sleep.__name__: 'random_sleep' =
= random_sleep().__name__: 'wrapper' =

ご覧のとおりrandom_sleep()、別の名前のジェネレーター オブジェクトを返します。装飾されたコルーチンの名前を保持したいと思います。これが特定の問題であるかどうかはわかりasyncio.coroutinesません。デコレータの順序を変えてコードも試してみましたが、すべて同じ結果になりました。私がコメントすれば@functools.wraps(coro)、私が期待したとおりにrandom_sleep.__name__なります。wrapper

編集: この問題を Python Issue Tracker に投稿し、R. David Murray から次の回答を受け取りました。随分前。"

4

2 に答える 2

2

問題は、functools.wraps変化するだけwrapper.__name__で、wrapper().__name__とどまることwrapperです。__name__読み取り専用のジェネレータ属性です。exec適切な名前を設定するために使用できます：

import asyncio
import functools
import uuid
from textwrap import dedent

def wrap_coroutine(coro, name_prefix='__' + uuid.uuid4().hex):
    """Like functools.wraps but preserves coroutine names."""
    # attribute __name__ is not writable for a generator, set it dynamically
    namespace = {
        # use name_prefix to avoid an accidental name conflict
        name_prefix + 'coro': coro,
        name_prefix + 'functools': functools,
        name_prefix + 'asyncio': asyncio,
    }
    exec(dedent('''
        def {0}decorator({0}wrapper_coro):
            @{0}functools.wraps({0}coro)
            @{0}asyncio.coroutine
            def {wrapper_name}(*{0}args, **{0}kwargs):
                {0}result = yield from {0}wrapper_coro(*{0}args, **{0}kwargs)
                return {0}result
            return {wrapper_name}
        ''').format(name_prefix, wrapper_name=coro.__name__), namespace)
    return namespace[name_prefix + 'decorator']

使用法：

def time_resulted(coro):
    @wrap_coroutine(coro)
    def wrapper(*args, **kargs):
        # ...
    return wrapper

それは機能しますが、おそらく を使用するよりも良い方法がありexec()ます。

于 2014-05-04T06:08:47.153 に答える
0

この質問が出されてから、コルーチンの名前を変更できるようになりました。__qualname__これは( ではなく)設定することによって行われます__name__:

async def my_coro(): pass

c = my_coro()
print(repr(c))
# <coroutine object my_coro at 0x7ff8a7d52bc0>

c.__qualname__ = 'flimflam'
print(repr(c))
# <coroutine object flimflam at 0x7ff8a7d52bc0>

import asyncio
print(repr(asyncio.ensure_future(c)))
# <Task pending name='Task-737' coro=<flimflam() running at <ipython-input>:1>>

__qualname__コルーチン オブジェクトでのの使用法は、CPython ソース__repr__で定義されています。

于 2021-05-19T19:00:06.610 に答える