# -*- coding: utf-8 -*-
"""
=============================== 
@Time    : 2021/5/18 15:56
@Author  : flora.chen
@FileName: handle_mysql.py
@Software: PyCharm
===============================
"""
import pymysql

class MysqlDB:
    """
    操作mysql數(shù)據(jù)庫(kù)
    """
    def __init__(self, host, user, pwd, database=None, port=3306):
        """
        初始化數(shù)據(jù)庫(kù)鏈接
        :param host: 主機(jī)地址
        :param user: 用戶(hù)名
        :param pwd: 密碼
        :param database: 數(shù)據(jù)庫(kù)名稱(chēng)，默認(rèn)為空
        :param port: 端口號(hào)，默認(rèn)3306
        """
        self.conn = pymysql.connect(
            host=host,
            user=user,
            password=pwd,
            database=database,
            port=port,
            cursorclass=pymysql.cursors.DictCursor
        )
        # 創(chuàng)建一個(gè)游標(biāo)對(duì)象
        self.cur = self.conn.cursor()

    def update(self, sql):
        """
        進(jìn)行增刪改操作
        :param sql: 需要執(zhí)行的SQL
        :return:
        """
        # 執(zhí)行SQL
        result = self.cur.execute(sql)
        # 提交事務(wù)
        self.conn.commit()
        return result

    def query(self, sql, one=False):
        """
        進(jìn)行查詢(xún)操作
        :param one: 判斷是要返回所有查詢(xún)數(shù)據(jù)還是第一條，默認(rèn)是所有
        :param sql: 要執(zhí)行的SQL
        :return:
        """
        # 執(zhí)行SQL
        self.cur.execute(sql)
        if one:
            return self.cur.fetchone()
        else:
            return self.cur.fetchall()

    def close(self):
        """
        斷開(kāi)游標(biāo)，關(guān)閉數(shù)據(jù)庫(kù)連接
        :return:
        """
        self.cur.close()
        self.conn.close()


if __name__ == "__main__":
    db = MysqlDB(host="localhost", user="root", pwd="root")
    print(db.query("select * from bookmanage.books"))
    # db.update("insert into bookmanage.books(name, position) value ('python從入門(mén)到放棄', 'A-1-1');")

# -*- coding: utf-8 -*-
"""
=============================== 
@Time    : 2021/5/18 16:39
@Author  : flora.chen
@FileName: bookmanager.py
@Software: PyCharm
===============================
"""
from handle_mysql import MysqlDB

db = MysqlDB(host="localhost", database="bookmanage", user="root", pwd="root")


