实现一个简单的DBUtil工具类(Python)

pymysql操作mysql,虽然简单,但每次都要链接数据库,获取cursor,关闭cursor,关闭链接。这些操作无技术含量,还要重复编写!!我们应该想法提高开发效率

下面提供一个简易的DBUtil工具类,帮我们解决代码繁琐问题

import pymysql as pq

class DBUtil:
    #连接你的数据库
    config = {
        "host":"localhost",
        "user":"root",
        "passwd":'123',
        "charset":'utf8',
        "db":"your databaseName"
    }
    def __init__(self):
        self.con = pq.connect()
        self.cursor = self.con.cursor()
    
    def close(self):
        """
        创建connect
        创建cursor
        """
        if self.con:
            self.con.close()
        if self.cursor:
            self.cursor.close()

    def execute_DML(self,sql,*args):
        #执行DML语句
        try:
            self.cursor.execute(sql,args)
            self.con.commit()
        except Exception as e:
            print(e)
        finally:
            self.close()

    def query_one(self,sql,*args):
        """
        查询一条语句
        """
        try:
            self.cursor.execute(sql,args)
            return self.cursor.fetchone()
        except Exception as e:
            print(e)
        finally:
            self.close()
    
    def query_many(self,sql,*args):
        """
        查询多条语句
        """
        try:
            self.cursor.execute(sql,args)
            rs = self.cursor.fetchmany(*args)
        except Exception as e:
            print(e)
        finally:
            self.close()

      
    def query_all(self,sql,*args):
        """
        查询全部语句
        """
        try:
            self.cursor.execute(sql,args)
            rs = self.cursor.fetchall()
        except Exception as e:
            print(e)
        finally:
            self.close()
if __name__ == '__main__':
    db = DBUtil()

  • 8
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值