Python封装pymysql数据库操作类详解

林舟
林舟
发布时间:2026-10-16| 5人浏览过
原创

本文介绍如何封装一个Python的pymysql操作类,简化数据库连接与查询流程。该类支持配置连接参数,提供fetch、insert等常用方法,并自动管理游标与事务提交,适用于需要频繁进行MySQL交互的Python脚本开发场景。

pymysql操作类封装背景

近期在编写Python脚本时,使用pymysql进行数据库交互的频率显著增加。为了方便后续开发与维护,这里对常用的数据库操作进行简单封装,构建一个通用的pymysql操作类。

核心代码实现

import pymysql


class MysqlDB:

    def __init__(
        self, 
        host=None, 
        port=None, 
        db=None, 
        account=None,
        password=None, 
        connect_timeout=20,
        read_timeout=20,
        write_timeout=20
        ):
        self.conn = pymysql.connect(
            host=self.host,
            port=self.port,
            db=self.db,
            user=self.account,
            passwd=self.password,
            connect_timeout=self.connect_timeout,
            read_timeout=self.read_timeout,
            write_timeout=self.write_timeout
        )
    
    def fetch(self, table_name=None, fields=(), where=None, many=False):
        cur = self.conn.cursor()

        if where:
            sql = f'select {",".join(fields)} from {table_name} where {where}'
        else:
            sql = f'select {",".join(fields)} from {table_name}'
        cur.execute(sql)
        if many:
            data = cur.fetchmany()
        else:
            data = cur.fetchone()
        cur.close()
        return data


    def update(self, table_name=None, field=None, value=None, where=None):
        cur = self.conn.cursor()
        sql = f'update {table_name} set {field} = {value}'
        if where:
            sql += f'where {where}'
        cur.execute(sql)
        self.conn.commit()
        cur.close()
    
    def insert(self, table_name=None, single=True, data_list: list = []):
        cur = self.conn.cursor()
        for data in data_list:
            sql = f'insert into {table_name}({",".join([key for key in data.keys()])}) values({",".join(["%s" for _ in range(len(data.keys()))])})'
            cur.execute(sql, data)
        self.conn.commit()
        cur.close()

    def quit(self):
        self.conn.close()

以上就是对Python封装pymysql操作类的详细介绍。如需了解更多相关内容,可查阅本站往期文章或继续浏览后续教程。感谢大家的支持!

相关标签:

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。