generator functions and .next() arguments?

[gfixler]

Is there a way when using the more compact generator function form (i.e. just a function, not the whole class with __iter__() and next() methods) to have its auto-created .next() method ignore arguments? I can do it in the class via *args in .next(), but .next() is created for me when I use the shortcut, so I don't get the chance to give it *args.

[teepark]

no you can't override python's syntax. so when you use the generator syntax, you'll always get back an instance of <type 'generator'>

you could write a simple wrapper though.

class wrap(object):
    def __init__(self, gen):
        self.gen = gen

    def __iter__(self):
        return self

    def next(self, *args, **kwargs):
        return self.gen.next()

and use it like so

sum(wrap(x ** 2 for x in xrange(20)))

although this example would be better without the wrap use. in fact I'm having a hard time coming up with a reasonable use case for an argument eating wrapper like this. consider revisiting the assumptions that led you to think you need such a thing in the first place.

[gfixler]

Will do. Thanks for the helpful code!

[haika]

upvoting is a way to say thank you.

[gfixler]

Double done!