コード例 #1
0
ファイル: generator.py プロジェクト: Downtownapp/aiostripe
    def __wrap(self, func, args, kwargs):
        async def inner():
            try:
                value = await func(*args, **kwargs)
            except:
                exc = to_exception(*sys.exc_info())
            else:
                exc = StopAsyncIteration(value)

            self.__finished = True
            await self.__queue_yield.put((None, exc))

        return update_wrapper(inner, func)
コード例 #2
0
ファイル: generator.py プロジェクト: Downtownapp/aiostripe
    def inner(*args, **kwargs):
        def async_yield(*args):
            return gen._agi_yield(*args)

        def async_yield_from(*args):
            return gen._agi_yield_from(*args)

        ns = uniondict(builtins.__dict__, func.__globals__)
        ns['async_yield'] = async_yield
        ns['async_yield_from'] = async_yield_from
        ns['__builtins__'] = builtins.__dict__

        hooked_func = types.FunctionType(func.__code__, ns, func.__name__, func.__defaults__, func.__closure__)
        hooked_func = update_wrapper(hooked_func, func)

        gen = asyncgenerator(hooked_func, args, kwargs)

        return gen