python 访问数据库

最新推荐文章于 2024-06-05 22:32:42 发布

学员-枫

最新推荐文章于 2024-06-05 22:32:42 发布

阅读量113

点赞数

分类专栏： Python

本文链接：https://blog.csdn.net/u014627536/article/details/103147300

版权

Python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

# -*- coding: utf-8 -*-
import pymysql

"""
数据库操作工具类
"""


class MySQLClient():
    def __init__(self, host="host", user="root", password="password", db="db"):
        self.connect = pymysql.connect(host=host, port=3306, user=user, passwd=password, db=db, charset='utf8')
        self.cursor = self.connect.cursor(pymysql.cursors.DictCursor)

    def queryOneData(self, sql):
        try:
            self.cursor.execute(sql)
        except Exception as e:
            print(e)
        else:
            return self.cursor.fetchone()

    def queryDataList(self, sql):
        try:
            self.cursor.execute(sql)
        except Exception as e:
            print(e)
        else:
            return self.cursor.fetchall()

    def insertOneData(self, sql):
        try:
            self.cursor.execute(sql)
            self.connect.commit()
        except Exception as e:
            self.connect.rollback()
            print(e)

    def updateData(self, sql):
        try:
            self.cursor.execute(sql)
            self.connect.commit()
        except Exception as e:
            self.connect.rollback()
            print(e)

    def deleteData(self, sql):
        try:
            self.cursor.execute(sql)
            self.connect.commit()
        except Exception as e:
            self.connect.rollback()
            print(e)

    def close(self):
        self.cursor.close()
        self.connect.close()