Python 操作MySql数据库（封装、优雅）

学了pymysql第三方库（pip install pymysql）来操作MySql数据库后，浅记一下对MySql进行 《关于我的MySql之优雅封装这件事儿》。这里只涉及了单表的增、删、查、改这些简单地封装操作，便于对简单的sql语句进行快速使用。

首页，利用Navicat连接本地的MySql数据库来快速新建一个（pythondemo）数据库，在里面快速创建两张空表（tb_test，tb_user）用于测试,其中主要使用tb_user表来进行数据测试，如下图进行tb_user的简单设计（将其中的id字段设为自动递增，便于在insert插入数据时少写一个字段。）：

---

---

能实现简单的功能有：

• 以元组类型快速insert插入数据
• 通过id删除数据
• 通过传入id快速查询返回一条数据，返回结果
• 直接调用方法查询表中所有数据，返回结果
• 指定查询返回多少条数数据,可根据简单条件查询（where 字段=”“），返回结果
• 通过id修改数据
• 可通过自定义sql语句执行

将以下代码可复制到一个py模块中即可运行（注意：要修改自己连接的数据库配置）！！！

---

---

# -*- coding:utf-8 -*-"""@File : my_mysql.py@Time : 2022/9/3 17:32"""import pymysqlclass MySql:    def __init__(self, operate_tablename:str,my_sqldb_config_param:dict):        assert isinstance(my_sqldb_config_param,dict),"请以字典类型的格式传入！"        self._operate_tablename = operate_tablename        try:            self._conn = pymysql.connect(**my_sqldb_config_param) # 连接数据库，配置参数            self._cursor = self._conn.cursor() # 创建一个游标，用来执行查询            self._get_field() # 获取此表中的字段名        except Exception as e:            raise Exception(f"数据库连接失败！！！\n请检查表名、配置参数是否正确或检查本地数据库是否已启动！\n{e}")    # 获取_conn对象    @property    def get_connect(self):        return self._conn    # 获取_cursor对象    @property    def get_cursor(self):        return self._cursor    # 获取__desc对象    @property    def get_description(self):        # print(f"{self._operate_tablename}表中的字段属性：",self._desc)        return self._desc    # 获取正在操作的表名    @property    def operate_tablename(self):        return f"正在操作 {self._operate_tablename}表！！！"    # 修改要操作的表    @operate_tablename.setter    def operate_tablename(self,operate_tablename):        assert operate_tablename !="", "请输入要操作的表名！"        print(f"{self._operate_tablename} 表已被更换！")        self._operate_tablename = operate_tablename        self._get_field()    # 获取此表中的字段名    def _get_field(self):        self._cursor.execute(f"select * from {self._operate_tablename}")        self._desc = self._cursor.description        self._field_ = []        for field in self._desc:            self._field_.append(field[0])    # 执行sql语句    def _sql(self,sql,msg=""):        try:            self._cursor.execute(sql)  # 执行sql语句            self._conn.commit() # 执行sql语句后，进行提交            if msg:print(f"数据{msg}成功！")            return True        except Exception as e:            if msg:print(f"\033[31m数据{msg}失败！！！\n{e} \033[0m")            self._conn.rollback()  # 执行sql语句失败，进行回滚            return False    # 插入数据    def insert(self, *value):        if not isinstance(value[0],tuple): raise Exception("要求传入的参数类型为tuple元组！！！")        if len(value) == 1: value=value[0]        else:value = str(value)[1:-1]        sql = f"insert into {self._operate_tablename}({','.join(self._field_[1:])}) values {value}"        if not self._sql(sql,f"{value}插入"):            print("\n\033[31m：请检查每一条记录字段是否正确！！！\033[0m\n")    # 插入：自定义sql语句插入数据    def insert_by_sql(self, sql):        self._sql(sql,"插入")    # 删除：通过id删除该条数据    def delete_by_id(self,id_:int):        sql = f"delete from {self._operate_tablename} where id = {id_}"        if self._sql(sql):print(f"id={id_}记录，删除成功！")        else:print(f"\n\033[31m：id = {id_}记录，删除失败！！！\033[0m\n")    # 删除：自定义sql语句删除数据    def delete_by_sql(self, sql):        self._sql(sql,"删除")    # 修改：通过id修改数据    def update_by_id(self, id_:int, set_field:dict):        assert isinstance(set_field,dict),"请以字典类型的格式传入！"        tempset_field = []        for i in set_field:            tempset_field.append(f"{i}='{set_field[i]}'")        set_field = ",".join(tempset_field)        sql = f"update {self._operate_tablename} set {set_field} where id = {id_}"        if self._sql(sql):print(f"id={id_}记录，{set_field}修改成功！")        else:print(f"\n\033[31m：id = {id_}记录，{set_field}修改失败！！！\033[0m\n")    # 修改：自定义sql语句修改数据    def update_by_sql(self, sql):        self._sql(sql,"修改")    # 查询：通过id查询一条数据    def select_by_id(self,id_:int,field="*"):        if field != "*": field = ','.join(field)        sql = f"select {field} from {self._operate_tablename} where id={id_}"        self._cursor.execute(sql)        return self._cursor.fetchone()    # 查询：指定查询多少条数数据,可根据简单条件查询（where 字段=”“）    def select_many(self,num:int,query_builder=None,field="*"):        if field != "*": field = ','.join(field)        sql = f"select {field} from {self._operate_tablename}"        if query_builder:            if isinstance(query_builder,dict) and len(query_builder) == 1:                query_builder = list(query_builder.items())[0]                sql = f"select {field} from {self._operate_tablename} where {query_builder[0]}='{query_builder[1]}'"            else: raise Exception("要求输入的条件为dict(字典)类型并且只能有一对键值（：len(dict）=1）！！！")        self._cursor.execute(sql)        return self._cursor.fetchmany(num)    # 查询：所有数据    def select_all(self, field="*"):        if field != "*": field = ','.join(field)        sql = f"select {field} from {self._operate_tablename}"        self._cursor.execute(sql)        return self._cursor.fetchall()    # 查询：自定义sql语句查询数据    def select_by_sql(self, sql):        try:            self._cursor.execute(sql)            return self._cursor.fetchall()        except Exception as e:            print(f"\033[31m：数据查询失败！！！\n{e} \033[0m")    # 当对象被销毁时，游标先关闭,连接后关闭    def __del__(self):        self._cursor.close()        self._conn.close()        

