データの保存忘れを防ぐために警告を表示する処理を施したい。

Question

前提・実現したいこと

Ruby初心者です。
現在簡単な蔵書管理アプリを作成しています。
実行すると初期化、登録、表示、別ファイルへ保存、終了を選択することができます。
入力したデータを保存せずに終了しようとすると警告が表示されるような処理を実装してみたいのですが、何をどうしたらいいやら全く見当がつきません。

助力お願いいたします。

発生している問題・エラーメッセージ

該当のソースコード

require 'date'

class BookInfo
  attr_accessor :title, :author, :pages, :publish_date
  def initialize(title, author, pages, publish_date)
    @title = title
    @author = author
    @pages = pages
    @publish_date = publish_date
  end

  def to_s
    "#{@title},#{@author},#{@pages},#{@publish_date}"
  end

  def toFormattedString(sep="\n")
    "書籍名:#{@title}#{sep}著者名:#{@author}#{sep}ページ数:#{@pages.to_s}#{sep}発刊日:#{@publish_date.to_s}#{sep}"
  end
end


class BookInfoManager

  def initialize(csv_fname = "book_info.csv")
    @book_infos = []
    @csv_fname = csv_fname
    if File.exist?(@csv_fname)
      open(@csv_fname, "r") do |file|
        file.each do |line|
          title, author, pages, publish_date = line.chomp.split(",")
          @book_infos << BookInfo.new( title, author, pages.to_i, Date.parse(publish_date))
        end
      end
    end
  end

  def addBookInfo
    book_info = BookInfo.new("", "", 0, Date.new)
    print "書籍名:"
    book_info.title = gets.chomp
    print "著者名:"
    book_info.author = gets.chomp
    print "ページ数:"
    book_info.pages = gets.chomp.to_i
    print "発刊年:"
    year = gets.chomp.to_i
    print "発刊月:"
    month = gets.chomp.to_i
    print "発刊日:"
    day = gets.chomp.to_i
    book_info.publish_date = Date.new( year, month, day)

    @book_infos << book_info
  end

  def listAllBookInfos
    @book_infos.each do |info|
      print info.toFormattedString
    end
  end

  def saveAllBookInfos
    open(csv_fname = "book_info.csv", "w") do |file|
      @book_infos.each do |book|
        file.puts book.to_s
      end
    end
  end

  def run
    while true
      print "蔵書管理アプリ
      0. 蔵書データベースの初期化
      1. 蔵書データの登録
      2. 蔵書データの表示
      8.蔵書データをファイルへ保存
      9. 終了
      番号を選んでください(0, 1, 2, 8, 9):"

      num = gets.chomp

      case num
        when '0'
          initialize
        when '1'
          addBookInfo
        when '2'
          listAllBookInfos
        when '8'
          saveAllBookInfos
        when '9'
        break
      end
    end
  end
end

bim = BookInfoManager.new()
bim.run

試したこと

1でデータを追加した場合のみ、9を押した際に事前に8で保存していなければ警告メッセージを表示したいのですが、思ったより複雑で何をどうしたらいいかわかりません。

Answer 1

@dirty_flag (名前は適当) ってのを用意して、最初は false、登録したら true、保存したら false にするようにして、終了時にこの値をチェックして警告を出せば良いのでは。

 class BookInfoManager
 
   def initialize(csv_fname = "book_info.csv")
     @book_infos = []
     @csv_fname = csv_fname
+    @dirty_flag = false
     # 略
   end
 
   def addBookInfo
     # 略
 
     @book_infos << book_info
+    @dirty_flag = true
   end
 
   def saveAllBookInfos
     open(csv_fname = "book_info.csv", "w") do |file|
       @book_infos.each do |book|
         file.puts book.to_s
       end
     end
+    @dirty_flag = false
   end
 
   def run
     while true
       # 略
 
       case num
         # 略
         when '9'
+          if @dirty_flag
+            print "保存してない変更は失われます。よろしいですか？"
+            next if gets.chomp != "y"
+          end
         break
       end
     end
   end
 end

Comments

[popyRon]

感謝いたします。