【数据库】pymysql数据库事务操作

封装代码骨架:

  1. 创建连接
  2. 创建游标
  3. 执行sql
try:
	# 获取游标对象
	# 调用游标对象
	# 如果是 查询:
		#返回所有数据
	# 否则:
		# 提交事务
		# 返回受影响的行数
except:
	# 回滚事务
	# 抛出异常
finally:
	# 关闭游标
	# 关闭连接
  1. 关闭游标
  2. 关闭连接

封装代码骨架如下:

# 导包
import pymysql

# 创建工具类
class DBUtil():

    # 创建连接
    @classmethod
    def __get_conn(cls):
        pass

    # 获取游标
    @classmethod
    def __get_cursor(cls):
        pass
    # 执行sql
    @classmethod
    def exe_sql(cls, sql):
        try:
            pass
        except Exception as e:
            pass
        finally:
            pass

    # 关闭游标
    @classmethod
    def __close_cursor(cls):
        pass

    # 关闭连接
    @classmethod
    def __close_conn(cls):
        pass

创建连接方法完整代码:

# 导包
import pymysql

# 创建工具类
class DBUtil():
    # 初始化
    __conn = None
    __cursor = None

    # 创建连接
    @classmethod
    def __get_conn(cls):
        if cls.__conn is None:
            cls.__conn = pymysql.connect(host="localhost",
                                         port=3307,
                                         user="root",
                                         password="root",
                                         database="books"
                                         )
        return  cls.__conn

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

    # 执行sql
    @classmethod
    def exe_sql(cls, sql):
        try:
            # 获取游标对象
            cursor = cls.__get_cursor()
            # 调用游标对象的execute方法,执行sql
            cursor.execute(sql)
            # 如果是查询
            if sql.split()[0].lower() == "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()

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

    # 关闭连接
    @classmethod
    def __close_conn(cls):
        if cls.__conn:
            cls.__conn.close()
            cls.__conn = None
上一篇:python高级学习笔记Day03-- Ubuntu安装Mysql,navicat,python连接数据库,函数参数,闭包,SQL注入


下一篇:[Snowflake+Python]Python中如何使用Snowflake