---

---

注意下面传入数据的格式！！！

def my_mysql_test(operate_tablename:str, my_sqldb_config_param:dict):    print("-------------my_mysql_test：注意下面传入数据的格式---------------")    # 创建自己的mysql连接对象，operate_tablename是要进行操作的表名，my_sqldb_config_param是pymysql连接本机MySQL所需的配置参数    mysql = MySql(operate_tablename=operate_tablename, my_sqldb_config_param=my_sqldb_config_param)    print(mysql.operate_tablename)    mysql.operate_tablename = "tb_user"   # 修改要操作的表    print(mysql.operate_tablename)    print("----------------insert-----------------")    # 自定义sql语句，插入多条数据，这里我们在创建表时，设置了id自动增加，所以这里不需要设置id字段    mysql.insert_by_sql('insert into tb_user(name,age,gender) values ("111", 12, "男"), ("222", 22, "女"),("333", 32, "女")')    mysql.insert(("444", 42, "男"))  # 插入一条数据    mysql.insert(("555",52,"男"),("666",62,"女"))  # 插入多条数据    # mysql.insert("777",72,"男") # 异常    # mysql.insert(["888",82,"女"]) # 异常    mysql.insert(("999", 92, "男"), ("000", '28', "女", "xxx"))  # 异常    print("----------------select-----------------")    result = mysql.select_by_sql("select * from tb_user where gender='男'")  # 自定义sql语句查询数据    print("查询：自定义sql查询数据：\n", result)    result = mysql.select_all() # 查询表中所有数据，返回表中所有数据    print("查询：表中所有数据：\n", result)    result = mysql.select_by_id(1) # 通过id查询，返回一条数据    print("\n查询：通过id：", result)    result = mysql.select_many(1, {"gender": "女"}) # 指定查询返回多少条数数据,可根据简单条件查询（where 字段=”“）    print('查询：指定查询多少条数数据,可根据简单条件查询（where 字段=”“）：', result)    print("----------------delete-----------------")    mysql.delete_by_sql('delete from tb_user where gender="男"')  # 自定义sql语句删除数据    mysql.delete_by_id(4) # 通过id删除数据    result = mysql.select_all()    print("删除数据后查询表中所有数据：\n", result)    print("----------------update-----------------")    mysql.update_by_sql("update tb_user set name='update_name',gender='男' where id = 6")  # 自定义sql语句更新数据    mysql.update_by_id(3, {"age": "180"})   # 通过id更新数据    mysql.update_by_id(2, {"name": "update_name", "age": "999"})    mysql.update_by_id(6, {"xxx": "updateName", "yyy": "18"}) # 异常    # mysql.update_by_id(1, ("age","180"))  # 异常    result = mysql.select_all()    print("更新数据后查询表中所有数据：\n", result)if __name__ == '__main__':    my_sqldb_config_param = {        "host": "127.0.0.1", #连接主机的ip        "port": 3306,   #连接主机的端口        "user": "root", #本地数据库的用户名        "password": "***", #本地数据库的密码        "database": "pythondemo", #连接的数据库        "charset": "utf8" #设置编码格式    }    operate_tablename = "tb_test" #设置该数据库准备操作的表名    my_mysql_test(operate_tablename,my_sqldb_config_param)    

---

结果展示

---

控制台打印的结果：

---

数据库中的结果：

---

来源地址：https://blog.csdn.net/qq_59142194/article/details/126681062