pymysql操作mysql封装

Posted

衣带渐宽终不悔,为伊消得人憔悴

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了pymysql操作mysql封装相关的知识,希望对你有一定的参考价值。

 1 import pymysql
 2 from lib.logger import logger
 3 from warnings import filterwarnings
 4 filterwarnings("ignore",category=pymysql.Warning) # 忽略mysql警告信息
 5 
 6 class MysqlDb():
 7 
 8     logger = logger
 9 
10     def __init__(self, host, port, user, passwd):
11         # 建立数据库连接
12         self.conn = pymysql.connect(
13             host=host,
14             port=port,
15             user=user,
16             passwd=passwd,
17             charset=utf8
18         )
19         # 通过 cursor() 创建游标对象,并让查询结果以字典格式输出
20         self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)
21 
22     def __del__(self): # 对象资源被释放时触发,在对象即将被删除时的最后操作
23         # 关闭游标
24         self.cur.close()
25         # 关闭数据库连接
26         self.conn.close()
27 
28     def select_db(self,db_name):
29         """
30         选择数据库
31         :return:
32         """
33         try:
34             self.conn.select_db(db_name)
35             self.logger.logger.info("选择 {0} 执行成功".format(db_name))
36         except Exception as e:
37             self.logger.logger.exception("操作出现错误:{0}".format(e))
38 
39     def query_db(self, sql,state="all"):
40         """
41         查询
42         :param sql:sql语句
43         :param state:all查询所有,outhor查询单条
44         :return:
45         """
46         # 使用 execute() 执行sql
47         self.logger.logger.info(sql)
48         self.cur.execute(sql)
49         if state == "all":
50             # 使用 fetchall() 获取查询结果
51             data = self.cur.fetchall()
52         else:
53             data = self.cur.fetchone()
54         return data
55 
56     def execute_db(self, sql):
57         """更新/插入/删除"""
58         try:
59             # 使用 execute() 执行sql
60             self.cur.execute(sql)
61             # 提交事务
62             self.conn.commit()
63             self.logger.logger.info("{0} 执行成功".format(sql))
64         except Exception as e:
65             self.logger.logger.exception("操作出现错误:{}".format(e))
66             # 回滚所有更改
67             self.conn.rollback()
68 
69 if __name__ == __main__:
70     pass

 

以上是关于pymysql操作mysql封装的主要内容,如果未能解决你的问题,请参考以下文章

数据库操作 pymysql 封装处理(中级)

python-pymysql常用功能封装

python-pymysql常用功能封装

Python通过pymysql连接数据库并进行查询和更新SQL方法封装

python自动化之pymysql库连接mysql数据库封装成类

python 操作数据库,代码封装