class BookManage:
    """
    圖書(shū)管理系統(tǒng)
    """

    @staticmethod
    def print_menu():
        """
        菜單打印
        :return:
        """
        print("---------------------菜單-------------------------")
        print("[1]: 添加圖書(shū)")
        print("[2]: 修改圖書(shū)")
        print("[3]: 刪除圖書(shū)")
        print("[4]: 查詢(xún)圖書(shū)")
        print("[5]: 圖書(shū)列表")
        print("[6]: 出借圖書(shū)")
        print("[7]: 歸還圖書(shū)")
        print("[8]: 退出")

    def add_book(self):
        """
        [1]: 添加圖書(shū)
        :return:
        """
        print("****************添加圖書(shū)****************")
        name = input("請(qǐng)輸入書(shū)名：")
        position = input("請(qǐng)輸入圖書(shū)位置：")
        if name and position:
            db.update("insert into books(name, position) value ('{}', '{}');".format(name, position))
            print("圖書(shū)添加成功")
        else:
            print("書(shū)名或者圖書(shū)位置不能為空，請(qǐng)重新輸入！")

        num = input("繼續(xù)添加請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.add_book()

    def update_book(self):
        """
        [2]: 修改圖書(shū)
        :return:
        """
        print("****************修改圖書(shū)****************")
        book_id = input("請(qǐng)輸入需要修改的圖書(shū)ID：")
        result = db.query("select * from books where id={};".format(book_id), one=True)
        if result:
            print("當(dāng)前數(shù)據(jù)為：{}".format(result))
            name = input("重新輸入書(shū)名，不修改輸入回車(chē)：") or result["name"]
            position = input("重新輸入位置，不修改輸入回車(chē)：") or result["position"]
            db.update("update books set name='{}', position='{}' where id={};".format(name, position, book_id))
            print("修改成功")
        else:
            print("您輸入的圖書(shū)ID不存在，請(qǐng)重新輸入~")

        num = input("繼續(xù)修改請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.update_book()

    def delete_book(self):
        """
        [3]: 刪除圖書(shū)
        :return:
        """
        print("****************刪除圖書(shū)****************")
        book_id = input("請(qǐng)輸入需要修改的圖書(shū)ID：")
        result = db.query("select * from books where id={};".format(book_id), one=True)
        if result:
            print("當(dāng)前數(shù)據(jù)為：{}".format(result))
            confirm_num = input("確定需要?jiǎng)h除這本書(shū)嗎？確認(rèn)請(qǐng)按1，取消請(qǐng)按2：")
            if confirm_num == "1":
                db.update("delete from books where id={};".format(book_id))
                print("刪除成功")
            else:
                print("已確認(rèn)不刪除該書(shū)籍~")
        else:
            print("系統(tǒng)中未找到該書(shū)籍！")

        num = input("繼續(xù)刪除請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.delete_book()

    def query_book(self):
        """
        [4]: 查詢(xún)圖書(shū)
        :return:
        """
        print("****************查詢(xún)圖書(shū)****************")
        name = input("請(qǐng)輸入您要查詢(xún)的圖書(shū)名稱(chēng)（模糊匹配）：")
        if name:
            result = db.query("select * from books where name like '%{}%';".format(name))
            if result:
                print("當(dāng)前查詢(xún)到如下書(shū)籍信息：{}".format(result))
            else:
                print("未查詢(xún)到相關(guān)書(shū)籍信息~")
        else:
            print("書(shū)名不能為空！")

        num = input("繼續(xù)查詢(xún)請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.query_book()

    def book_list(self):
        """
        [5]: 圖書(shū)列表
        :return:
        """
        print("****************圖書(shū)列表****************")
        result = db.query("select * from books;")
        for i in result:
            print("編號(hào)：{}， 書(shū)籍名：{}， 位置：{}， 狀態(tài)：{}， 借閱人：{}".format(i["id"], i["name"], i["position"], i["status"],
                                                               i["borrower"]))

    def borrow_book(self):
        """
        [6]: 出借圖書(shū)
        :return:
        """
        print("****************出借圖書(shū)****************")
        book_id = input("請(qǐng)輸入需要借閱的圖書(shū)ID：")
        result = db.query("select * from books where id={};".format(book_id), one=True)
        if result:
            if result["status"] == "出借":
                print("抱歉，該書(shū)已經(jīng)借出！")
            else:
                while True:
                    borrower = input("請(qǐng)輸入借閱者的名字：")
                    if borrower:
                        db.update("update books set borrower='{}' where id={};".format(borrower, book_id))
                        db.update("update books set status='出借' where id={};".format(book_id))
                        print("圖書(shū)借閱成功~")
                        break
                    else:
                        print("借閱者的名字不能為空， 請(qǐng)重新輸入")
        else:
            print("未查詢(xún)到相關(guān)書(shū)籍信息~")

        num = input("繼續(xù)借閱請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.borrow_book()

    def back_book(self):
        """
        [7]: 歸還圖書(shū)
        :return:
        """
        print("****************歸還圖書(shū)****************")
        book_id = input("請(qǐng)輸入需要?dú)w還的圖書(shū)ID：")
        result = db.query("select * from books where id={};".format(book_id), one=True)
        if result:
            if result["status"] == "在庫(kù)":
                print("該書(shū)是在庫(kù)狀態(tài)，請(qǐng)確認(rèn)圖書(shū)編號(hào)是否正確！")
            else:
                db.update("update books set status='在庫(kù)' where id={};".format(book_id))
                db.update("update books set borrower='' where id={};".format(book_id))
                print("書(shū)籍歸還成功~")
        else:
            print("未查詢(xún)到相關(guān)書(shū)籍信息~")

        num = input("繼續(xù)歸還書(shū)籍請(qǐng)輸入1， 回車(chē)退回主菜單")
        if num == "1":
            self.borrow_book()

    def quit(self):
        """
        [8]: 退出
        :return:
        """
        print("****************退出****************")
        db.close()

    def main(self):
        """
        程序運(yùn)行的流程控制
        :return:
        """
        print("---------------歡迎進(jìn)入圖書(shū)管理系統(tǒng)----------------")
        while True:
            self.print_menu()
            num = input("請(qǐng)輸入選項(xiàng)：")
            if num == "1":
                self.add_book()
            elif num == "2":
                self.update_book()
            elif num == "3":
                self.delete_book()
            elif num == "4":
                self.query_book()
            elif num == "5":
                self.book_list()
            elif num == "6":
                self.borrow_book()
            elif num == "7":
                self.back_book()
            elif num == "8":
                self.quit()
                break
            else:
                print("您的輸入有誤~ 請(qǐng)按照菜單提示輸入，謝謝！")


if __name__ == "__main__":
    book = BookManage()
    book.main()