封装Python连接MySQL数据库为工具类

import pymysql


class DBUtil:
    __conn = None
    __cursor = None

    @classmethod
    def __get_conn(cls):  # 获取连接
        if cls.__conn == None:
            cls.__conn = pymysql.connect(host="localhost",
                                         port=3306,
                                         user="root",
                                         password="123456",
                                         database="books")
            return cls.__conn

    @classmethod
    def __get_cursor(cls):  # 获取游标
        if cls.__cursor == None:
            cls.__cursor = cls.__get_conn().cursor()
            return cls.__cursor

    @classmethod
    def __close_cursor(cls):  # 关闭游标
        if cls.__cursor != None:
            cls.__cursor.close()
            cls.__cursor = None

    @classmethod
    def __close_conn(cls):  # 关闭连接
        if cls.__conn != None:
            cls.__conn.close()
            cls.__conn = None

    @classmethod
    def mysql(cls, sql_str):  # 程序入口
        try:
            cursor = cls.__get_cursor()
            cursor.execute(sql_str)
            str = sql_str.split(" ")[0]
            if str == "select":
                return cursor.fetchall()
            else:
                cls.__conn.commit()
                return cursor.rowcount
        except Exception as e:
            cls.__conn.rollback()
            print(e)
        finally:
            cls.__close_cursor()
            cls.__close_conn()

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值