Config.py
DBHOST = ""
DBPORT = 3306
DBUSER = "root"
DBPWD = ""
DBCHAR = "utf8"
DBNAME = ""
mysql_DB.py
#!/usr/bin/env python
# -*- coding:utf-8 -*-
import re
import pymysql, os, configparser
from pymysql.cursors import DictCursor
from DBUtils.PooledDB import PooledDB
import Config
'''
在配置的文件中进行配置 Config.py
DBHOST = ""
DBPORT = 3306
DBUSER = "root"
DBPWD = ""
DBNAME = ""
DBCHAR = "utf8"
'''
class BasePymysqlPool(object):
def __init__(self, host, port, user, password, db_name=None):
self.db_host = host
self.db_port = int(port)
self.user = user
self.password = str(password)
self.db = db_name
self.conn = None
self.cursor = None
class MyPymysqlPool(BasePymysqlPool):
"""
MYSQL数据库对象,负责产生数据库连接 , 此类中的连接采用连接池实现获取连接对象:conn = Mysql.getConn()
释放连接对象;conn.close()或del conn
"""
# 连接池对象
__pool = None
def __init__(self, conf_name=None):
# self.conf = Config().get_content(conf_name)
self.conf = {
'host': Config.DBHOST,
'port': Config.DBPORT,
'user': Config.DBUSER,
'password': Config.DBPWD,
'db_name': Config.DBNAME
}
super(MyPymysqlPool, self).__init__(**self.conf)
# 数据库构造函数,从连接池中取出连接,并生成操作游标
self._conn = self.__getConn()
self._cursor = self._conn.cursor()
def __getConn(self):
"""
@summary: 静态方法,从连接池中取出连接
@return MySQLdb.connection
"""
if MyPymysqlPool.__pool is None:
MyPymysqlPool.__pool = PooledDB(creator=pymysql,
mincached=1,
maxcached=20,
host=self.db_host,
port=self.db_port,
user=self.user,
passwd=self.password,
db=self.db,
use_unicode=False,
charset="utf8",
cursorclass=DictCursor)
return MyPymysqlPool.__pool.connection()
def getAll(self, sql, param=None):
"""
@summary: 执行查询,并取出所有结果集
@param sql:查询SQL,如果有查询条件,请只指定条件列表,并将条件值使用参数[param]传递进来
@param param: 可选参数,条件列表值(元组/列表)
@return: result list(字典对象)/boolean 查询到的结果集
"""
self.start()
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchall()
else:
result = False
self.dispose()
return result, count
def getOne(self, sql, param=None):
"""
@summary: 执行查询,并取出第一条
@param sql:查询SQL,如果有查询条件,请只指定条件列表,并将条件值使用参数[param]传递进来
@param param: 可选参数,条件列表值(元组/列表)
@return: result list/boolean 查询到的结果集
"""
self.start()
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchone()
else:
result = False
self.dispose()
return result
def getMany(self, sql, num, param=None):
"""
@summary: 执行查询,并取出num条结果
@param sql:查询SQL,如果有查询条件,请只指定条件列表,并将条件值使用参数[param]传递进来
@param num:取得的结果条数
@param param: 可选参数,条件列表值(元组/列表)
@return: result list/boolean 查询到的结果集
"""
self.start()
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
if count > 0:
result = self._cursor.fetchmany(num)
else:
result = False
self.dispose()
return result
def insertMany(self, sql, values):
"""
@summary: 向数据表插入多条记录
@param sql:要插入的SQL格式
@param values:要插入的记录数据tuple(tuple)/list[list]
@return: count 受影响的行数
"""
self.start()
count = self._cursor.executemany(sql, values)
self.dispose()
return count
def __query(self, sql, param=None):
self.start()
if param is None:
count = self._cursor.execute(sql)
else:
count = self._cursor.execute(sql, param)
self.dispose()
return count
def update(self, sql, param=None):
"""
@summary: 更新数据表记录
@param sql: SQL格式及条件,使用(%s,%s)
@param param: 要更新的 值 tuple/list
@return: count 受影响的行数
"""
return self.__query(sql, param)
def insert(self, sql, param=None):
"""
@summary: 更新数据表记录
@param sql: SQL格式及条件,使用(%s,%s)
@param param: 要更新的 值 tuple/list
@return: count 受影响的行数
"""
return self.__query(sql, param)
def delete(self, sql, param=None):
"""
@summary: 删除数据表记录
@param sql: SQL格式及条件,使用(%s,%s)
@param param: 要删除的条件 值 tuple/list
@return: count 受影响的行数
"""
return self.__query(sql, param)
def execute(self,sql,param=None):
return self.__query(sql, param)
def begin(self):
"""
@summary: 开启事务
"""
self._conn.autocommit(0)
def end(self, option='commit'):
"""
@summary: 结束事务
"""
if option == 'commit':
self._conn.commit()
else:
self._conn.rollback()
def start(self):
self._conn = self.__getConn()
self._cursor = self._conn.cursor()
def dispose(self, isEnd=1):
"""
@summary: 释放连接池资源
"""
if isEnd == 1:
self.end('commit')
else:
self.end('rollback')
self._cursor.close()
self._conn.close()
def getDB():
return MyPymysqlPool()
def sqlr(sql, data):
m = re.findall(r'(\$\w+)', sql)
for i in m:
val = "\"%s\"" % eval("data.%s" % i.replace('$', ''))
sql = sql.replace(i, val)
m = re.findall(r'(\#\w+)', sql)
for i in m:
val = eval("data.%s" % i.replace('#', ''))
sql = sql.replace(i, val)
return sql
def sqlk(sql, **pars):
m = re.findall(r'(\$\w+)', sql)
for i in m:
key = i.replace('$', '')
val = "\"%s\"" % pars[key]
sql = sql.replace(i, val)
m = re.findall(r'(\#\w+)', sql)
for i in m:
key = i.replace('#', '')
val = pars[key]
sql = sql.replace(i, val)
return sql
if __name__ == '__main__':
mysql = getDB()
'''
数据库进行修改
'''
sql = 'insert into user(name) value ("test")'
mysql.execute(sql)
sql = 'delete from user where id=2;'
mysql.delete(sql)
id = 2
sql = sqlk("update user set gender=0 where id=$id", id=id)
mysql.update(sql)
sqlAll = "select * from user where gender='%s';" % (0)
result, count = mysql.getAll(sqlAll)
print(result, count)