图书管理系统图书管理系统
某图书馆所藏图书如表1所示：
| 书号 |
书名 |
出版社 |
作者 |
价格 |
库存 |
| 10001 |
C语言程序设计 |
清华大学出版社 |
张三 |
51 |
5 |
| 10002 |
Python程序设计基础 |
高等教育出版社 |
李四 |
45 |
6 |
| … |
… |
… |
… |
… |
… |
借阅信息如表2所示：
| 学号 |
书号 |
借阅日期 |
| 1001 |
10001 |
20190309 |
| 1002 |
10002 |
20190401 |
| … |
… |
… |
学生信息如表3所示
| 学号 |
姓名 |
性别 |
班级 |
| 1001 |
张三 |
男 |
化学1801 |
| 1002 |
李四 |
女 |
化学1802 |
主要实现以下功能：
（1）能分别输入表1和表3的信息，并将其分别保存到两个文件
（2）从分别从文件中读出图书信息和学生信息
（3）实现借阅功能：输入学号和书号，如果借阅成功（学号所对应的学生在表3中并且书号所对应的图书在表1中且库存大于等于1），修改表1和表2，并保存到文件
（4）实现还书功能：从表2中删除该学生的借阅信息，并修改表1的库存信息，并保存到文件
（5）输入某学生姓名，可以查询该生的借阅图书信息
（6）输入某书号，可以查询借阅该书的学生信息
（7）统计某出版社的藏书量，统计某学生当前借书量
（8）利用第三方库matplotlib中的pyplot绘制统计图，如绘制各出版社的藏书量折线图
import time
import pandas as pd
import os
import matplotlib.pyplot as plt
# 先定义一个配置类
from Tools.scripts.treesync import raw_input
class Config:
# 定义表一
table1_url = "C:\\Users\\Shinelon\\Desktop\\untitled\\resource\\table1.csv"
table1_header = ["书号", "书名", "出版社", "作者", "价格", "库存"]
# 定义表二
table2_url = "C:\\Users\\Shinelon\\Desktop\\untitled\\resource\\table2.csv"
table2_header = ["学号", "书号", "借阅日期"]
# 定义表三
table3_url = "C:\\Users\\Shinelon\\Desktop\\untitled\\resource\\table3.csv"
table3_header = ["学号", "姓名", "性别", "班级"]
class library_system(object):
def __init__(self):
print("图书馆系统初始化中...")
self.input_table1_table2()
self.book_table, self.borrow_table, self.student_table = self.get_info_from_file()
print("图书馆系统初始化完成！")
self.menu()
def input_table1_table2(self):
print("请输入图书馆藏书表")
table1 = []
print("输入藏书表:书号 书名 出版社 作者 价格 库存\n")
x = raw_input('请输入数据：')
while x != "exit\n":
y = x.split()
if len(y) != 6:
x = raw_input('列数不一致\n请重新输入:')
else:
table1.append(y)
x = raw_input('下一条记录:')
print("请输入学生信息表")
table3 = []
print("输入学生表:学号 姓名 性别 班级\n")
x = raw_input('请输入数据:')
while x != "exit\n":
y = x.split()
if len(y) != 4:
x = raw_input('数据列数不一致\n请重新输入:')
else:
table3.append(y)
x = raw_input('下一条记录:')
table1 = pd.DataFrame(table1)
table3 = pd.DataFrame(table3)
table1.to_csv(Config.table1_url, header=Config.table1_header, index=None)
table3.to_csv(Config.table3_url, header=Config.table3_header, index=None)
def get_info_from_file(self):
book_table = pd.read_csv(Config.table1_url)
borrow_table = []
student_table = pd.read_csv(Config.table3_url)
print("信息读取成功...")
return book_table, borrow_table, student_table
def save_file(self):
self.book_table.to_csv(Config.table1_url, header=Config.table1_header, index=None)
boo = pd.DataFrame(self.borrow_table)
boo.to_csv(Config.table2_url, header=Config.table2_header, index=None)
self.student_table.to_csv(Config.table3_url, header=Config.table3_header, index=None)
print("保存文件成功")
# 实现借阅功能：输入学号和书号，如果借阅成功（学号所对应的学生在表3中并且书号所对应的图书在表1中且库存大于等于1），修改表1和表2，并保存到文件
def borrow(self, student_id, book_id):
flag1=True
for i in range(len(self.student_table["学号"])):
if int(self.student_table["学号"][i])==int(student_id):
print("查询到该学生")
flag1=False
for j in range(len(self.book_table["书号"])):
if int(self.book_table["书号"][j])==int(book_id) and int(self.book_table["库存"][j])>0:
print("库存充足")
self.book_table["库存"][j] = str(int(self.book_table["库存"][j])-1)
borrow = [student_id,book_id, time.strftime("%d/%m/%Y")]
self.borrow_table .append(borrow)
self.save_file()
break
if not flag1:
break
if flag1:
print("未查询到该学生")
print("借阅图书")
# 实现还书功能：从表2中删除该学生的借阅信息，并修改表1的库存信息，并保存到文件
def let_back(self, student_id, book_id):
for j in range(len(self.book_table["书号"])):
if int(self.book_table["书号"][j])==int(book_id):
self.book_table["库存"][j] = str(int(self.book_table["库存"][j])+1)
for i in range(len(self.borrow_table)):
if int(self.borrow_table[i][0])==int(student_id) and int(self.borrow_table[i][1])==int(book_id):
del self.borrow_table[i]
self.save_file()
break
break
print("还书")
# 输入某书号，可以查询借阅该书的学生信息
def find_by_book_id(self, book_id):
stu_ids=set()
#先获取对应人的学号
for i in range(len(self.borrow_table)):
if int(self.borrow_table[i][1])==int(book_id):
stu_ids.add(int(self.borrow_table[i][0]))
stu=[]
for i in range(len(self.student_table["学号"])):
if int(self.student_table["学号"][i]) in stu_ids:
stu.append(self.student_table.iloc[i])
print("查询借了某本书的学生信息")
print(stu)
def sum_by_student_id(self, student_id):
sum=0
for i in range(len(self.borrow_table)):
if int(self.borrow_table[i][0])==int(student_id):
sum+=1
print("统计某学生当前借书量"+str(sum))
# 统计某出版社的藏书量，统计某学生当前借书量
def sum_by_publish(self, publish_name):
book_id = set()
sum=0
for i in range(len(self.book_table["出版社"])):
if str(self.book_table["出版社"][i])==str(publish_name):
book_id.add(int(self.book_table["出版社"][0]))
sum+=int(self.book_table["库存"][i])
for i in range(len(self.borrow_table)):
if self.borrow_table[i][1] in book_id:
sum+=1
print(str(publish_name)+"统计藏书量"+str(sum))
return sum
# 输入某学生姓名，可以查询该生的借阅图书信息
def find_by_student_name(self, student_name):
stu_id=set()
for i in range(len(self.student_table["姓名"])):
if str(self.student_table["姓名"][i])==str(student_name):
stu_id.add(int(self.student_table["学号"][i]))
print("找到")
book_id=set()
res=[]
for i in range(len(self.borrow_table)):
if int(self.borrow_table[i][0]) in stu_id:
book_id.add(int(self.borrow_table[i][1]))
for i in range(len(self.book_table["书号"])):
if int(self.book_table["书号"][i]) in book_id:
res.append(self.book_table.iloc[i])
print("查询某学生的借书信息")
print(res)
#获取各出版社的藏书量折线图
def get_publish(self):
pub = set()
for i in range(len(self.book_table["出版社"])):
pub.add(str(self.book_table["出版社"][i]))
pubL = list(pub)
x1 = range(len(pubL))
numL = []
for i in range(len(pubL)):
numL.append(self.sum_by_publish(pubL[i]))
plt.title('各出版社的藏书量折线图')
plt.xlabel('出版社名字')
plt.ylabel('藏书量')
plt.plot(pubL, numL, 'r', label='藏书量')
plt.xticks(x1, pubL, rotation=0)
plt.legend()
plt.grid()
plt.show()
# 利用第三方库matplotlib中的pyplot绘制统计图，如绘制各出版社的藏书量折线图，绘制各学生借书量的饼图等
def get_stu(self):
stu_id = set()
for i in range(len(self.student_table["学号"])):
stu_id.add(int(self.student_table["学号"][i]))
stuL = list(stu_id)
numL = []
for i in range(len(stuL)):
numL.append(self.sum_by_student_id(stuL[i]))
plt.rcParams['font.sans-serif'] = 'SimHei' # 设置中文显示
plt.figure(figsize=(6, 6)) # 将画布设定为正方形，则绘制的饼图是正圆
plt.pie(numL, labels=stuL, autopct='%1.1f%%') # 绘制饼图
plt.title('2018年饼图') # 绘制标题
plt.show()
# 1绘制各出版社的藏书量折线图，2绘制各学生借书量的饼图
def show_diaglo(self,chos):
if str(chos)=="1":
self.get_publish()
if str(chos)=='2':
self.get_stu()
print("展现图表")
def cls(self):
os.system("cls")
def menu(self):
level_1_choose = raw_input("按回车继续")
while level_1_choose != "exit\n":
self.cls()
print("图书管理系统菜单\n")
print("1 借阅功能 请输入： 1 学号 书号")
print("2 还书功能 请输入： 2 学号 书号")
print("3 查询学生借阅信息 请输入： 3 学生姓名")
print("4 查询借该书的学生信息 请输入： 4 书号")
print("5 统计某出版社的藏书量 请输入： 5 出版社名")
print("6 统计某学生当前的借书量 请输入： 6 学生学号")
print("7 绘图功能 请输入：7")
level_1_choose = raw_input("请选择1-7你要选择的功能,输入exit退出")
turn = True
while turn:
sp = level_1_choose.split()
# 1-7
if sp[0] == "1":
if len(sp) == 3:
self.borrow(sp[1], sp[2])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "2":
if len(sp) == 3:
self.let_back(sp[1], sp[2])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "3":
if len(sp) == 2:
self.find_by_student_name(sp[1])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "4":
if len(sp) == 2:
self.find_by_book_id(sp[1])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "5":
if len(sp) == 2:
self.sum_by_publish(sp[1])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "6":
if len(sp) == 2:
self.sum_by_student_id(sp[1])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
elif sp[0] == "7":
if len(sp) == 2:
self.show_diaglo(sp[1])
turn = False
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
else:
turn = True
level_1_choose = raw_input("输入错误，请重新输入\n")
if __name__ == '__main__':
ls = library_system()
