# -*- coding: utf-8 -*-
"""
@Time ： 2023/4/6 10:09
@Auth ： otnios@qq.com
@File ： books_system.py
"""
import os# 图书管理系统
books_path = r'E:\\a1\\books_system\\books.txt'
users_path = r'E:\\a1\\books_system\\users.txt'
user_book_path = r'E:\\a1\\books_system\\user_book.txt'def dict_user(src):"""传入一个用户路径 将文件内容以列表中的字典形式返回:param src: 用户文件路径:return: list"""with open(src, encoding='utf-8') as rStream:user_list = []# 以列表形式拿到文件内容src_list = rStream.readlines()for user in src_list:one_user = user.split(' ')# print(one_user)user_list.append({'username': one_user[0],'password': one_user[1]})return user_list# 注册功能
def register():print('--------开始会员注册--------')username = input('请输入会员名:')password = input('请输入密码:')re_password = input('请确认密码:')if password == re_password:# 先判断用户名是否存在 如果存在则注册失败user_list = dict_user(users_path)for user in user_list: # user = {'username': 'daming', 'password': '123456\\n'},if username == user.get('username'):print('用户名已被注册,注册失败!')breakelse:# 两次密码相同 用户名不存在则写入文件with open(users_path, 'a', encoding='utf-8') as wStream:wStream.write(f'{username} {password}\\n')print(f'用户<{username}>注册成功!')else:print('两次输入密码不一致,注册失败!')# 登录功能
def login():print('--------开始会员登录--------')user_name = input('请输入会员名:')pass_word = input('请输入密码:')user_list = dict_user(users_path)# 开始判断用户名与密码是否匹配文件for user in user_list:if user_name == user.get('username'): # user = {'username': 'daming', 'password': '123456\\n'},if pass_word == user.get('password').strip('\\n'):print('登录成功!')return user_name, pass_wordbreakelse:print('密码错误,请重新输入密码!')breakelse:print('会员账号未注册,请先注册!')# 查询所有图书
def select_books():with open(books_path, encoding='utf-8') as rStream:books_list = rStream.readlines()print('--------图书馆藏书有--------')for book in books_list:print(book, end='')# 添加书本
def add_book(book_name):add_book_name = f'{book_name} \\n'with open(books_path, 'a', encoding='utf-8') as aStream:aStream.write(add_book_name)print(f'<{book_name}>添加完毕!')# 添加用户借书记录user_login_info = []
username = ''
password = ''
user_borrow_books = []
borrow_list = []
while True:print('--------欢迎来到图书管理系统--------')choice1 = input('1-会员注册 2-会员登录 3-借书功能 ''4-查询藏书 5-借书情况 6-退出系统:')if choice1 == '1':register()elif choice1 == '2':# 如果完成登录则将登录信息保存username, password = login()user_login_info.append(username)user_login_info.append(password)# print(user_login_info[0], user_login_info[1])elif choice1 == '3':if not user_login_info:print('您还未登录,请登录后再进行操作!')else:borrow_book = input(f'用户<{username}>,请选择您要借阅的书:')with open(user_book_path, 'a', encoding='utf-8') as aStream:aStream.write(f'{user_login_info[0]}:{borrow_book}\\n')print('借书成功!')elif choice1 == '4':select_books()elif choice1 == '5':print(f'用户<{username}>当前借书详情如下:')with open(user_book_path, encoding='utf-8') as rStream:user_borrow_list = rStream.readlines()for user_borrow in user_borrow_list:new_borrow = user_borrow.split(':')user_borrow_books.append({'username': new_borrow[0],'bookname': new_borrow[1]})# 根据当前登录username判断所借的书for borrow_info in user_borrow_books:if username == borrow_info.get('username'):borrow_list.append(borrow_info.get('bookname'))print(borrow_list)elif choice1 == '6':print('正在退出系统,请稍后...')breakelse:print('指令输入错误,请重新输入!')
