Python编程：封装MySQL通用辅助类

已于 2022-12-15 12:37:20 修改
阅读量870 收藏 4
点赞数 1
分类专栏： Python 文章标签： mysql 数据库 python
于 2022-08-24 14:35:15 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/wsfsp_4/article/details/126503627
版权

一.前言

通过Python使用pymysql库来操作MySQL数据库时，发现该库只提供一个execute()函数来操作，SQL完全需要自己来写，操作多个数据库或多个表时，写SQL语句特别麻烦，于是想建立一个通用辅助类，能对不同数据库和不同表提供便捷和通用的接口，比如常用操作通过以下代码方式就能调用：

if __name__ == "__main__":
    db = SQLHelper("source")

    # 插入-字典形式
    proxy = {"ip": "127.0.0.1", "port": 10002}
    db.setTable("proxies").setObj(proxy).insert()

    # 插入-普通形式
    fileds = ["ip", "port"]
    values = ["127.0.0.1", 10002]
    db.setTable("proxies").setFields(fileds).setValues(values).insert()

    # 更新-字典形式
    proxy = {"ip": "127.0.0.1", "port": 8080}
    wheres = [['ip', "=", "127.0.0.1"]]
    db.setTable("proxies").setObj(proxy).setWheres(wheres).update()

    # 更新-普通形式
    fileds = ["ip", "port"]
    values = ["127.0.0.1", 8080]
    wheres = [['ip', "=", "127.0.0.1"]]
    db.setTable("proxies").setFields(fileds).setValues(values).setWheres(wheres).update()

    # 查询
    orders = ["ip", ("port", "DESC")]
    wheres = [["port", ">", 1000]]
    proxies = db.setTable("proxies").query()
    proxies = db.setTable("proxies").setLimit(10).query()
    proxies = db.setTable("proxies").setOrders(orders).setWheres(wheres).query()

    # 复杂SQL语句，可以直接调用execute()
    cursor = db.setTable("proxies").execute("your sql")

二.正文

辅助类代码如下

from copy import deepcopy
import pymysql


