From 03f5bf6f029ca252bf37926b30fe42936f614d3a Mon Sep 17 00:00:00 2001
From: p7sng5buz <prs252964695@163.com>
Date: Mon, 10 Jun 2024 16:16:24 +0800
Subject: [PATCH] ADD file via upload

---
 main.py | 132 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 132 insertions(+)
 create mode 100644 main.py

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()