python實現一個無介面的小型圖書管理系統

# -*- coding: utf-8 -*-

"""

===============================

@Time    : 2021/5/18 15:56

@Author  : flora.chen

@FileName: handle_mysql.py

@Software: PyCharm

===============================

"""

import pymysql

class MysqlDB:

     """

     操作 mysql 資料庫

     """

     def __init__(self, host, user, pwd, database=None, port=3306):

         """

         初始化資料庫連結

         :param host: 主機地址

         :param user: 使用者名稱

         :param pwd: 密碼

         :param database: 資料庫名稱，預設為空

         :param port: 埠號，預設 3306

         """

         self.conn = pymysql.connect(

             host=host,

             user=user,

             password=pwd,

             database=database,

             port=port,

             cursorclass=pymysql.cursors.DictCursor

         )

         # 建立一個遊標物件

         self.cur = self.conn.cursor()

     def update(self, sql):

         """

         進行增刪改操作

         :param sql: 外匯跟單gendan5.com 需要執行的 SQL

         :return:

         """

         # 執行 SQL

         result = self.cur.execute(sql)

         # 提交事務

         self.conn.commit()

         return result

     def query(self, sql, >

         """

         進行查詢操作

         :param one: 判斷是要返回所有查詢資料還是第一條，預設是所有

         :param sql: 要執行的 SQL

         :return:

         """

         # 執行 SQL

         self.cur.execute(sql)

         if one:

             return self.cur.fetchone()

         else:

             return self.cur.fetchall()

     def close(self):

         """

         斷開遊標，關閉資料庫連線

         :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 從入門到放棄 ', 'A-1-1');")