Source code for freeflux.utils.context

'''Define the Context class.'''




[docs]
__author__ = 'Chao Wu'






[docs]
class Context():
    
    def __init__(self):
        


[docs]
        self.operations = []


        
    


[docs]
    def add_undo(self, op):
        '''
        Parameters
        ----------
        op: callable
            Operation to reset the model.
        '''
        
        self.operations.append(op)






[docs]
    def undo(self):
        
        while self.operations:
            op = self.operations.pop()
            op()