check if object is a generator

[ttmetro]

How do I check if an object is a coroutine (generator) in micropython?

Code: Select all

from inspect import iscoroutine

is not available.

This appears to work, but (???):

Code: Select all

def iscoroutine(obj):
    return str(type(obj)) == "<class 'generator'>"

[pythoncoder]

Code: Select all

type_gen = type((lambda: (yield))())

Test with

Code: Select all

isinstance(callback, type_gen)

It's sometimes also necessary to test for a generator function:

Code: Select all

type_gen = type((lambda: (yield))())  # Generator type
type_genf = type((lambda: (yield)))  # Generator function

def foo():
    for x in range(5):
        yield x

isinstance(foo, type_gen)  # False
isinstance(foo, type_genf)  # True

foo_gen = foo()
isinstance(foo_gen, type_gen)  # True
isinstance(foo_gen, type_genf)  # False

[ttmetro]

Thanks, works perfect!

[pythoncoder]

I should acknowledge my debt to @pfalcon here.