Python简单图书管理系统

'''
图书管理系统
library = [{},{},{}...]
1.借书
2.还书
3.查询（书名/作者查询）
4.查看所有
5.退出

'''
import time

library = [{'bookname': '水浒传', 'author': '施耐庵', 'price': 100, 'number': 40},
           {'bookname': '红楼梦', 'author': '曹雪芹', 'price': 90, 'number': 40},
           {'bookname': '西游记', 'author': '施耐庵', 'price': 98, 'number': 40},
           {'bookname': '三体', 'author': '刘慈欣', 'price': 88, 'number': 40}
           ]

while True:
    operation = input('请选择功能:\n 1.借书 \n 2.还书 \n 3.查询'
                      '\n 4.显示所有图书 \n 5.退出系统\n')
    if operation == '1':
        print('借书')
        name = input('请输入书的名称：')
        for book in library:
            # print(book.get('bookname'))
            if name == book.get('bookname') and book.get('number') != 0:
                print('借书{}成功'.format(book.get('bookname')))
                s = book.get('number') - 1
                book['number'] = s
                break
        else:
            print('未查到您的图书')
    elif operation == '2':
        print('还书')
        name = input('请输入您需要还书的书名：')
        for book in library:
            if name == book['bookname']:
                book['number'] += 1
                print('还书成功！！！')
                break
    elif operation == '3':
        print('查询')
        way = input('请输入您需要查询的方式：（1.按书名查询/2.按作者查询）')
        if way == '1':
            name = input('请输入您需要查询的书名')
            for book in library:
                if name == book['bookname']:
                    print(book['bookname'], book['author'], book['price'], book['number'])
                    break
            else:
                print('未查询到您所需要的图书')
        elif way == '2':
            author = input('请输入您需要查询的作者')
            for book in library:
                if author == book['author']:
                    print(book['bookname'], book['author'], book['price'], book['number'])
            else:
                print('未查询到您所需要的图书')
        else:
            print('输入错误！')
    elif operation == '4':
        print('显示所有图书')
        print('书名', '作者', '价格', '剩余图书')
        for book in library:
            print(book['bookname'], book['author'], book['price'], book['number'])
    elif operation == '5':
        print('正在退出本系统')
        time.sleep(3)
        print('退出成功！！！感谢您的使用')
        break
    else:
        print('输入有误，请重新输入')