Python交互MySQL -- 有缘人拿去享用

背景

• Python连接MySQL的时候
• 比如使用pymysql的时候
• 前提工作就非常琐碎呢
• 真是烦银呢
• 先要获取连接对象，然后再依靠游标对象来进行SQL *** 作
• 执行SQL语句，得先用游标执行SQL
• 然后调用游标对象的fetchone方法或者fetchall方法获取返回结果
• 最后，还要关闭连接
• 真是烦银呢

缘由

• 基于以上等问题
• 写多了之后这些琐碎的东西后
• 痛定思痛
• 博主决定对其进行简单包装
• 便于快捷使用

代码

# -*- coding: utf-8 -*-
# @Author   : zbz

import pymysql


class OperationMysql:
    def __init__(self, mysql_config: dict):
        self.conn = pymysql.Connection(**mysql_config)
        self.curs = self.conn.cursor()

    def exe_sql(self, sql, args=None, commit=False, value=None):
        try:
            self.conn.ping(reconnect=True)
            self.curs.execute(sql, args=args)
            if commit:
                self.conn.commit()
        except Exception as e:
            self.conn.rollback()
            print("ERROR ==> exe_sql ==> {}".format(e))
            return False
        else:
            if value == 1:
                return self.curs.fetchone()
            elif value == 2:
                return self.curs.fetchall()
            else:
                return True

如何使用

2022-4-9 更新哦