pattern/memento.py

# Поведенческий шаблон проектирования ("Снимок")
# Позволяет сохранять и восстанавливать прошлые состояния объектов, не раскрывая подробностей их реализации. Не нарушает инкапсуляцию. Тратит много памяти


import abc
import typing


class IMemento(metaclass=abc.ABCMeta):
    @abc.abstractmethod
    def get_dollars(self) -> int:
        pass

    @abc.abstractmethod
    def get_euro(self) -> int:
        pass


class ExchangeMemento(IMemento):
    def __init__(self, d: int, e: int):
        self.__dollars = d
        self.__euro = e

    def get_dollars(self) -> int:
        return self.__dollars

    def get_euro(self) -> int:
        return self.__euro


class Exchange:
    def __init__(self, d: int, e: int):
        self.__dollars = d
        self.__euro = e

    def get_dollars(self):
        print('Долларов: {}'.format(self.__dollars))

    def get_euro(self):
        print('Евро {}'.format(self.__euro))

    def sell(self):
        if self.__dollars > 0:
            self.__dollars -= 1

    def buy(self):
        self.__euro += 1

    def save(self) -> ExchangeMemento:
        return ExchangeMemento(self.__dollars, self.__euro)

    def restore(self, exchange_memento: IMemento):
        self.__dollars = exchange_memento.get_dollars()
        self.__euro = exchange_memento.get_euro()


class Memory:
    def __init__(self, exchange: Exchange):
        self.__exchange = exchange
        self.__history: typing.Deque[IMemento] = []

    def backup(self):
        self.__history.append(self.__exchange.save())

    def undo(self):
        if len(self.__history) == 0:
            return
        else:
            self.__exchange.restore(
                self.__history.pop()
            )


if __name__ == "__main__":
    exchange = Exchange(d=10, e=10)
    memory = Memory(exchange)

    exchange.get_dollars()
    exchange.get_euro()

    print('----- Продажа доллара, покупка евро -----')

    exchange.sell()
    exchange.buy()

    exchange.get_dollars()
    exchange.get_euro()

    print('----- Сохраним состояние  -----')
    memory.backup()

    print('----- Продажа доллара, покупка евро -----')
    exchange.sell()
    exchange.buy()

    exchange.get_dollars()
    exchange.get_euro()

    print('----- Востановим состояние из памяти -----')
    memory.undo()

    exchange.get_dollars()
    exchange.get_euro()

    # Долларов: 10
    # Евро 10
    # ----- Продажа доллара, покупка евро -----
    # Долларов: 9
    # Евро 11
    # ----- Сохраним состояние  -----
    # ----- Продажа доллара, покупка евро -----
    # Долларов: 8
    # Евро 12
    # ----- Востановим состояние из памяти -----
    # Долларов: 9
    # Евро 11