MYSQL封装

mysql简单封装

配置文件congig,py

# 服务端口
# SERVER_PORT = 5000 			# 服务器端口

# mysql配置
MYSQL_HOST = "127.0.0.1" 		# host连接
MYSQL_PORT = 3306				# 端口
MYSQL_USER = "user"				# 用户
MYSQL_PASSWD = "123456"			# 密码
MYSQL_DB = "db_mysql"			# 数据库
MYSQL_CHARSET = "UTF8"			# 字符格式

mysql封装

# 导入配置文件
from config import *
import pymysql			# pip install pymysql

class PyMysql():
    def __init__(self):
       self.conn = pymysql.connect(
           host=MYSQL_HOST,
           port=MYSQL_PORT,
           user=MYSQL_USER,
           passwd=MYSQL_PASSWD,
           db=MYSQL_DB,
           charset=MYSQL_CHARSET,
       )
       self.cur = self.conn.cursor(cursor=pymysql.cursors.DictCursor)

    def __del__(self):
       # 关闭游标
       self.cur.close()
       # 关闭数据库连接
       self.conn.close()

    def select_db(self, sql):
        """
        查询
        :param sql: sql语句
        :return:
        """
        # 检查连接是否断开，如果断开就进行重连
        self.conn.ping(reconnect=True)
        # 使用execute() 执行sql
        self.cur.execute(sql)
        # 使用 fetchall()获取查询结果
        data = self.cur.fetchall()
        return data

    def execute_db(self,sql):
        """
        更新、新增、删除
        :param sql: sql语句
        :return:
        """
        try:
            # 检查连接是否断开，如果断开就进行重连
            self.conn.ping(reconnect=True)
            # 使用execute() 执行sql
            self.cur.execute(sql)
            # 提交事务
            self.conn.commit()
        except Exception as e:
            print('操作错误：{}'.format(e))
            # 回滚所有更改
            self.conn.rollback()
   
if __name__ == '__main__':
	db = PyMysql()