1.添加书籍
2.查询数据
3.借书
存储方式 ,用excel保存到硬盘上或者用.txt文件保存
1.用excel存储
# 一、介绍 # 主要功能实现 # 1、借书 # 2、添加新书 # 3、查找图书 # 数据存储:excel表 import xlwt import xlrd import xlutils.copy import os #book = {"位置":"","书名":"","价格":"","作者":""} #存储方式 用excel title =["位置","书名","价格","作者"] #查看当前的书本数,也就行号 def read_book_num(): path = os.path.join(os.getcwd()+r'\图书.xls') print(path) flag = os.path.exists(path) if(flag): book_excel = xlrd.open_workbook("图书.xls") sheet1 = book_excel.sheets()[0] book_num = sheet1.nrows else: book_num = 0 return book_num def add_book(book_num): #判断excel是否存在,如果不存在,就创建 path = os.path.join(os.getcwd()+r'\图书.xls') flag = os.path.exists(path) print("flag",flag) if(flag): #如果存在,就打开excel book_excel = xlrd.open_workbook("图书.xls") #并复制之前的已经存在的数据 book_excel = xlutils.copy.copy(book_excel) sheet1 = book_excel.get_sheet(0) #sheet1 = book_excel.sheets()[0] else: book_excel = xlwt.Workbook("图书.xls") #新建excel sheet1 = book_excel.add_sheet(sheetname="图书表单",cell_overwrite_ok=True) while(1): #打印提示 button_num = input("请选择你的操作\n:"+"1.添加新书\n"+"2.退出请按q\n") if(button_num == 'q'): break elif (button_num == "1"): #输入一本书的所有信息,并且先存储到book里面 book = [] #清空书本信息 input_value = '' #清空输入 for i in range(4): print("请输入:",title[i]) input_value = input() book.append(input_value) #存储到硬盘(将输入的数据存储到excel中) for i in range(4): #写入第book_num行数据 sheet1.write(book_num,i,book[i]) book_num = book_num +1 #总书数量加1 book_excel.save("图书.xls") print("添加成功") else: print("输入无效,请重新输入!") def search_book(): #打开excel book_excel = xlrd.open_workbook("图书.xls") sheet1 = book_excel.sheets()[0] book_num = sheet1.nrows while(1): #输入书名 chose= input("请输入你的操作:\n"+"1.查询书籍:\n"+"2.退出请按q\n") if(chose == 'q'): break elif (chose == '1'): bookname = input("请输入书名:") for i in range(0,book_num): if(bookname == sheet1.cell(i,0).value): print("查询成功,查询结果为\n",sheet1.row_values(i)) return else: print("查询失败,本书库没有此书") return else: print("操作有误,请重新输入!") def borrow_book(): #打开excel book_excel = xlrd.open_workbook("图书.xls") sheet1 = book_excel.sheets()[0] book_num = sheet1.nrows book_excel_copy = xlutils.copy.copy(book_excel) sheet1_copy = book_excel_copy.get_sheet(0) #重新创建一个excel,用于保存更新后的数据 # book_excel_new = xlwt.Workbook("图书.xls") #新建excel # sheet1_new = book_excel_new.add_sheet(sheetname="1",cell_overwrite_ok=True) while(1): #输入书名 print("1.请输入借书书名\n2.按q退出借书界面") bookname = input() if(bookname == 'q'): return else: #查找 a = 0 for i in range(0, book_num): if( bookname == sheet1.cell(i, 0).value ): for j in range(4): a = i + 1 while(book_num-a): sheet1_copy.write(i,j,sheet1.cell(a,j).value)#清除位置 a += 1 print("借阅成功") book_excel_copy.save('图书.xls') return # else: # a = i # sheet1_copy.write(i,j,sheet1.cell(a,j).value)#清除位置 #book_excel_copy.save('图书.xls') if __name__ == '__main__': book_num = read_book_num() print(book_num) while(1): print("******图书管理系统****") print("******1.添加新书******") print("******2.查询书籍******") print("******3.借书*********") print("******4.退出*********") op = input("请输入你的操作:") if(op == "1"): add_book(book_num) elif (op == "2"): search_book() elif (op == "3"): borrow_book() elif (op == "4"): break else: print("输入无效,请重新输入!")
2.用txt文件方式存储
def add_book(): file = open("图书管理系统.txt","a+") print("请输入要添加的书籍信息:") id = input("id:") name = input("bookname:") author = input("author:") #table = [name,id,author] file.write(id+" "+name+" "+author+"\r") print("书籍添加成功!") file.close() def serch_book(): file = open("图书管理系统.txt","r") name = input("请输入要查询书籍名称:") read_data_all = [] count = len(file.readlines()) #print(count) file.seek(0,0) #需要将文件指针移动到开头 for i in range(count): read_data = file.readline().split() read_data_all.append(read_data) for read_data in read_data_all: # print(type(read_data)) if(name==read_data[0]): print("查询到的数据信息为:",read_data) break else: print("查找失败") file.close() return read_data def borrow_book(): file = open("图书管理系统.txt","r+") #先查找书籍存不存在,如果存在就借出 count = len(file.readlines()) read_data_all= [] file.seek(0,0) #需要将文件指针移动到开头 for i in range(count): read_data = file.readline().split() read_data_all.append(read_data) print(read_data_all) file.close() book = serch_book() file = open("图书管理系统.txt","w") for line in read_data_all: if book==line: continue line_to_str = ' '.join(line) #将列表装换成字符串 file.write(line_to_str+"\n") if __name__ == "__main__": #open直接打开一个文件,如果文件不存在则创建文件 while(1): print("******图书管理系统****") print("******1.添加新书******") print("******2.查询书籍******") print("******3.借书**********") print("******4.退出**********") op = input("请输入你的操作:") if(op == "1"): add_book() elif(op == "2"): serch_book() elif(op == "3"): borrow_book() else: break