# Update class names from the immediately prior version only
# to ensure that cycles are not possible

from apptools.persistence.updater import Updater

def cleanup_foo(self, state):

    print 'cleaning up Foo0'
    state['firstname'] = state['prenom']
    state['lastname'] = state['surnom']

    del state['prenom']
    del state['surnom']

    '''for key in state:
        print '%s state ---> %s' % (key, state[key])
    '''

    #self.__setstate_original__(state)
    self.__dict__.update(state)


def update_project(self, state):
    print 'updating to v1'
    metadata = state['metadata']
    metadata['version'] = 1
    metadata['diesel'] = 'E300TD'
    return state


class Update1(Updater):

    def __init__(self):

        self.refactorings = {
            ("__main__", "Foo0"): ("__main__", "Foo"),
        }

        self.setstates = {
            ("cplab.project", "Project"):  update_project
        }