class SQLHelper:
    def __init__(self, database, host="localhost", user="root", password="") -> None:
        self.host = host
        self.user = user
        self.password = password
        self.database = database
        self.db = pymysql.connect(
            host=self.host, user=self.user, passwd=self.password, database=self.database)
        self.init()

    def close(self):
        try:
            self.db.close()
        except:
            pass

    def init(self):
        """
        重置条件变量;
        """
        self.table = None
        self.wheres = None
        self.fields = None
        self.orders = None
        self.limit = None
        self.offset = None
        self.values = None

    def setDB(self, db: str):
        """
        (可选)设置数据库,不设置则使用上次操作的数据库;
        """
        self.db.select_db(db)
        return self

    def setTable(self, table: str):
        """
        (必选)设置要操作的表;
        """
        self.table = table
        return self

    def setFields(self, fields: list):
        """
        (可选)设置insert、query或update字段,对于insert不设置则默认选择全部字段;
        """
        self.fields = list(fields)
        return self

    def setValues(self, values: list):
        """
        (必选)设置insert或update字段的值;
        """
        self.values = list(values)
        #字符串类型值用''包括
        for i in range(len(self.values)):
            if(isinstance(self.values[i], str)):
                self.values[i] = f"'{self.values[i]}'"        
        return self

    def setObj(self, obj: dict):
        """
        (可选)设置insert或update的对象;
        """
        self.setFields(obj.keys())
        self.setValues(obj.values())
        return self

    def setWheres(self, wheres: list):
        """
        (可选)设置update或query的条件;
        """
        #多层对象必须需要深拷贝
        self.wheres = deepcopy(wheres)
        for i in range(len(self.wheres)):
            #kv默认用=号连接
            if(len(self.wheres[i]) == 2):
                self.wheres[i].insert(1,"=")
            #字符串类型值用''包括
            if(len(self.wheres[i]) == 3 and isinstance(self.wheres[i][2],str)):
                self.wheres[i][2] = f"'{self.wheres[i][2]}'"
        return self

    def setLimit(self, limit: int):
        """
        (可选)设置query的数量;
        """
        self.limit = limit
        return self

    def setOffset(self, offset: int):
        """
        (可选)设置query的偏移量;
        """
        self.offset = offset
        return self

    def setOrders(self, orders: list):
        """
        (可选)设置query排序字段和排序方式(默认升序);
        """
        self.orders = deepcopy(orders)
        return self

    def builderSet(self) -> str:
        """
        构建SET子句
        """
        if(self.fields == None or self.values == None or len(self.fields) != len(self.values)):
            return None
        kvs = []
        for i in range(len(self.fields)):
            field = self.fields[i]
            value = self.values[i]
            kvs.append(f"{field} = {value}")
        return f"SET {','.join(kvs)} "

    def buildWhere(self) -> str:
        """
        构建WHERE子句;
        """
        if(self.wheres == None):
            return None
        isConnect = False
        ws = []
        for where in self.wheres:
            if(len(where) == 1):
                ws.append(where)
            else:
                # 默认AND连接条件
                if(isConnect):
                    ws.append("AND")
                ws.append(f"{where[0]} {where[1]} {where[2]}")
            isConnect = not isConnect
        return f"WHERE {' '.join(ws)} "

    def buildOrders(self) -> str:
        """
        构建ORDER BY子句
        """
        if(self.orders == None):
            return None
        ods = []
        for order in self.orders:
            # 默认升序
            if(isinstance(order, str)):
                ods.append(f"{order} ASC")
            # 自定义排序
            else:
                ods.append(f"{order[0]} {order[1]}")
        return f"ORDER BY {','.join(ods)} "

    def buildUpdate(self) -> str:
        """
        构建UPDATE语句;
        """
        if(self.table == None):
            return None
        sql = f"UPDATE {self.table} "
        setW = self.builderSet()
        if(setW == None):
            return None
        sql += setW
        where = self.buildWhere()
        if(where == None):
            return None
        sql += where
        return sql

    def buildInsert(self) -> str:
        """
        构建SELECT语句;
        """
        if(self.table == None):
            return None
        sql = f"INSERT INTO {self.table}({','.join(self.fields)}) VALUES({','.join(self.values)})"
        return sql

    def buildDelete(self) -> str:
        """
        构建DELETE语句,不允许没有where条件
        """
        if(self.table == None):
            return None
        where = self.buildWhere()
        if(not where):
            return None
        sql = f"DELETE FROM {self.table} {where}"
        return sql    

    def buildQuery(self) -> str:
        """
        构建SELECT语句;
        """
        # 初步判断
        if(self.table == None):
            return None
        # 查询语句
        sql = "SELECT "
        # 选择字段
        if(self.fields != None):
            sql += f"{','.join(self.fields)} "
        else:
            sql += "* "
        # 选择表
        sql += f"FROM {self.table} "
        # 选择条件
        where = self.buildWhere()
        if(where):
            sql += where
        # 选择排序
        orders = self.buildOrders()
        if(orders):
            sql += orders
        # 选择数量限制
        if(self.limit != None):
            sql += f"LIMIT {self.limit} "
        # 选择偏移
        if(self.offset != None):
            sql += f"OFFSET {self.offset} "
        return sql

    def update(self) -> bool:
        """
        执行update操作,不允许没有where条件
        """
        sql = self.buildUpdate()
        self.init()
        cursor = self.execute(sql)
        if(cursor != None):
            return True
        else:
            return False

    def insert(self) -> bool:
        """
        执行insert操作;
        """
        sql = self.buildInsert()
        self.init()
        cursor = self.execute(sql)
        if(cursor != None):
            return True
        else:
            return False

    def delete(self) -> bool:
        """
        执行delete操作;
        """
        sql = self.buildDelete()
        self.init()
        cursor = self.execute(sql)
        if(cursor != None):
            return True
        else:
            return False

    def query(self) -> tuple:
        """
        执行query操作;
        """
        sql = self.buildQuery()
        self.init()
        cursor = self.execute(sql)
        if(cursor != None):
            return cursor.fetchall()
        else:
            return None

    def execute(self, sql: str):
        """
        执行SQL语句;
        """
        if(not sql):
            return None
        try:
            cursor = self.db.cursor()
            cursor.execute(sql)
            self.db.commit()
            return cursor
        except Exception as e:
            print(str(e))
            return None

