change stats

gistfile1.py

#https://coderwall.com/p/jo39na/python-decorators-using-self

# todo it inlater

states = {"a":"b", "b":"c", "c":"d", "d","end"}
class Foo:
     #skip some staff
    def will_change_state(f):
            def wrapper(*args):
                ret =  f(*args)
                args[0].change_state()
                return ret
            return wrapper

     def change_state(self):
            self.state = states[self.state]

     @will_change_state
     def func1(self):
           pass

     @will_change_state
     def func2(self):
           pass