python編寫圖書管理系統

# coding = utf-8
import csv
from itertools import islice
import pandas as pd
import os


class Book:
? ? def __init__(self, name, author, recommendation, type, status=0): ?# 對實例屬性進行初始化
? ? ? ? self.name = name
? ? ? ? self.author = author
? ? ? ? self.recommendation = recommendation
? ? ? ? self.status = status
? ? ? ? self.type = type

? ? def __str__(self):
? ? ? ? if self.status == 1:
? ? ? ? ? ? status = '已借出'
? ? ? ? else:
? ? ? ? ? ? status = '未借出'
? ? ? ? return '名稱：《%s》作者：%s 推薦語：%s 類別: %s 狀態：%s ' % (self.name, self.author, self.recommendation, self.type, status)


class FictionBook(Book):
? ? def __init__(self, type='奇幻'):
? ? ? ? self.type = type


class NovelBook(Book):
? ? def __init__(self, type='小說'):
? ? ? ? self.type = type


class JottingsBook(Book):
? ? def __init__(self, type='隨筆'):
? ? ? ? self.type = type


class BookManager:
? ? books = []

? ? def __init__(self):
? ? ? ? f = open('BookInfos.csv', 'r')
? ? ? ? readers = csv.reader(f)
? ? ? ? for row in islice(readers, 1, None):
? ? ? ? ? ? self.books.append(Book(row[0], row[1], row[2], row[3], row[4]))
? ? ? ? f.close()

? ? def menu(self): ?# 顯示選擇菜單，根據不同菜單的選項調用不同的方法
? ? ? ? print('歡迎使用大數據團隊圖書節約系統，每本書都是好書，希望鐘老師的市場越來越好。\n')
? ? ? ? while True:
? ? ? ? ? ? print(' 1.查詢書籍\n 2.添加書籍\n 3.借閱書籍\n 4.歸還書籍\n 5.退出系統\n')
? ? ? ? ? ? choice = int(input('請輸入數字選項對應的功能，例如：1（指....）'))
? ? ? ? ? ? if choice == 1:
? ? ? ? ? ? ? ? self.show_all_book()
? ? ? ? ? ? elif choice == 2:
? ? ? ? ? ? ? ? self.add_book()
? ? ? ? ? ? elif choice == 3:
? ? ? ? ? ? ? ? self.lend_book()
? ? ? ? ? ? elif choice == 4:
? ? ? ? ? ? ? ? self.return_book()
? ? ? ? ? ? elif choice == 5:
? ? ? ? ? ? ? ? print('感謝使用！歡迎下次再來！')
? ? ? ? ? ? ? ? break

? ? def show_all_book(self): ?# 顯示書籍信息
? ? ? ? print("1.查詢作者相關書籍\n2.查詢類別相關書籍\n3.查詢所有書籍")
? ? ? ? ch = eval(input('請輸入你的選擇:例如：1（指....）'))
? ? ? ? if ch == 1:
? ? ? ? ? ? author = input("請輸入作者名字：")
? ? ? ? ? ? for book in self.books:
? ? ? ? ? ? ? ? if book.author == author:
? ? ? ? ? ? ? ? ? ? print('')
? ? ? ? ? ? ? ? ? ? print(book)
? ? ? ? elif ch == 2:
? ? ? ? ? ? type = input("請輸入書籍類別：")
? ? ? ? ? ? for book in self.books:
? ? ? ? ? ? ? ? if book.type == type:
? ? ? ? ? ? ? ? ? ? print('')
? ? ? ? ? ? ? ? ? ? print(book)
? ? ? ? elif ch == 3:
? ? ? ? ? ? for book in self.books:
? ? ? ? ? ? ? ? print('')
? ? ? ? ? ? ? ? print(book)
? ? ? ? else:
? ? ? ? ? ? print("您的輸入有誤噢！")

