class Originator:
    """
    Đối tượng Originator lưu giữ một số trạng thái có thể thay đổi theo thời gian. Nó cũng định nghĩa một phương thức để lưu trạng thái của nó vào trong một Memento và một phương thức khác để khôi phục trạng thái từ Memento.
    """
    _state = ""

    def set(self, state):
        """
        Đặt trạng thái của Originator.
        """
        print(f"Originator: Đặt trạng thái thành {state}")
        self._state = state

    def save_to_memento(self):
        """
        Lưu trạng thái hiện tại vào trong Memento.
        """
        return self.Memento(self._state)

    def restore_from_memento(self, m):
        """
        Khôi phục trạng thái của Originator từ một đối tượng Memento.
        """
        self._state = m.get_saved_state()
        print(f"Originator: Trạng thái sau khi khôi phục từ Memento: {self._state}")

    class Memento:
        """
        Memento lưu trữ trạng thái nội bộ của đối tượng Originator. Memento có thể tiết lộ trạng thái cho thế giới, nhưng phải theo cách ngăn chặn việc thao túng đối tượng Originator.
        """
        def __init__(self, state):
            self._state = state

        def get_saved_state(self):
            """
            Trả về trạng thái đã lưu.
            """
            return self._state

class Caretaker:
    """
    Caretaker không phụ thuộc vào lớp Memento cụ thể. Do đó, nó không có quyền truy cập vào trạng thái của Originator, được lưu trong Memento. Nó làm việc với tất cả các Memento thông qua giao diện Memento cơ bản.
    """
    def __init__(self, originator):
        self.originator = originator
        self.saved_states = []

    def save_state(self):
        """
        Lưu trạng thái hiện tại của Originator.
        """
        print("Caretaker: Đang lưu trạng thái của Originator...")
        self.saved_states.append(self.originator.save_to_memento())

    def undo(self):
        """
        Khôi phục trạng thái đã lưu cuối cùng.
        """
        if not self.saved_states:
            print("Caretaker: Không có trạng thái đã lưu để khôi phục.")
            return
        print("Caretaker: Đang khôi phục trạng thái...")
        self.originator.restore_from_memento(self.saved_states.pop())

if __name__ == '__main__':
    originator = Originator()
    caretaker = Caretaker(originator)

    originator.set("Trạng thái 1")
    caretaker.save_state()

    originator.set("Trạng thái 2")
    caretaker.save_state()

    originator.set("Trạng thái 3")
    caretaker.save_state()

    originator.set("Trạng thái 4")
    caretaker.save_state()

    print("\nClient: Bây giờ, hãy quay lại!\n")
    caretaker.undo()

    print("\nClient: Một lần nữa!\n")
    caretaker.undo()

    print("\nClient: Một lần cuối để chứng minh nó xử lý ngăn xếp rỗng!\n")
    caretaker.undo()
    caretaker.undo()   

Originator: Đặt trạng thái thành Trạng thái 1
Originator: Đặt trạng thái thành Trạng thái 2
Originator: Đặt trạng thái thành Trạng thái 3
Originator: Đặt trạng thái thành Trạng thái 4
Client: Bây giờ, hãy quay lại!
Originator: Trạng thái sau khi khôi phục từ Memento: Trạng thái 4
Client: Một lần nữa!
Originator: Trạng thái sau khi khôi phục từ Memento: Trạng thái 3