#獲取圖書
#從文件中讀取圖書，并寫入列表
def getBook():
    bookList =[]
    f = open("book.txt","r")
    tempList = f.readlines()    
    for temp in tempList:
        temp = temp[:-1]
        bookList.append(temp)
    f.close()
    return bookList

#保存圖書    
#將添加的圖書從列表寫入文件
def saveBook(bookList):
    f = open("book.txt","w")
    for temp in bookList:
        f.write(temp+"\n")
    f.close()
    
#添加圖書
#將圖書添加至列表    
def addBook(book):
    global books
    books.append(book)
    
#刪除圖書
#將圖書從列表中刪除    
def removeBook(book):
    global books
    if book in books:
        books.remove(book)
    else:
        print("沒有找到",book,"這本書。")
        
        
        
#菜單        
print("""
1.添加圖書
2.刪除圖書
3.展示所有圖書
4.保存并退出
""")

book = getBook()

while True:
    x = input("請輸入功能鍵(1-4):")
    if x == "1":
        name = input("請輸入要添加的圖書名稱: ")
        addBook(name)
    elif x == "2":
        tempBook = input("請輸入要刪除的圖書名稱: ")
        removeBook(tempBook)
    elif x == "3":
        print(books)
    elif x == "4":
        saveBook(books)
        break
    else:
        print("輸入有誤，請重新輸入!")