Return or yield from a function that calls a generator?

Question

I have a generator generator and also a convenience method to it - generate_all.

def generator(some_list):   for i in some_list:     yield do_something(i)  def generate_all():   some_list = get_the_list()   return generator(some_list) # <-- Is this supposed to be return or yield? 

Should generate_all return or yield? I want the users of both methods to use it the same, i.e.

for x in generate_all() 

should be equal to

some_list = get_the_list() for x in generate(some_list) 

Answer 1

You're probably looking for Generator Delegation (PEP380)

For simple iterators, yield from iterable is essentially just a shortened form of for item in iterable: yield item

def generator(iterable):   for i in iterable:     yield do_something(i)  def generate_all():   yield from generator(get_the_list()) 

It's pretty concise and also has a number of other advantages, such as being able to chain arbitrary/different iterables!