Decorator to convert Python generator functions to lists or dicts. Also supports asynchronous functions.

listify.py

from functools import wraps, partial
import inspect


def _ify(func, factory=list):
    if inspect.isasyncgenfunction(func):
        @wraps(func)
        async def new_func(*args, **kwargs):
            return factory([x async for x in func(*args, **kwargs)])
    else:
        @wraps(func)
        def new_func(*args, **kwargs):
            return factory(func(*args, **kwargs))
    return new_func


listify = partial(_ify, factory=list)
dictify = partial(_ify, factory=dict)

test_listify.py

from listify import listify, dictify

def test_listify():
    @listify
    def func():
        yield from range(10)
    assert func() == list(range(10))

def test_dictify():
    @dictify
    def func():
        for i in range(10):
            yield i, i+1
    assert func() == dict(zip(range(10), range(1, 11)))