python3 mysql操作封装_Python3操作MySQL基于PyMySQL封装的类

Python3操作MySQL基于PyMySQL封装的类

在未使用操作数据库的框架开发项目的时候，我们需要自己处理数据库连接问题，今天在做一个Python的演示项目，写一个操作MySQL数据库的类，基于PyMySQL库在Python3上实现。在写业务逻辑代码的时候，可以方便很多，时间关系，没有写太完善，只写了常用的操作。

直接上代码：

#!/usr/bin/env python

# -*- coding: utf-8 -*-

'Python连接到 MySQL 数据库及相关操作(基于Python3)'

import pymysql.cursors

class Database:

""" Python连接到 MySQL 数据库及相关操作 """

"""

conf: 类参数，数据库的连接参数配置字典，含host、port、user、pw、db、charset(可选,默认utf8)

connected: 属性，True数据库连接成功，False连接失败

insert(self, table, val_obj): 方法，插入数据到数据表

table: 数据表名称

val_obj: 待插入数据的字段名和值的键值对字典

返回: 成功则返回新插入数据的主键ID，失败返回False

update(self, table, val_obj, range_str): 方法，更新数据表中的数据

table: 数据表名称

val_obj: 待更新数据的字段名和值的键值对字典

range_str: 更新范围的条件语句字符串

返回: 成功返回更新的行数，失败返回False

delete(self, table, range_str): 方法，在数据表中删除数据

table: 数据表名称

range_str: 删除范围的条件语句字符串

返回: 成功返回删除的行数，失败返回False

select_one(self, table, factor_str, field='*'): 方法，查询表中符合条件唯一的一条数据

table: 数据表名称

factor_str: 查询唯一条件语句字符串

field: 查询结果返回哪些字段，多个用逗号分隔，可选参数，默认返回所有字段

返回: 成功返回一条数据的字段名与值的一维字典，失败返回False

select_more(self, table, range_str, field='*'): 方法，查询表中符合条件的所有数据

table: 数据表名称

range_str: 查询条件语句字符串

field: 查询结果返回哪些字段，多个用逗号分隔，可选参数，默认返回所有字段

返回: 成功返回多条数据的字段名与值的二维字典，失败返回False

count(self, table, range_str='1'): 方法，统计数据表中符合条件的总函数

table: 数据表名称

range_str: 查询条件语句字符串，可选参数，默认表中所有行数

返回: 成功返回符合条件的行数，失败返回False

sum(self, table, field, range_str='1'): 方法，对数据表中某数值类型字段求和

table: 数据表名称

field: 需要求和的字段，可以是多个字段的计算公式

range_str: 需要求和的条件语句字符串，可选参数，默认表中所有行

返回: 成功返回求和结果，失败返回False

close(self): 方法，关闭数据库连接，对象销毁时也会自动关闭，所以多数时候不用特意调用

"""

connected = False

__conn = None

# 构造函数，初始化时直接连接数据库

def __init__(self, conf):

if type(conf) is not dict:

print('错误: 参数不是字典类型！')

else:

for key in ['host', 'port', 'user', 'pw', 'db']:

if key not in conf.keys():

print('错误: 参数字典缺少 %s' % key)

if 'charset' not in conf.keys():

conf['charset'] = 'utf8'

try:

self.__conn = pymysql.connect(

host=conf['host'],

port=conf['port'],

user=conf['user'],

passwd=conf['pw'],

db=conf['db'],

charset=conf['charset'],

cursorclass=pymysql.cursors.DictCursor)

self.connected = True

except pymysql.Error as e:

print('数据库连接失败:', end='')

# 插入数据到数据表

def insert(self, table, val_obj):

sql_top = 'INSERT INTO ' + table + ' ('

sql_tail = ') VALUES ('

try:

for key, val in val_obj.items():

sql_top += key + ','

sql_tail += val + ','

sql = sql_top[:-1] + sql_tail[:-1] + ')'

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return self.__conn.insert_id()

except pymysql.Error as e:

self.__conn.rollback()

return False

# 更新数据到数据表

def update(self, table, val_obj, range_str):

sql = 'UPDATE ' + table + ' SET '

try:

for key, val in val_obj.items():

sql += key + '=' + val + ','

sql = sql[:-1] + ' WHERE ' + range_str

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.rowcount

except pymysql.Error as e:

self.__conn.rollback()

return False

# 删除数据在数据表中

def delete(self, table, range_str):

sql = 'DELETE FROM ' + table + ' WHERE ' + range_str

try:

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.rowcount

except pymysql.Error as e:

self.__conn.rollback()

return False

# 查询唯一数据在数据表中

def select_one(self, table, factor_str, field='*'):

sql = 'SELECT ' + field + ' FROM ' + table + ' WHERE ' + factor_str

try:

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.fetchall()[0]

except pymysql.Error as e:

return False

# 查询多条数据在数据表中

def select_more(self, table, range_str, field='*'):

sql = 'SELECT ' + field + ' FROM ' + table + ' WHERE ' + range_str

try:

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.fetchall()

except pymysql.Error as e:

return False

# 统计某表某条件下的总行数

def count(self, table, range_str='1'):

sql = 'SELECT count(*)res FROM ' + table + ' WHERE ' + range_str

try:

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.fetchall()[0]['res']

except pymysql.Error as e:

return False

# 统计某字段（或字段计算公式）的合计值

def sum(self, table, field, range_str='1'):

sql = 'SELECT SUM(' + field + ') AS res FROM ' + table + ' WHERE ' + range_str

try:

with self.__conn.cursor() as cursor:

cursor.execute(sql)

self.__conn.commit()

return cursor.fetchall()[0]['res']

except pymysql.Error as e:

return False

# 销毁对象时关闭数据库连接

def __del__(self):

try:

self.__conn.close()

except pymysql.Error as e:

pass

# 关闭数据库连接

def close(self):

self.__del__()