欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python备忘录模式

程序员文章站 2024-03-23 23:12:10
...

备忘录模式,类似于撤销功能,提供保存并恢复之前状态的功能。

class Momento(object):
    def __init__(self):
        super().__init__()
        self.dct = {}

    def set_status(self, dct):
        self.dct = {}
        for key in dct:
            if not key.startswith('_'):
                self.dct[key] = dct[key]

    def get_status(self):
        return self.dct


class Obj(object):
    def __init__(self, a):
        super().__init__()
        self.attribute_a = a

        self._momento = Momento()

    def set_a(self, status):
        self.backup_current_status()
        self.attribute_a = status

    def restore_previous_status(self):
        self.__dict__.update(self._momento.get_status())

    def backup_current_status(self):
        self._momento.set_status(self.__dict__)

    def __str__(self):
        return "attribute_a: " + self.attribute_a + '\n'


def main():
    obj = Obj('a')
    obj.set_a('aa')
    print('current_status:\n', obj)
    obj.restore_previous_status()
    print('previous_status:\n', obj)


if __name__ == '__main__':
    main()