38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
#!/usr/bin/python
|
|
# -*- coding: UTF-8 -*-
|
|
import config
|
|
import pymysql
|
|
from DBUtils.PooledDB import PooledDB
|
|
|
|
class mysql_obj():
|
|
pool = None
|
|
|
|
def __init__(self, config_temp=None):
|
|
if config_temp is None:
|
|
configs = {
|
|
'host': config.mysql_host,
|
|
'port': config.mysql_port,
|
|
'user': config.mysql_user,
|
|
'password': config.mysql_pass,
|
|
'db': config.mysql_db,
|
|
'charset': config.mysql_charset,
|
|
'maxconnections': 70, # 连接池最大连接数量
|
|
}
|
|
else:
|
|
configs = config_temp
|
|
configs['creator'] = pymysql
|
|
configs['cursorclass'] = pymysql.cursors.DictCursor
|
|
self.pool = PooledDB(**configs)
|
|
|
|
def __enter__(self):
|
|
self.conn = self.pool.connection()
|
|
self.cursor = self.conn.cursor()
|
|
return self
|
|
|
|
def __exit__(self, type, value, trace):
|
|
self.cursor.close()
|
|
self.conn.close()
|
|
|
|
if __name__ == '__main__':
|
|
|
|
pass |