Python连接MySQL -- 使用连接池

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


import pymysql
from dbutils.pooled_db import PooledDB


class MysqlConnectionPool:

    # mark为True，查询语句的返回结果为dict
    def __init__(self, cfg, mark=False):
        self.pool = PooledDB(pymysql, **cfg)
        self.mark = mark

    # 获取连接、游标
    def get_conn_curs(self):
        conn = self.pool.connection()
        curs = conn.cursor(pymysql.cursors.DictCursor) if self.mark else conn.cursor()
        return conn, curs

    # 关闭游标、连接
    def close_conn_curs(self, curs, conn):
        curs.close()
        conn.close()

    # 执行sql
    def exe_sql(self, sql, args=None, way=None):
        conn, curs = self.get_conn_curs()
        try:
            curs.execute(sql, args=args)
            conn.commit()
        except Exception as e:
            conn.rollback()
            print("error  >>  exe_sql>  >>  {}".format(e))
            return False
        else:
            if way == 1:
                return curs.fetchone()
            elif way == 2:
                return curs.fetchall()
            else:
                return True
        finally:
            self.close_conn_curs(curs, conn)
            

if __name__ == '__main__':
    cfg = {
        'host': 'localhost',
        'port': 3306,
        'user': 'root',
        'passwd': 'Chinais9?',
        'charset': 'utf8mb4',

        'maxconnections': 10,  # 连接池允许的最大连接数
        'mincached': 2,  	   # 初始化连接池时创建的连接数。默认为0，即初始化时不创建连接
        'maxcached': 0,        # 连接池中空闲连接的最大数量。默认0，即无最大数量限制
        'maxusage': 5,         # 连接的最大使用次数。默认0，即无使用次数限制
        'blocking': True       # 连接数达到最大时，新连接是否可阻塞。默认False，即达到最大连接数时，再取新连接将会报错
    }

    handler = MysqlConnectionPool(cfg)
    result = handler.exe_sql('show databases', way=2)
    print(result)