Python函数成为生成器

Question

我正在努力解决以下问题：

def function(param):
    if(param == 'notgen'):
        return 'some text'
    elif(param == 'gen'):
        for x in range(10):
            yield x
gen = function('notgen')
print(type(gen))
>>> <class 'generator'>

我想，这是预期的行为，但我找不到合适的文档。 我需要根据情况从一个函数中生成并返回值。 它甚至可行吗？

感谢大家的建议！