Python工具類（一）—— 操作Mysql數據庫

 

如何調用直接看__main__函數里如何調用此工具類就闊以啦！

# encoding=utf-8
import pymysql

# 導入所有Mysql配置常量,請自行指定文件
from conf.settings import *


class MysqlConnection(object):
    """
    mysql操作類，對mysql數據庫進行增刪改查
    """

    def __init__(self, config):
        # Connect to the database
        self.connection = pymysql.connect(**config)
        self.connection.autocommit(True)
        self.cursor = self.connection.cursor()

    def QueryAll(self, sql):
        """
        查詢所有數據
        :param sql:
        :return:
        """
        # 數據庫若斷開即重連
        self.reConnect()

        self.cursor.execute(sql)
        return self.cursor.fetchall()

    def QueryMany(self, sql, n):
        """
        查詢某幾條數據數據
        :param sql:
        :return:
        """
        # 數據庫若斷開即重連
        self.reConnect()

        self.cursor.execute(sql)
        return self.cursor.fetchmany(n)

    def QueryOne(self, sql):
        """
        查詢某幾條數據數據
        :param sql:
        :return:
        """
        # 數據庫若斷開即重連
        self.reConnect()

        self.cursor.execute(sql)
        return self.cursor.fetchone()

    # return self.cursor.fetchone()

    def reConnect(self):
        """
        重連機制
        :return:
        """
        try:
            self.connection.ping()
        except:
            self.connection()

    def Operate(self, sql, params=None, DML=True):
        """
        數據庫操作:增刪改查
        DML: insert / update / delete
        DDL: CREATE TABLE/VIEW/INDEX/SYN/CLUSTER
        """
        try:
            # 數據庫若斷開即重連
            self.reConnect()

            with self.connection.cursor() as cursor:
                cursor.execute(sql, params)

                self.connection.commit()

        except Exception as e:
            if DML:
                # 涉及DML操作時,若拋異常需要回滾
                self.connection.rollback()
            print(e)

    def __del__(self):
        """
        MysqlConnection實例對象被釋放時調用此方法,用於關閉cursor和connection連接
        """
        self.cursor.close()
        self.connection.close()


if __name__ == "__main__":
    # 初始化MysqlConnection實例對象需要傳Mysql配置信息的字典
    config = {'host': MYSQL_HOST, 'charset': CHARSET, 'db': DB, 'user': USER, 'port': MYSQL_PORT, 'password': PASSWORD}
    msc = MysqlConnection(config)
    sql = "delete from users where username ='%s'" % "123456"

    print(msc.Operate(sql))