利用Python封裝MySQLHelper類實現數據庫的增刪改查功能

更新時間：2023年06月16日 15:31:10 作者：小小鳥愛吃辣條

Python 連接 MySQL 的方法有很多，常用的有 pymysql 和 mysql-connector-python 兩種庫，本文主要介紹了如何封裝一個MySQLHelper類，實現對數據庫的增刪改查功能，感興趣的可以了解一下

Python 連接 MySQL 的方法有很多，常用的有 pymysql 和 mysql-connector-python 兩種庫。本文介紹使用 pymysql 庫連接 MySQL，并實現基本的增刪改查操作。

在使用 pymysql 前需要先安裝該庫，可使用 pip 命令進行安裝：

pip install pymysql

首先，我們可以封裝一個 MySQLHelper 類來管理數據庫連接和操作：

import pymysql


class MySQLHelper:
    def __init__(self, host, port, user, password, database):
        self.host = host
        self.port = port
        self.user = user
        self.password = password
        self.database = database
        self.conn = pymysql.connect(host=host, port=port, user=user, password=password, database=database)

    def execute(self, sql: str, params: tuple = None):
        """執(zhí)行 SQL 語句，無返回值"""
        cursor = self.conn.cursor()
        cursor.execute(sql, params or ())
        self.conn.commit()
        cursor.close()

    def query_one(self, sql: str, params: tuple = None):
        """查詢并返回單條數據"""
        cursor = self.conn.cursor()
        cursor.execute(sql, params or ())
        result = cursor.fetchone()
        cursor.close()
        return result

    def query_all(self, sql: str, params: tuple = None):
        """查詢并返回所有數據"""
        cursor = self.conn.cursor()
        cursor.execute(sql, params or ())
        result = cursor.fetchall()
        cursor.close()
        return result

    def insert(self, table, data: dict):
        """插入一條數據"""
        fields = ','.join(data.keys())
        values = ','.join(['%s'] * len(data))
        sql = 'INSERT INTO %s (%s) VALUES (%s)' % (table, fields, values)
        self.execute(sql, tuple(data.values()))

    def update(self, table, data: dict, where: str = '1=2', where_params: tuple = None):
        """更新數據"""
        set_values = ','.join(['%s=%s' % (field, '%s') for field in data.keys()])
        sql = 'UPDATE %s SET %s WHERE %s' % (table, set_values, where)
        params = tuple(data.values())
        if where_params:
            params += where_params
        self.execute(sql, params)

    def delete(self, table, where: str = '1=2', where_params: tuple = None):
        """刪除數據"""
        sql = 'DELETE FROM %s WHERE %s' % (table, where)
        if where_params:
            self.execute(sql, where_params)
        else:
            self.execute(sql)

    def close(self):
        """關閉數據庫連接"""
        self.conn.close()

以上 MySQLHelper 類封裝了基本的增刪改查操作，說明如下：

execute 方法用于執(zhí)行 SQL 語句，無返回值；
query_one 方法用于查詢并返回單條數據；
query_all 方法用于查詢并返回所有數據；
insert 方法用于插入一條數據；
update 方法用于更新數據；
delete 方法用于刪除數據；
close 方法用于關閉數據庫連接。

在使用 MySQLHelper 類時，需要先實例化一個對象，并傳遞數據庫連接參數。以下是一個簡單的示例：

if __name__ == '__main__':
    helper = MySQLHelper('localhost', 3306, 'root', 'your_password', 'mydatabase')
    helper.insert('mytable', {'column1': 'value1', 'column2': 'value2'})
    helper.update('mytable', {'column1': 'new_value'}, 'id=%s', (1,))
    helper.delete('mytable', 'id=%s', (1,))
    result = helper.query_one('SELECT column1 FROM mytable WHERE id=%s', (1,))
    helper.close()

以上示例代碼使用 MySQLHelper 連接到名為 mydatabase 的數據庫，并操作名為 mytable 的表。具體實現了數據庫的增、刪、改、查等功能。執(zhí)行完畢后，還需調用 close 方法關閉數據庫連接。

總體而言，使用 Python 和 pymysql 庫連接和操作 MySQL 數據庫非常簡單方便，可以輕松地完成各種數據庫操作任務。

到此這篇關于利用Python封裝MySQLHelper類實現數據庫的增刪改查功能的文章就介紹到這了,更多相關Python數據庫增刪改查內容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: