From d1a7e0e466eb82c0877213f8234788de521f2233 Mon Sep 17 00:00:00 2001 From: p8vr56wsb <1315189220@qq.com> Date: Fri, 5 Nov 2021 21:05:52 +0800 Subject: [PATCH] ADD file via upload --- main.py | 526 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 526 insertions(+) create mode 100644 main.py diff --git a/main.py b/main.py new file mode 100644 index 0000000..36e6dea --- /dev/null +++ b/main.py @@ -0,0 +1,526 @@ +# 图书管理系统 +##1900301627陆洋 +##1800301439郑依贤 + + + + +# 用户注册 + +def logon(): + print("欢迎来到图书管理系统注册页面~") + username = input("请输入用户名：") + if len(username)<6: + print("用户名不能小于6个字符") + else: + email = input("请输入邮箱：") + password = input("请输入密码：") + if len(password)<8: + print("密码不能少于8位") + else: + rpassword = input("请确认密码：") + if password ==rpassword: + print("注册成功！") + # 函数调用，每追加一列数据都进行换行 每个数据之间都有空格 + preserve_data(path,[username,' '+ email,' '+ password + '\n']) + login_tips = input('是否登录？(yes/no)') + if login_tips =='yes': + login() + else: + pass + return True + else: + print("两次输入的密码不一致，请重新输入！") + # 递归调用 + logon() + + + + +# 保存数据到文件 +path = r'C:\Users\酒骁\Desktop\user.txt' +def preserve_data(file_path,data): + # 将字符串转换为bytes，因为write写入的是字节流，不能是字符串 当为w时需要解码 + # data = data.encode() + # 打开文件，追加数据到文件 + with open(file_path,'a') as wstream: + # 判断是否可写 + if wstream.writable(): + wstream.writelines(data) + else: + print("没有权限！") + + + + + +# 用户登录 +def login(): + print("欢迎来到图书管理系统登录页面~") + tips = input("是否已经注册？(yes/no)") + if tips =='yes': + while True: + username = input("输入用户名：") + password = input("输入密码：") + # 读取文件时可能会出现找不到文件的异常，因此使用try except + try: + # 读取文件中的内容 + with open(path, 'rb') as stream: + # 读取多行保存到列表中，列表中保存的是二进制，字节 + result = stream.readlines() + # print(result) + # 列表推导式，循环遍历列表，将字节解码为字符串放在一个新列表uesr_list + uesr_list = [i.decode() for i in result] + # print(uesr_list) + # 循环遍历列表，检查输入的用户名和密码是否在字符串中 + for i in uesr_list: + info = i.split(' ') + # print(info) + if username == info[0] and password == info[2].rstrip('\r\n'): + print("登录成功") + operate(book_path,username) + break + else: + raise Exception("用户名或密码错误，请重新输入！") + + except Exception as err: + print(err) + # 没有异常时执行else语句 + else: + break + else: + print("您还未注册，请先注册后再登录！") + # 递归 + logon() + + + + + +#查询图书 +def find_books(path): + try: + with open(path, 'r') as rstream: + # readlines读到的内容是一个列表 + container = rstream.readlines() + # 通过列表推导式得到新的列表，即每个元素去掉后面的换行符号 + new_container = [books_name.rstrip('\n') for books_name in container] + for b_name in new_container: + # 打印图书+《》 + print("《{}》".format(b_name)) + + except Exception as err: + print("错误原因：",err) + + + + + +#添加图书 +def add_book(b_path,username): + # 添加前首先判断是否是管理员 + permission(b_path, username) + # 追加图书 不能是w ,否则会清空之前的内容 + with open(b_path, 'a') as wstream: + # 判断是否可写 + if wstream.writable: + msg = input("请输入书名：") + try: + # 添加书籍之前判断某本书是否已经添加 + with open(b_path) as rstream: + while True: + line = rstream.readline() + # 去掉右边的换行 + line = line.rstrip('\n') + # 当找到空行是如果还没有找到与输入的书名一致的时候，就添加输入的书名 + if not line: + book = '\n' + msg + wstream.write(book) + print("添加成功") + break + else: + # 输入的图书和读到的行有一致的则提示不能重复添加 + if line == msg: + print("{}已添加,请不要重复添加哦~".format(msg)) + break + except Exception as err: + print("错误原因：", err) + else: + print("没有权限") + + + + + +#修改图书 +def update_book(b_path,username): + permission(b_path, username) + try: + with open(b_path, 'r') as rstream: + container = rstream.read() + # 通过'\n'来分割字符串，返回结果是列表 + container = container.split('\n') + # print(container) + # 删除前先展示有哪些图书 + find_books(book_path) + book_name = input("请输入需要修改的图书书名：") + # 循环遍历修改书名 + for i in range(len(container)): + if book_name == container[i]: + rbook_name = input("请输入修改后的图书书名：") + container[i] = rbook_name + '\n' + else: + # 列表中的每个书名后面加换行符，用于写入文件时换行 + container[i] = container[i] + '\n' + # print(container) + # 将书名更新后的内容以writelines写入文件中 writelines(可迭代) + with open(b_path, 'w') as wwstream: + wwstream.writelines(container) + print("修改成功") + except Exception as err: + print("错误原因：", err) + + + + +#删除图书 +def del_book(b_path,username): + permission(path, username) + try: + with open(b_path, 'r') as rstream: + container = rstream.read() + # 通过'\n'来分割字符串，返回结果是列表 + container = container.split('\n') + # print(container) + # 展示有哪些图书 + find_books(book_path) + book_name = input("请输入需要删除的图书书名：") + # 循环遍历修改书名 + for i in range(len(container) - 1): + if book_name == container[i]: + container.remove(container[i]) + else: + # 列表中的每个书名后面加换行符，用于写入文件时换行 + container[i] = container[i] + '\n' + # print(container) + # 将书名删除后的内容以writelines写入文件中 writelines(可迭代) + with open(b_path, 'w') as wwstream: + wwstream.writelines(container) + print("删除成功") + except Exception as err: + print("错误原因：", err) + + + +#借书 + +def borrow_book(username): + while True: + print("图书列表：") + find_books(book_path) + borrow_books = input("请选择图书：") + try: + with open('C:\\Users\\酒骁\Desktop\\user_books.txt') as rstream: + # 每次读取一行 + lines = rstream.readline() + lines = lines.rstrip('\n') + # 将读到的内容通过空格分割保存到列表 + lines = lines.split(' ') + # 判断输入的书是否已被借走 + if borrow_books not in lines: + # print(lines) + # 借书之前先判断该用户之前是否借过，如果借过，就在后面累加图书，用，分割图书 + # for user_book in lines: + if username in lines: + with open(r'C:\Users\酒骁\Desktop/user_books.txt', 'a') as wstream: + # 判断之前是否借过某本书 + if borrow_books not in lines: + wstream.write(' {}'.format(borrow_books)) + print("借书成功") + break + else: + print("您已借过此书，请从新选择！") + break + else: + # 将选择的图书与用户名一起保存到文件中 + with open(r'C:\Users\酒骁\Desktop/user_books.txt', 'a') as wstream: + wstream.write('\n{} {}\n'.format(username, borrow_books)) + print("借书成功") + break + else: + print("<<{}>>已被用户{}借走，请重新选择~".format(borrow_books,lines[0])) + except Exception as err: + print("错误原因：", err) + + + + +#还书 +def return_book(username): + try: + with open(r'C:\Users\酒骁\Desktop/user_books.txt') as rstream: + # print("{}您已借阅，未归还图书如下：".format(username)) + # 读到的结果是列表 + lines = rstream.readlines() + # 遍历列表，将里面的元素再拆分为列表 + for i in range(len(lines)): + # 去掉换行 + lines[i] = lines[i].rstrip('\n') + lines[i] = lines[i].rstrip(' ') + lines[i] = lines[i].split(' ') + for ii in range(len(lines[i])-1): + # 只打印登录用户借阅的图书 + if username == lines[i][0]: + print("{}您已借阅，未归还图书如下：".format(username)) + print(lines[i][ii+1]) + msg = input("请选择你要归还的图书：") + with open(r'C:\Users\酒骁\Desktop/user_books.txt') as rstream: + lines = rstream.readlines() + for i in range(len(lines)): + if username in lines[i] and msg in lines[i]: + # 用空字符串替换msg,即表示删除归还的图书 + lines[i] = lines[i].replace(msg, '') + with open(r'C:\Users\酒骁\Desktop/user_books.txt', 'w') as wstream: + # 将变更后的列表再写入文件，只变更当前用户的图书信息 + wstream.writelines(lines) + print("归还成功！") + break + + with open(r'C:\Users\酒骁\Desktop/user_books.txt') as rstream: + lines = rstream.readlines() + for i in range(len(lines)): + lines[i] = lines[i].rstrip('\n') + lines[i] = lines[i].rstrip(' ') + lines[i] = lines[i].split(' ') + # print(type(lines[i])) + for ii in range(len(lines[i])): + # 图书归还成功后判断列表中只有用户名了，如果只有用户名则将用户名用空字符串代替 + if username == lines[i][0] and len(lines[i]) == 1: + lines[i][0] = lines[i][0].replace(lines[i][0], '') + lines.append(lines[i][0]) + # print(lines) + str = '' + for i in range(len(lines)): + for ii in range(len(lines[i])): + # 将嵌套列表中的元素取出来拼接成字符串 + str += lines[i][ii] + ' ' + str += '\n' + # 遍历完毕删除之前列表里面嵌套的列表，追加字符串str + lines.clear() + lines.append(str) + + # print(lines) + # 将更新后的列表写入文件 + with open(r'C:\Users\酒骁\Desktop/user_books.txt', 'w') as wstream: + wstream.writelines(lines) + else: + print("您还没有借阅记录哦~") + + except Exception as err: + print("错误原因：", err) + + + + +#查看个人信息 +def look_person_info(path, username): + with open(path) as rstream: + lines = rstream.readlines() + # print(lines) + for info in lines: + # 分割成一个列表 + info = info.split(' ') + # print(info) + if username in info: + print("----个人信息----") + print("用户名：", info[0]) + print("邮箱：", info[1]) + print("密码：", info[2].rstrip(' ')) + + + + +#修改个人信息 +def update_password(path,username): + tips = input("请选择操作：\n 1.修改邮箱\n 2.修改密码\n") + + # 修改邮箱 + if tips =='1': + new_email = '' + line = [] + try: + with open(path) as rstream: + while True: + line = rstream.readline() + if not line: + break + line = line.split(' ') + # 去掉密码后面的换行符 + line[2] = line[2].rstrip('\n') + if username == line[0]: + new_email = input("请输入新邮箱：") + line[1] = new_email + break + except Exception as err: + print(err) + + else: + # 将新修改邮箱后的用户的所有信息追加到文件夹 + with open(path, 'a') as wstream: + for i in range(len(line)): + if i == 0: + # 遍历列表，第一个列表元素需要前面需要加换行，后面需要加空格与其他元素分割 + line[i] = '\n' + line[i] + ' ' + else: + line[i] = line[i] + ' ' + wstream.writelines(line) + print("修改成功") + # 删除修改邮箱之前用户的信息 + with open(path) as rstream: + # 读取多行 + lines = rstream.readlines() + i = 0 + l = len(lines) + while i < l: + # 当 当前用户名在用户信息行且新的邮箱不在时就删除之前的用户信息，不会删除其他用户的信息 + if username in lines[i] and new_email not in lines[i]: + lines.remove(lines[i]) + i += 1 + l -= 1 + # 删除旧邮箱对应的当前用户信息后，再将新邮箱对应的用户信息以及其他用户的信息从新写入到文件 + with open(path, 'w') as wstream: + wstream.writelines(lines) + + + + + + # 修改密码 + elif tips =='2': + new_password = '' + line = [] + try: + with open(path) as rstream: + while True: + line = rstream.readline() + if not line: + break + line = line.split(' ') + # 去掉密码后面的换行符 + line[2] = line[2].rstrip('\n') + if username == line[0]: + new_password = input("请输入新密码：") + # 判断新密码与旧密码是否一致 + if new_password ==line[2]: + # 抛出异常 + raise Exception("新密码不能与旧密码相同哦~") + else: + line[2] =new_password + break + # 可以捕获到前面raise抛出的异常 + except Exception as err: + print(err) + + else: + # 将新修改密码后的用户的所有信息追加到文件夹 + with open(path,'a') as wstream: + for i in range(len(line)): + if i ==0: + # 遍历列表，第一个列表元素需要前面需要加换行，后面需要加空格与其他元素分割 + line[i] = '\n'+line[i]+' ' + else: + line[i] = line[i] +' ' + wstream.writelines(line) + print("修改成功") + # 删除修改密码之前用户的信息 + with open(path) as rstream: + # 读取多行 + lines = rstream.readlines() + i =0 + l = len(lines) + while i < l: + # 当 当前用户名在用户信息行且新的密码不在时就删除之前的用户信息，不会删除其他用户的信息 + if username in lines[i] and new_password not in lines[i]: + lines.remove(lines[i]) + i+=1 + l-=1 + # 删除旧密码对应的当前用户信息后，再将新密码对应的用户信息以及其他用户的信息从新写入到文件 + with open(path,'w') as wstream: + wstream.writelines(lines) + + + +#管理员授权 +def permission(user_path,username): + try: + with open(user_path) as rstream: + while True: + line = rstream.readline() + # 读到空行跳出循环 + if not line: + break + # 通过3个空格将字符串line分割为一个列表，存储三个值 + line = line.split(' ') + # 循环遍历列表，去掉列表中每个元素后面的换行，如果有就去掉，没有就不取掉 + for i in range(len(line)): + line[i] = line[i].rstrip('\n') + # 判断是否管理员，如果是就可以进行添加操作 + if username == 'admin123': + pass + else: + print("只有管理员{}才可以进行该操作~".format(username)) + # 不是管理员将回到操作页面 + operate(path, username) + except Exception as err: + print("错误原因：",err) + + +book_path = r'C:\Users\酒骁\Desktop\book.txt' +#book_list = ['水浒传\n','红楼梦\n','廊桥遗梦'] + + +#个人信息 +def person_information(path, username): + tips = input("请选择操作：\n 1.查看个人信息\n 2.修改个人信息\n") + if tips == '1': + look_person_info(path, username) + elif tips == '2': + update_password(path, username) + + + +#图书增删改借还操作 +def operate(b_path,username): + while True: + msg = input("请选择操作:\n 1.查询图书\n 2.添加图书\n 3.修改图书\n 4.删除图书\n 5.借书\n 6.归还图书\n 7.个人信息\n 8.退出登录\n") + # 查询图书 + if msg =='1': + find_books(book_path) + # 添加图书 + elif msg =='2': + add_book(b_path, username) + # 修改图书 + elif msg =='3': + update_book(b_path, username) + # 删除图书 + elif msg =='4': + del_book(b_path, username) + # 借书 + elif msg =='5': + borrow_book(username) + # 还书 + elif msg =='6': + return_book(username) + #查看、修改个人信息 + elif msg =='7': + person_information(path, username) + + # 退出登录 + elif msg =='8': + msg = input("确定退出登录吗？(yes/no)") + if msg == 'yes': + break + + + +login()