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