Generator Attributes

generator_attributes.py

import functools
# forked from http://code.activestate.com/recipes/577057-generator-attributes/
# doesn't rely on exec or locals, works in python 2 and 3
def generator_with_attributes(fn):
    @functools.wraps(fn)
    def fnwrapper(*args, **kwargs):
        wrapper = type(
            'GeneratorWithAttributes', (object,),
            {'__iter__': lambda self: gen})()
        gen = fn(wrapper, *args, **kwargs)
        return functools.wraps(gen, ('close', 'next', 'send', 'throw',
                               'gi_code', 'gi_frame', 'gi_running', '__name__'
                                     ), ())(wrapper)
    return fnwrapper

xxexample.py

@generator_with_attributes
def flagrange(self, min_, max_, flag):
    '''
    >>> g = flagrange(0, 10, 5)
    >>> for i in g:
    ...    print(g.flag)
    False
    False
    False
    False
    False
    True
    True
    True
    True
    True
    '''

    self.flag = False
    for x in range(min_, flag):
        yield x
    self.flag = True
    for x in range(flag, max_):
        yield x