mysql实现with上下文管理器----python

最新推荐文章于 2024-08-14 22:10:38 发布

转载最新推荐文章于 2024-08-14 22:10:38 发布 · 526 阅读

本文介绍了一种使用pymysql模块在Python中操作数据库的方法。通过定义一个DBOpen类，实现了数据库的连接、操作及自动提交和关闭功能。代码示例展示了如何使用with语句执行SQL查询并获取所有数据。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >


from pymysql import connect
class DBOpen(object):
	# 初始化
	def __init__(self, databasename):
		# 创建Connection连接
		self.conn = connect(host='localhost', port=3306, database=databasename, user='root', password='mysql',
		                    charset='utf8')
		# 获得Cursor对象
		self.cs1 = self.conn.cursor()
	# 返回操作符
	def __enter__(self):
		return self.cs1
	# 退出
	def __exit__(self, exc_type, exc_val, exc_tb):
		# 关闭之前做一个统一的提交
		self.conn.commit()
		# 1.3 关闭
		self.cs1.close()
		self.conn.close()
with DBOpen("stock_db") as cs:
	cs.execute("select * from info;")
	data = cs.fetchall()
for temp in data:
	print(temp)