import pymysql
from warningsimport filterwarnings
# 忽略MySql告警信息
filterwarnings("ignore", category=pymysql.Warning)
class MysqlDb:
def __init__(self):
# 建立数据库连接
self.conn = pymysql.connect(host="127.0.0.1", port=3306, user="root", passwd="123456", db="case_demo",
charset="utf8")
# 使用cursor方法获取操作游标，得到一个可以执行的sql语句，并且操作结果作为字典返回的游标
self.cursor =self.conn.cursor(cursor=pymysql.cursors.DictCursor)
def __del__(self):
# 关闭游标
self.cursor.close()
# 关闭连接
self.conn.close()
def query(self, sql, state="all"):
"""
查询
:paramsql:
:param state: all默认查询全部
:return:
"""
self.cursor.execute(sql)
if state =="all":
# 查询全部
data =self.cursor.fetchall()
else:
# 查询单条
data =self.cursor.fetchone()
return data
def execute(self, sql):
"""
更新、删除、新增
:paramsql:
:return:
"""
try:
# 使用execute操作sql
rows =self.cursor.execute(sql)
# 提交事务
self.conn.commit()
return rows
except Exception as e:
print(f"数据库操作异常{e}")
# 回滚修改
self.conn.rollback()
if __name__ =='__main__':
mydb = MysqlDb()
# r = mydb.query(sql="select * from config", state="all")
r=mydb
print(r)