三.结语

该辅助类功能还不是很完善，可以基于自己的需求来拓展接口，比如自增、自减等。

本教程基于本身需求经验编写，只供参考学习，不足之处还请指正，欢迎伙伴们来一起探讨交流！

开发大观园
关注
专栏目录
Python MySQL 操作类（有使用例子）
03-08
Python 操作数据库的 类库，使用MySQLdb ,调用简单，有批量新增，查询，更新 # db=MySQLClass('127.0.0.1','root','123456') # db.setdb('test')
python mysql 帮助类
长门有希的博客
09-16 380
https://github.com/ItisDL/python-mysql-helper 欢迎点赞~ Thanks♪(･ω･)ﾉ
python封装一个 MySQL 连接类，继承自 MySQLConnector
最新发布
cyj0919的博客
08-13 320
在Python中，你可以创建一个MySQL连接类，通常会继承自`mysql-connector-python`库中的`mysql.connector.connection.MySQLConnection`基础类。封装这个类可以帮助管理数据库连接、执行查询以及关闭连接等操作。在这个例子中，`MysqlDBConnection`类初始化时连接到数据库，并提供了一些额外的方法如`execute_query`用于执行SQL查询，以及`close`方法负责关闭连接。
python mysql 通用类_python mysql类
weixin_31000553的博客
02-09 145
python操作mysql的简单封装类1.[代码][Python]代码#!/usr/bin/env python# -*- coding: utf-8 -*-import MySQLdbclass db():CURSOR_DICT = 0;CURSOR_DEFAULT = 1;def __init__(self, *args, **kwargs):self.__connect = None;se...
python mysql 辅助类_Python实现一个简单的MySQL类
weixin_42524750的博客
02-19 134
本文实例讲述了Python实现一个简单的MySQL类。分享给大家供大家参考。具体实现方法如下：#!/usr/bin/env python# -*- coding:utf-8 -*-# Created on 2011-2-19# @author: xiaoxiaoimport MySQLdbimport sys__all__ = [‘MySQL‘]class MySQL(object):‘‘‘MyS...
用python写一个连mysql的帮助类
weixin_42577243的博客
02-15 198
好的，下面是一个使用Python连接MySQL数据库的帮助类： import mysql.connector class MySQLHelper: def __init__(self, host, user, password, database): self.host = host self.user = user self.passw...
Python-爬取淘宝MM用户相册及图片数据存储到MySQL数据库中
08-12
- `mm-master`可能是项目文件夹，里面可能包含了`requirements.txt`（列出所需的Python库）、`main.py`（主程序）、`models.py`（定义数据库模型）、`parsers.py`（解析HTML）、`utils.py`（通用辅助函数）等文件。...
【愚公系列】《AIGC辅助软件开发》011-AI辅助编写技术文档：技术文档
热门推荐
时光隧道
07-24 1万+
在当今快速发展的技术环境中，技术文档的编写变得愈加重要。无论是软件开发、产品设计，还是系统架构，清晰、准确的技术文档不仅能帮助团队成员快速理解项目，还能为用户提供必要的使用指导。然而，编写高质量的技术文档常常是一项繁琐且耗时的任务。随着人工智能技术的不断进步，AI工具的辅助作用逐渐显现，特别是在技术文档的编写过程中。借助AI，开发者和技术作家可以更高效地生成、编辑和维护文档，从而专注于更高层次的创作和思考。
黑马程序员Python学习笔记，代码使用Pycharm编写。_黑马编程python笔记(1)
2401_84141332的博客
05-02 773
单引号定义法：ｎａｍｅ＝＇ｈｅｌｌｏ！双引号定义法：ｎａｍｅ＝＂ｈｅｌｌｏ！三引号定义法：ｎａｍｅ＝＂＇ｈｅｌｌｏ！＇＂三引号定义法，和多行注释的写法一样，同样支持换行操作。使用变量接收它，它就是字符串不使用变量接收它，就可以作为多行注释使用。问题：如果想要定义的字符串本身　，是包含：单引号／双引号怎么写？单引号定义法，可以内含双引号双引号定义法，可以内含单引号可以使用转义字符（＼）来讲引号解除效用，变成普通字符串。
开启全民编程时代！Python小学生都要学的编程语言！
IM_Hank的博客
02-14 1959
如要问现在最流行与时髦的计算机编程语言是什么？毫无疑问，Python领衔主演，网络上一句“人生苦短，我用Python” 不知换走了多少程序员的真心！ Python近些年随着大数据，云计算、AI等领域的大规模应用，稳步的在编程语言排行榜上逐步上升，业内也把Python提到了一个相当高的地位。可是Python并不是大家想象中的一个新兴的计算机编程语言，它其实在1989年就被荷兰人Guido van R...
数据库整理一：Mysql基础知识
weixin_45486448的博客
09-08 1104
文章目录一、当使用两列做主键时：primary key(列名，列名)即可，注意：这两列都要not null外键的名字不可以重复，外键可以节省空间，可以起到约束的作用外键：一对一，一对多，多对多当外键关联的表中使用两列做主键时，外键sql语句中references后面可以使用这两个列名 一、 创建与删除操作 create database 数据库名称 charset=utf8 drop database 数据库名称 数据表： create table 表名(
Python3实现的Mysql数据库操作封装类
09-09
主要介绍了Python3实现的Mysql数据库操作封装类,涉及Python针对mysql数据库的连接、查询、更新及关闭连接等相关操作技巧,需要的朋友可以参考下
python操作mysql封装库
03-03
python连接mysql，操作mysql函数封装
python mysql操作类 MySQL.py
04-10
最近用到Python操作数据库，MySql的操作封装类，简单写了一个Mysql封装操作类，分享一下，需要安装pymsql，安装方法：https://blog.csdn.net/sinat_28984567/article/details/105316092
Python操作Mysql的简单封装
什么都干的派森
05-16 2435
文章目录一、安装依赖二、实现类 一、安装依赖 pip install mysqlclient 二、实现类 # encoding: utf-8 # sql导出 # mysqldump -u root -p db_name > test_db.sql import MySQLdb class MySqlOp(object): def __init__(self, host, port, username, password, db): self.host = host
【专题】python封装MySQL类
qq_36581961的博客
07-07 651
Python创建连接MySQL类
python mysql 通用类_Python访问MySQL封装的常用类
weixin_34851528的博客
02-01 118
python访问mysql比较简单，细节请参考我的另一篇文章：链接自己平时也就用到两个mysql函数：查询和更新，下面是自己常用的函数的封装，大家拷贝过去直接可以使用。文件名：DBUtil.py# -*- encoding:utf8 -*-'''@author: crazyant.net@version: 2013-10-22封装的mysql常用函数'''import MySQLdbclass D...
Python操作mySql数据库封装类
星如雨落
09-06 823
这是自己在做项目过程中，整理封装的操作mysql数据库封装类，自己可以修改下使用，节省大量时间。 myGlobal.py # -*- coding: utf-8 -*- global globalListIPValid # 保存一个VLAN全局可用IP列表 myPySql.py # -*- coding: utf-8 -*- import re import pymysql import myGlobal # myGlobal全局变量文件 class MySql: """ pyt
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

开发大观园

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值