? ? def add_book(self): ?# 添加書籍
? ? ? ? new_name = input('請輸入書籍的名字：')
? ? ? ? new_author = input('請輸入書籍的作者：')
? ? ? ? new_comment = input('請輸入書籍的推薦語：')
? ? ? ? new_type = input('請輸入書籍的類別：')
? ? ? ? new_book = [new_name, new_author, new_comment, new_type, 0]
? ? ? ? # 寫入CSV文件
? ? ? ? f = open('BookInfos.csv', 'a+')
? ? ? ? writer = csv.writer(f)
? ? ? ? writer.writerow(new_book)
? ? ? ? f.close()
? ? ? ? # 改變books[]
? ? ? ? self.books.append(Book(new_book[0], new_book[1], new_book[2], new_book[3], new_book[4]))
? ? ? ? print('書籍錄入成功！\n')

? ? def check_book(self, name): ?# 檢查是否存在該書籍
? ? ? ? for book in self.books:
? ? ? ? ? ? if book.name == name:
? ? ? ? ? ? ? ? return book
? ? ? ? else:
? ? ? ? ? ? return -1

? ? def lend_book(self): ?# 借閱書籍
? ? ? ? name = input('請輸入你想借閱的書籍名字：')
? ? ? ? res = self.check_book(name)
? ? ? ? if res != -1:
? ? ? ? ? ? if res.status == 1:
? ? ? ? ? ? ? ? print('你來晚了，這本書已經被借走了')
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print('借閱成功！借了不看會變胖~~~')
? ? ? ? ? ? ? ? res.status = 1
? ? ? ? ? ? ? ? manager.change_status(name)
? ? ? ? else:
? ? ? ? ? ? print('這本書暫時沒有在系統里！')

? ? def return_book(self): ?# 歸還書籍
? ? ? ? name = input('請輸入歸還書籍的名字：')
? ? ? ? res = self.check_book(name)
? ? ? ? if res == -1:
? ? ? ? ? ? print('沒有這本書，你恐怕輸錯書名了~')
? ? ? ? else:
? ? ? ? ? ? if res.status == 0:
? ? ? ? ? ? ? ? print('這本書沒有被借走！')
? ? ? ? ? ? else:
? ? ? ? ? ? ? ? print('歸還成功')
? ? ? ? ? ? ? ? res.status = 0
? ? ? ? ? ? ? ? manager.change_status(name)
? ? ? ? ? ? ? ? # 這里我做實驗時有些些問題,去掉manager.會出現 name is not defined 學藝不精，還不夠了解python，見諒

? ? @staticmethod
? ? def change_status(name): ?# 修改CSV文件中借還書籍的狀態
? ? ? ? in_file = open("BookInfos.csv", "r")
? ? ? ? reader = csv.reader(in_file)
? ? ? ? out_file = open("BookInfos1.csv", "w", newline='')
? ? ? ? writer = csv.writer(out_file)
? ? ? ? for row in reader:
? ? ? ? ? ? if row[0] == name:

? ? ? ? ? ? ? ? if row[4] == '0':
? ? ? ? ? ? ? ? ? ? row[4] = '1'
? ? ? ? ? ? ? ? else:
? ? ? ? ? ? ? ? ? ? row[4] = '0'
? ? ? ? ? ? writer.writerow(row)
? ? ? ? in_file.close()
? ? ? ? out_file.close()
? ? ? ? os.remove("BookInfos.csv") ?# 刪除原文件
? ? ? ? os.rename("BookInfos1.csv", "BookInfos.csv") ?# 重命名寫入文件，此法不可取，但是我暫時找不到好的方法了
? ? ? ? print("已修改CSV文件")
? ? ? ? # 這里我做實驗時也有些些問題。1.不會修改特定單元格，2.選擇了修改后替換原文件的方法，此法若真實用于大型文件管理，會導致效率大幅度降低，不可取，見諒


manager = BookManager()
manager.menu()