Decorators Experiment

experiment.py

def state(state_value):
    def decorate(function):
        function.state = state_value
        return function
    return decorate


class ConsumerTest(object):
    @property
    def states(self):
        for attr in dir(self):
            obj = getattr(self, attr)
            if callable(obj) and hasattr(obj, 'state'):
                yield obj


class MyTest(ConsumerTest):
    @state('some books exist')
    def some_books_exist(self):
        print('THERE ARE FIVE BOOKS :)')

    @state('there are no books')
    def no_books_exist(self):
        print('THERE ARE ZERO BOOKS! :(')


def test_annotations():
    for foo in MyTest().states:
        print(foo.state)
        print(foo())
        print('===')
    assert False