from book import BookManage, Book
"""
代码优化：1、书籍编码自增 2、数据可以正常保存本地
"""
def welcome():
print("=========欢迎进入图书管理系统=========")
print("1、显示所有图书\n2、添加图书\n3、删除图书\n4、查找图书\n5、退出")
print("==================================")
def get_choose_number():
"""获取用户输入的菜单编号"""
choose_number = input("请输入菜单编码：") # 字符串类型
# 字符串编号不是数值 或者 不是1、2、3、4、5 返回 -1
if not choose_number.isdigit() or choose_number not in ["1", "2", "3", "4", "5"]:
return -1
return choose_number
def main():
bm = BookManage()
while 1:
welcome()
number = get_choose_number()
if number == -1:
print("您的输入有误，请重新输入")
continue
elif number == "1":
bm.show_book_list()
elif number == "2":
book_name = input("请输入图书名称：")
book_position = input("请输入图书位置：")
# 创建book对象，给与对象属性赋值
# bm.last_book_id获取列表最后一本书的num+1 写代码就是将自己的思维实现
book = Book(bm.last_book_id + 1, book_name, book_position)
# 调用添加书籍方法
bm.add_book(book)
elif number == "3":
name = input("请输入图书名称：")
bm.del_book_by_name(name)
elif number == "4":
# 查询某本书是否存在，不存在返回0 存在返回书籍名称的属性
name = input("请输入图书名称：")
bm.show_book_by_name(name)
else:
break
if __name__ == '__main__':
main()
