#-*- coding=utf8 -*- 
# @author:sololi
# date: 2020/11/12
# 文件說明 : data
import sys
def print_muen():
 print("----菜單提示----")
 print("[1]:添加圖書")
 print("[2]:查詢圖書")
 print("[3]:刪除圖書")
 print("[4]:修改圖書信息")
 print("[5]:顯示所有圖書")
 print("[6]:退出")

def add_book():
 book_id=input("請輸入圖書編號")
 book_name=input("請輸入圖書名稱")
 book_loc=input("請輸入圖書位置")
 for i in books:
 if book_id==i["book_id"]:
 print("編號重復(fù)")
 break
 else:
 book={}
 book["book_id"]=book_id
 book["book_name"]=book_name
 book["book_loc"]=book_loc
 books.append(book)
 new_file(books)
 print("添加成功")

def find_book():
 jg=0
 while True:
 book_name=input("輸入要找的書名")
 for i in books:
 if book_name==i["book_name"]:
 print(i)
 jg=1
 if jg==0:
 print("查找失敗請重新輸入")
 continue
 elif jg==1:
 break

def dell_book():
 book_name =input("請輸入圖書書名")
 i=0
 j=0
 jg=0
 flag=[]
 while i<len(books):
 if books[i]["book_name"]==book_name:
 id=books[i]["book_id"]
 flag.append(id)
 print("圖書編號:{}".format(id))
 jg=1
 i+=1
 if jg==0:
 print("找不到圖書名")
 if jg==1:
 book_id=input("請輸入刪除的圖書編號")
 i=0
 while j < len(flag):
 if flag[j] == book_id:
 while i < len(books):
  if books[i]["book_id"] == book_id:
  del books[i]
  new_file(books)
  print("刪除成功")
  break
  i+=1
 break
 j+=1
 else:
 print("輸入的編號有誤")
 return books

def modify_book():
 book_name = input("請輸入修改的圖書名稱")
 i = 0
 jg = 0
 flag = []
 while i < len(books):
 if books[i]["book_name"] == book_name:
 id = books[i]["book_id"]
 flag.append(id)
 print("本書圖書編號:{}".format(id))
 jg = 1
 i += 1
 if jg == 0:
 print("找不到圖書名")
 elif jg == 1:
 book_id = input("請輸入修改的圖書編號，或者其他需要修改的圖書編號")
 for i in books:
 if book_id == i["book_id"]:
 New_id = input("新的圖書編號")
 if New_id == i["book_id"]:
  print("輸入編號重復(fù)")
  break
 else:
  New_name = input("新的圖書名稱")
  New_loc = input("新的圖書位置")
  i["book_id"] = New_id
  i["book_name"] = New_name
  i["book_loc"] = New_loc
  new_file(books)
  break
 else:
 print("輸入有誤")

def all_book():
 for i in books:
 print(i)

def read_book(data):#將文本文件里的圖書信息讀取到books中
 f = open(data, mode="r", encoding="utf8")
 books = f.readlines()
 f.close()
 for i in range(0, len(books)):
 books[i] = eval(books[i].replace("\n", ""))
 return books

def new_file(books):#將books的內(nèi)容更新到文本文件里
 i=0
 nf = open('data', 'w+',encoding='utf8') # 以清空原文本文件內(nèi)容的方式打開文件寫入
 while i<len(books):
 new=str(books[i])
 nf.write(new)
 if i !=len(books)-1:
 nf.write("\n")
 i+=1
 nf.close()

if __name__=="__main__":#主要流程
 while True:
 data="data"
 books=read_book(data)
 print_muen()
 choice=input("請輸入您的選擇")
 if choice=='1':
 add_book()
 all_book()
 elif choice=='2':
 find_book()
 elif choice=='3':
 dell_book()
 all_book()
 elif choice=='4':
 modify_book()
 all_book()
 elif choice == '5':
 all_book()
 elif choice == '6':
 print("感謝您的使用")
 sys.exit(0)
 else:
 print("您的輸入有誤,請重新輸入")