图书管理系统用户

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":
            pass
        elif number == "4":
            # 查询某本书是否存在，不存在返回0 存在返回书籍名称的属性
            pass
        else:
            break


if __name__ == '__main__':
    main()