一、Python封装文件
mysql_server.py
#! /usr/bin/python
# -*- coding: UTF-8 -*-
import pymysql
from dbutils.pooled_db import PooledDB
from common.db.mysql_config import MysqlConfig
"""
pymysql封装总结
https://blog.csdn.net/zhj_1121/article/details/121070412
python操作mysql之只看这篇就够了
https://www.jianshu.com/p/4e72faebd27f
关于PooledDB使用autocommit的方法
https://blog.51cto.com/abyss/1736844
"""
class MysqlPool:
"""
MySQL 数据库连接池类 配置变量
"""
'''
:param
reset: how connections should be reset when returned to the pool
(False or None to rollback transcations started with begin(),
True to always issue a rollback for safety's sake)
:param
setsession: optional list of SQL commands that may serve to prepare
the session, e.g. ["set datestyle to ...", "set time zone ..."]
'''
'''
https://blog.51cto.com/abyss/1736844
其中的
setsession=['SET AUTOCOMMIT = 1']
就是用来设置线程池是否打开自动更新的配置，0为False，1为True
'''
# 初始化数据库连接池变量
__pool = None
# 创建连接池的最大数量
__MAX_CONNECTIONS = 100
# 连接池中空闲连接的初始数量
__MIN_CACHED = 10
# 连接池中空闲连接的最大数量
__MAX_CACHED = 20
# 共享连接的最大数量
__MAX_SHARED = 10
# 超过最大连接数量时候的表现，为True等待连接数量下降，为false直接报错处理
__BLOCK = True
# 单个连接的最大重复使用次数
__MAX_USAGE = 100
# 当返回到池时，连接应该如何重置
# (False或None回滚以begin()开始的事务，为了安全起见，总是发出回滚)
__RESET = True
# 设置自动提交
__SET_SESSION = ['SET AUTOCOMMIT = 1']
# 不能是 UTF-8
__CHARSET = 'UTF8'
def __init__(self, host, port, user, password, database):
"""
:param host: 数据库主机地址
:param port: 端口号
:param user: 用户名
:param password: 密码
:param database: 数据库名
"""
if not self.__pool:
# self代表当前类的实例，即为 MysqlPool（） 带小括号，执行后的数据。
# __class__，魔法函数，代表从当前类的实例中，获取当前类，即为 MysqlPool 不带小括号的类。
# __pool，这个代表的事类的变量，即为在类下面创建的初始化连接池，__pool
self.__class__.__pool = PooledDB(
creator=pymysql,
host=host,
port=port,
user=user,
password=password,
database=database,
maxconnections=self.__MAX_CONNECTIONS,
mincached=self.__MIN_CACHED,
maxcached=self.__MAX_CACHED,
maxshared=self.__MAX_SHARED,
blocking=self.__BLOCK,
maxusage=self.__MAX_USAGE,
setsession=self.__SET_SESSION,
reset=self.__RESET,
charset=self.__CHARSET
)
def get_connect(self):
return self.__pool.connection()
class MysqlCursor:
"""
从数据库配置环境，取出数据库配置参数
这里的参数，可以不从外部导入，直接手动写入也可以。
"""
host = MysqlConfig().current_config()[0]
port = MysqlConfig().current_config()[1]
user = MysqlConfig().current_config()[2]
password = MysqlConfig().current_config()[3]
database = MysqlConfig().current_config()[4]
def __init__(self, host=host, port=port, user=user, password=password, database=database) -> None:
"""
:param host: 数据库主机地址
:param port: 端口号
:param user: 用户名
:param password: 密码
:param database: 数据库名
"""
self.__host = host
self.__port = port
self.__user = user
self.__password = password
self.__database = database
# 初始化数据库连接池
self.connects_pool = MysqlPool(
host=self.__host,
port=self.__port,
user=self.__user,
password=self.__password,
database=self.__database
)
def __enter__(self):
"""
# with 上下文管理，魔法函数，进入with时调用
:return: 当前类
"""
# 从数据库链接池，获取一个数据库链接
connect = self.connects_pool.get_connect()
# 从获取的数据库链接，获取一个光标
cursor = connect.cursor(pymysql.cursors.DictCursor)
'''
# https://blog.51cto.com/abyss/1736844
# 如果使用连接池 则不能在取出后设置 而应该在创建线程池时设置
# connect.autocommit = False
'''
# 将数据库链接，赋值给当前类，方便__exit__函数调用
self._connect = connect
# 将数据库光标，赋值给当前类，方便__exit__函数调用
self._cursor = cursor
# __enter__函数，必须返回当前类
return self
def __exit__(self, *exc_info):
"""
# with 上下文管理，魔法函数，退出with时调用
:param exc_info: 异常信息，元祖
:return: None
"""
# 退出with上下文时，使用当前类链接，提交数据库语句
self._connect.commit()
# 关闭光标
self._cursor.close()
# 关闭链接
self._connect.close()
@property
def cursor(self):
"""
数据库连接池，取出链接，取出光标，转换为光标属性
:return: 数据库连接池的光标
"""
return self._cursor
if __name__ == "__main__":
with MysqlCursor() as db:
# 获取数据库的方法
sql = 'select count(id) as total from people'
db.cursor.execute("select count(id) as total from people")
data = db.cursor.fetchone()
print('--------统计数据条数', data)
网友评论