|
# 定义一个图书类
|
|
import atexit
|
|
import os
|
|
|
|
import yaml
|
|
|
|
|
|
class Book:
|
|
|
|
# 使用构造函数，接收实例化类时需要传入的参数
|
|
def __init__(self, bid, name, price, summary):
|
|
# 使用实例属性接收传入的值
|
|
self.bid = bid
|
|
self.name = name
|
|
self.price = price
|
|
self.summary = summary
|
|
|
|
# 重写对象的显示格式方法
|
|
def __str__(self):
|
|
return f"Bid: {self.bid} --- Name: {self.name} --- Price: {self.price} --- Summary: {self.summary}"
|
|
|
|
|
|
# 封装管理类
|
|
class BookManagement:
|
|
|
|
def __init__(self, filename='D:\\python脚本\\project_book\\book_data.yaml'):
|
|
# 定义一个实例变量，用来保存图书的信息，方便各个方法之间进行访问
|
|
# 把这个实例变量设置为私有，只能让类中方法对其进行操作
|
|
self.filename = filename
|
|
self.__books = []
|
|
atexit.register(self.save_data)
|
|
self.load_data()
|
|
|
|
def __menu(self):
|
|
'''
|
|
菜单函数：展示图书管理系统菜单信息，并获取用户输入
|
|
:return: 用户输入的编号字符串
|
|
'''
|
|
print("*****************************")
|
|
print("* 图书管理系统 *")
|
|
print("* 1. 添加新图书信息 *")
|
|
print("* 2. 通过编号修改图书信息 *")
|
|
print("* 3. 通过编号删除图书信息 *")
|
|
print("* 4. 通过书名删除图书信息 *")
|
|
print("* 5. 通过编号查询图书信息 *")
|
|
print("* 6. 通过书名查询图书信息 *")
|
|
print("* 7. 显示所有图书信息 *")
|
|
print("* 8. 退出系统 *")
|
|
print("*****************************")
|
|
select_op = input("输入编号选择操作：")
|
|
return select_op
|
|
|
|
def __get_bid(self):
|
|
'''
|
|
获取用户输入的图书编号
|
|
:return: 返回编号字符串
|
|
'''
|
|
bid = input("请输入图书编号:")
|
|
return bid
|
|
|
|
def __get_name(self):
|
|
'''
|
|
获取用户输入的名字
|
|
:return: 返回名字字符串
|
|
'''
|
|
name = input("请输入图书名字：")
|
|
return name
|
|
|
|
def __get_price(self):
|
|
'''
|
|
获取用户输入的价格
|
|
:return: 返回价格字符串
|
|
'''
|
|
while True:
|
|
price = input("请输入图书价格：")
|
|
if price.isdigit():
|
|
return int(price)
|
|
else:
|
|
print("输入价格不合法，请输入数字")
|
|
|
|
def __get_summary(self):
|
|
'''
|
|
获取用户输入的简介
|
|
:return: 返回简介字符串
|
|
'''
|
|
summary = input("请输入图书简介：")
|
|
return summary
|
|
|
|
def add_book(self, bid, name, price, summary):
|
|
'''
|
|
添加图书
|
|
:param bid: 图书编号 id
|
|
:param name: 图书名字
|
|
:param price: 图书价格
|
|
:param summary: 图书简介
|
|
:return: 操作提示信息
|
|
'''
|
|
for s in self.__books:
|
|
if s["bid"] == bid:
|
|
print("图书编号已存在，添加失败")
|
|
return "添加失败"
|
|
else:
|
|
book = Book(bid, name, price, summary)
|
|
self.__books.append({
|
|
'bid': bid,
|
|
'name': name,
|
|
'price': price,
|
|
'summary': summary
|
|
})
|
|
print(f"添加图书{book}成功")
|
|
return '添加成功'
|
|
|
|
def modify_book_by_id(self, bid, name, price, summary):
|
|
'''
|
|
通过编号修改图书信息
|
|
:param bid: 图书编号
|
|
:return: 操作提示信息
|
|
'''
|
|
for s in self.__books:
|
|
if s["bid"] == bid:
|
|
s['name'] = name
|
|
s["price"] = price
|
|
s["summary"] = summary
|
|
print("修改成功")
|
|
return "修改成功"
|
|
else:
|
|
print(f'没有 {bid} 对应的图书信息')
|
|
return "修改失败"
|
|
|
|
def delete_book_by_id(self, bid):
|
|
'''
|
|
通过ID删除图书信息
|
|
:param bid: 图书编号
|
|
:return: 操作提示信息
|
|
'''
|
|
for s in self.__books:
|
|
if s["bid"] == bid:
|
|
self.__books.remove(s)
|
|
print("删除成功")
|
|
return "删除成功"
|
|
else:
|
|
print(f'没有 {bid} 对应的学生信息')
|
|
return "删除失败"
|
|
|
|
def delete_book_by_name(self, name):
|
|
'''
|
|
通过图书名称 删除所有符合的图书
|
|
:param name: 图书名称
|
|
:return: 操作提示信息
|
|
'''
|
|
exist_s = []
|
|
# 找出所有要删除的图书
|
|
for s in self.__books:
|
|
if s["name"] == name:
|
|
exist_s.append(s)
|
|
|
|
# 开始删除
|
|
if len(exist_s) > 0:
|
|
for s in exist_s:
|
|
self.__books.remove(s)
|
|
print(f"名字为 {name} 的图书删除成功")
|
|
else:
|
|
print(f"成功删除 {len(exist_s)} 本图书")
|
|
return "删除成功"
|
|
else:
|
|
print(f"图书【{name}】不存在，无法删除")
|
|
return "删除失败"
|
|
|
|
def query_book_by_id(self, bid):
|
|
'''
|
|
通过编号查询图书信息
|
|
:param bid: 图书编号
|
|
:return: 操作提示信息
|
|
'''
|
|
for s in self.__books:
|
|
if s["bid"] == bid:
|
|
print(f"图书编号 {bid} 的图书信息如下：")
|
|
print(f"编号：{s['bid']}，名字：{s['name']}，价格：{s['price']}，简介：{s['summary']}")
|
|
return "查询成功"
|
|
else:
|
|
print(f"图书编号 {bid} 的图书不存在")
|
|
return "查询失败"
|
|
|
|
def query_book_by_name(self, name):
|
|
'''
|
|
通过名字查询图书信息
|
|
:param name:
|
|
:return: 操作提示信息
|
|
'''
|
|
exist_s = []
|
|
for s in self.__books:
|
|
if s["name"] == name:
|
|
exist_s.append(s)
|
|
|
|
if len(exist_s) > 0:
|
|
print(f"名字为 {name} 的图书共 {len(exist_s)} 本，信息如下：")
|
|
for s in exist_s:
|
|
print(f"编号：{s['bid']}，名字：{s['name']}，价格：{s['price']}，简介：{s['summary']}")
|
|
return "查询成功"
|
|
else:
|
|
print(f"名字为 {name} 的图书不存在")
|
|
return "查询失败"
|
|
|
|
def show(self):
|
|
'''
|
|
显示所有图书信息
|
|
:return: 格式化的图书信息
|
|
'''
|
|
print("所有图书信息如下：")
|
|
for s in self.__books:
|
|
print(f"编号：{s['bid']}，姓名：{s['name']}，价格：{s['price']}，简介：{s['summary']}")
|
|
|
|
def save_data(self):
|
|
with open(self.filename, 'w') as f:
|
|
yaml.dump(self.__books, f)
|
|
print(f"数据已保存至 {self.filename}")
|
|
|
|
def load_data(self):
|
|
if os.path.exists(self.filename):
|
|
with open(self.filename) as f:
|
|
self.__books = yaml.safe_load(f) or []
|
|
print(f"成功从 {self.filename} 加载数据")
|
|
else:
|
|
self.__books = []
|
|
print("数据文件不存在，已初始化空列表")
|
|
|
|
def manager(self):
|
|
'''
|
|
管理方法
|
|
:return:
|
|
'''
|
|
while True:
|
|
select_op = self.__menu()
|
|
if len(select_op) == 1 and select_op in "12345678":
|
|
if select_op == "1":
|
|
bid = self.__get_bid()
|
|
name = self.__get_name()
|
|
age = self.__get_price()
|
|
gender = self.__get_summary()
|
|
self.add_book(bid, name, age, gender)
|
|
elif select_op == "2":
|
|
bid = self.__get_bid()
|
|
name = self.__get_name()
|
|
age = self.__get_price()
|
|
gender = self.__get_summary()
|
|
self.modify_book_by_id(bid, name, age, gender)
|
|
elif select_op == "3":
|
|
bid = self.__get_bid()
|
|
self.delete_book_by_id(bid)
|
|
elif select_op == "4":
|
|
name = self.__get_name()
|
|
self.delete_book_by_name(name)
|
|
elif select_op == "5":
|
|
bid = self.__get_bid()
|
|
self.query_book_by_id(bid)
|
|
elif select_op == "6":
|
|
name = self.__get_name()
|
|
self.query_book_by_name(name)
|
|
elif select_op == "7":
|
|
self.show()
|
|
else:
|
|
break
|
|
else:
|
|
print("输入的数据不合法，请输入在合法范围内的操作编号！！！")
|
|
|
|
|
|
# 程序入口
|
|
if __name__ == '__main__':
|
|
# 获取图书管理系统实例
|
|
bm = BookManagement()
|
|
# 通过实例调用实例方法
|
|
bm.manager()
|