python图书管理系统

class Book(object):
    def __init__(self,name,author,atate=1):
        self.name = name
        self.author = author
        self.state =atate
    def __str__(self):
        return "书名:{} 作者:{} 状态：{}".format(self.name,self.author,self.state)

class BookManage(object):
    books = list()

    def start(self):
        self.books.append(Book("java","山客",1))
        self.books.append(Book("python","鬼叔",1))
        self.books.append(Book("c","吉姆",1))

    def menu(self):
        self.start()
        while True:
            print("""
                    图书管理系统
                    1. 查询图书
                    2. 增加图书
                    3. 借阅图书
                    4. 归还图书
                    5. 退出系统
            """)
            choice = int(input("请输入选择："))
            if choice == 1:
                name = input("书名：")
                self.findBook(name)
            elif choice == 2:
                self.addBook()
            elif choice == 3:
                self.borrowBook()
            elif choice == 4:
                self.returnBook()
            elif choice == 5:
                break
            else:
                print("参数有误，请输入正确的选择！'")

    def findBook(self,name):
        for book in self.books:
            if book.name == name:
                print(book)#输出调用格式化__str__函数输出
                return#执行return后不再执行函数后面的代码，相当于break

        else:
            print("没有找到")
            return None

    def addBook(self):
        name =input("书名:")
        author = input("作者：")
        self.books.append(Book(name,author))
        print("添加图书%s成功"%name)

    def borrowBook(self):
        name = input("借阅书籍名称: ")
        ret = self.findBook(name)
        if ret is not None:
            if ret.state == 0:
                print("书籍%s已经借出"%name)
            elif ret.state == 1:
                print("借阅%s成功:"%name)
                ret.state = 0
        else:
            print('书籍%s不存在' % name)

    def returnBook(self):
        name = input("归还书籍名称:")
        ret = self.findBook(name)
        if ret is not None:
            ret.state = 1
            print("书籍%s归还成功 "%name)
        else:
            print("书籍%s不存在"%name)
bookManage = BookManage()
bookManage.menu()