使用PyMySQL封装一个调用MySQL数据库的Python类

前言

---

我们在学习一门语言时必定要知道怎么连接数据库，这里给大家封装了一个Python调用MySQL的类，大家下载下来就能在自己的项目中直接使用了!

 大家要提前安装PyMySQL这个Python包，如果没有安装请打开cmd输入下面命令（也可以在pycharm中安装）

pip install PyMySQL

import pymysql


class MySQL:
    def __init__(self, host, user, password, database):
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.db = None
        self.cursor = None

    # 连接数据库
    def connect(self):
        self.db = pymysql.connect(self.host, self.user, self.password, self.database)
        self.cursor = self.db.cursor()

    # 关闭连接
    def close(self):
        self.cursor.close()
        self.db.close()

    # 查询一条数据
    def get_one(self, sql):
        res = None
        try:
            self.connect()
            self.cursor.execute(sql)
            res = self.cursor.fetchone()
            self.close()
        except self.db.Error:
            print("查询失败！")
        return res
    
    # 查询多条数据
    def get_all(self, sql):
        res = ()
        try:
            self.connect()
            self.cursor.execute(sql)
            res = self.cursor.fetchall()
            self.close()
        except self.db.Error:
            print("查询失败！")
        return res
    
    # 插入数据
    def insert(self, sql):
        return self.__edit(sql)
    
    # 修改数据
    def update(self, sql):
        return self.__edit(sql)

    # 删除数据
    def delete(self, sql):
        return self.__edit(sql)

    def __edit(self, sql):
        count = 0
        try:
            self.connect()
            count = self.cursor.execute(sql)
            self.db.commit()
            self.close()
        except self.db.Error:
            print("提交事物失败！")
            self.db.rollback()
        return count