diff --git a/main.py b/main.py new file mode 100644 index 0000000..1b31683 --- /dev/null +++ b/main.py @@ -0,0 +1,132 @@ +import os + + +# 新增图书信息 +def Add(): + id_book = int(input('请输入编号:')) + name_book = input('请输入书名:') + author_book = input('请输入作者:') + type_book = input('请输入种类:') + num_book = int(input('请输入数量:')) + surplus_book = int(input('请输入剩余:')) + book_information = {'编号': id_book, '书名': name_book, '作者': author_book, '种类': type_book, '数量': num_book, + '剩余': surplus_book} + with open('图书信息.txt', 'a', encoding='utf-8') as f: # 图书信息以字典形式存储,增加信息就用a追加到文件中 + if os.path.getsize('图书信息.txt') != 0: + f.write('\n') # 换行\n保证数据在文件中按行存储,这样在查询时用f.readlines()才能正确查询 + f.write(str(book_information)) + + +# 删除图书信息 +def Delete(): + del_num = int(input('请输入你要删除图书的编号:')) + with open('图书信息.txt', 'r', encoding='utf-8') as f: # 把文件信息全部提取出来 + book_inf = f.readlines() + for i in range(len(book_inf)): # 循环查找想要编号的那一项 + if eval(book_inf[i])['编号'] == del_num: + book_inf.pop(i) # 找到后把那一项删掉 + break + with open('图书信息.txt', 'w', encoding='utf-8') as f: # 把删掉一项信息后的book_inf覆盖存储到文件中 + for temp in book_inf: + f.write(str(temp)) # 存储必须以str类型,否则就报错 + + +# 修改图书信息 +def Alter(): + alt_num = int(input('请输入你要修改图书的编号:')) + with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息 + books_inf = f.readlines() + for temp in books_inf: # 循环遍历出哪一项是想要修改的 + b = eval(temp) + if b['编号'] == alt_num: + print(b) # 列出所选编号的图书信息,好看着修改 + books_inf.remove(temp) + b['编号'] = int(input('请输入编号:')) + b['书名'] = input('请输入书名:') + b['作者'] = input('请输入作者:') + b['种类'] = input('请输入种类:') + b['数量'] = int(input('请输入数量:')) + b['剩余'] = int(input('请输入剩余:')) + break + with open('图书信息.txt', 'w', encoding='utf-8') as f: + f.write(str(b) + '\n') + with open('图书信息.txt', 'a', encoding='utf-8') as f: + for temp in books_inf: + f.write(str(temp)) + + +# 查询图书信息 +def Search(): + sea_num = int(input('请输入你要查找图书的编号:')) + with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息 + books_inf = f.readlines() + for temp in books_inf: # 循环遍历出哪一项是想要查找的 + b = eval(temp) + if b['编号'] == sea_num: + print('编号'.center(5, ' '), '书名'.center(15, ' '), '作者'.center(15, ' '), '种类'.center(15, ' '), + '数量'.center(5, ' '), + '剩余'.center(5, ' ')) + format_data = '{:^5}\t{:^15}\t{:^10}\t{:^10}\t{:^5}\t{:^5}' + print(format_data.format(b.get('编号'), b.get('书名'), b.get('作者'), b.get('种类'), b.get('数量'), + b.get('剩余'))) + + +# 展示所有图书信息 +def Show(): + with open('图书信息.txt', 'r', encoding='utf-8') as f: # 取所有图书信息 + books_inf = f.readlines() + print('编号'.center(5, ' '), '书名'.center(15, ' '), '作者'.center(15, ' '), '种类'.center(15, ' '), + '数量'.center(5, ' '), + '剩余'.center(5, ' ')) + format_data = '{:^5}\t{:^15}\t{:^10}\t{:^10}\t{:^5}\t{:^5}' + for temp in books_inf: + b = eval(temp) + print(format_data.format(b.get('编号'), b.get('书名'), b.get('作者'), b.get('种类'), b.get('数量'), + b.get('剩余'))) + + +# 选择菜单 +def Menu(): + print('-----------------------------功能菜单-----------------------------') + print('\t\t\t\t1.新增图书信息') + print('\t\t\t\t2.删除图书信息') + print('\t\t\t\t3.修改图书信息') + print('\t\t\t\t4.查询图书信息') + print('\t\t\t\t5.展示所有图书信息') + print('\t\t\t\t0.退出') + print('----------------------------------------------------------------') + + +# 主程序 +def main(): + print('=========================资料室图书管理系统=========================') + while True: + Menu() + select = eval(input('请选择功能:')) + if select == 1: + Add() + print('新增信息完成!') + print('\n') + elif select == 2: + Delete() + print('删除信息完成!') + print('\n') + elif select == 3: + Alter() + print('修改信息完成!') + print('\n') + elif select == 4: + Search() + print('\n') + elif select == 5: + Show() + print('\n') + elif select == 0: + print('谢谢使用!') + break + else: + print("输入错误!请重新输入!") + + +if __name__ == '__main__': + main()