配置config.py

最新推荐文章于 2024-10-07 01:36:35 发布

yamaa9

最新推荐文章于 2024-10-07 01:36:35 发布

阅读量113

点赞数 3

文章标签： python

本文链接：https://blog.csdn.net/weixin_38656011/article/details/138212861

版权

文章介绍了如何在Python中通过Django配置和使用pymysql库来连接数据库，创建了一个名为SQLManager的类，包含了连接、查询、执行SQL语句以及关闭连接的方法。

摘要由CSDN通过智能技术生成

# 数据库操作类
# from django.db import close_old_connections
import pymysql

DB_CONFIG = {
	"host": "127.0.0.1",
	"port": 3306,
	"user": "root",
	"passwd": "root",
	"db": "huwenlong",
	"charset": "utf8"
}


class SQLManager(object):

	# 初始化实例方法
	def __init__(self):
		self.conn = None
		self.cursor = None
		self.connect()

	# 连接数据库
	def connect(self):
		self.conn = pymysql.connect(
			host=DB_CONFIG["host"],
			port=DB_CONFIG["port"],
			user=DB_CONFIG["user"],
			passwd=DB_CONFIG["passwd"],
			db=DB_CONFIG["db"],
			charset=DB_CONFIG["charset"],
		)
		sql = 'select *	from reportdata'
		self.conn.ping(reconnect=True)
		cur = self.conn.cursor()
		cur.execute(sql)
		self.conn.commit()
		self.cursor = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

	# 查询多条数据
	def get_list(self, sql, args=None):
		self.cursor.execute(sql, args)
		# close_old_connections()
		return self.cursor.fetchall()

	# 查询单条数据
	def get_one(self, sql, args=None):
		self.cursor.execute(sql, args)
		# close_old_connections()
		return self.cursor.fetchone()

	# 执行单条SQL语句
	def modify(self, sql, args=None):
		row = self.cursor.execute(sql, args)
		self.conn.commit()
		# close_old_connections()
		return row > 0

	# 执行多条SQL语句
	def multi_modify(self, sql, args=None):
		rows = self.cursor.executemany(sql, args)
		self.conn.commit()
		# close_old_connections()
		return rows > 0

	# 关闭数据库cursor和连接
	def close(self):
		self.cursor.close()
		self.conn.close()