import pymysql
class MysqlConnection(object):
def __init__(self, config):
self.connection = pymysql.connect(**config)
self.connection.autocommit(True)
self.cursor = self.connection.cursor()
def reConnect(self):
"""
重连机制
:return:
"""
try:
self.connection.ping()
except:
self.connection()
def close(self):
"""
断开连接
:return:
"""
self.cursor.close()
self.connection.close()
def __del__(self):
self.close()
def queryAll(self, sql):
"""
查询所有数据
:param sql: sql语句
:return:
"""
self.reConnect()
try:
self.cursor.execute(sql)
return self.cursor.fetchall()
except Exception as e:
return e
def queryMany(self, sql, n):
"""
查询几条数据
:param sql:
:return:
"""
self.reConnect()
try:
self.cursor.execute(sql)
return self.cursor.fetchmany(n)
except Exception as e:
return e
def queryOne(self, sql):
"""
查询一条记录
:param sql:
:return:
"""
self.reConnect()
try:
self.cursor.execute(sql)
return self.cursor.fetchone()
except Exception as e:
return e
def operate(self, sql, params=None, DML=True):
"""
"""
self.reConnect()
try:
with self.connection.cursor() as cursor:
cursor.execute(sql, params)
self.connection.commit()
return 0
except Exception as e:
if DML:
self.connection.rollback()
return e
# 举个例子
if __name__ == "__main__":
config = {
'host':'192.168.153.32',
'port':3306,
'user':'root',
'password':'123456',
'db':'webapp'
}
msc = MysqlConnection(config)
sql = "select * from students"
print(msc.queryAll(sql))