pymysql的简单封装代码实例

针对您提出的问题，以下是“pymysql的简单封装代码实例”的完整攻略。

概述

pymysql是Python编程语言对MySQL数据库进行操作的库。使用pymysql封装一些常用的数据库操作可以让我们编写数据库相关代码时更加方便快捷。

在封装pymysql时，可以考虑将数据库的连接和关闭等基本操作进行封装，以适应不同场景和需求。本攻略将讲解如何使用Python基于pymysql进行数据库操作的简单封装。

安装pymysql

在进行封装之前，需要先安装pymysql库。使用pip命令可以进行安装。

pip install pymysql

基本封装代码

下面是一个最简单的pymysql封装代码。代码封装了一个数据库连接，和一个返回连接的函数。在需要进行数据库操作时，只需要调用返回连接的函数即可。

import pymysql

class MyDatabase:
    def __init__(self):
        self.db = None

    def connect(self):
        self.db = pymysql.connect(host='localhost',
                                  port=3306,
                                  user='root',
                                  password='password',
                                  db='mydb',
                                  charset='utf8')

    def get_conn(self):
        if not self.db:
            self.connect()
        return self.db

增加数据

下面是一个示例，向指定表中插入数据。

class MyDatabase:
    # ...

    def insert_data(self, table, data):
        conn = self.get_conn()
        try:
            with conn.cursor() as cursor:
                columns = ', '.join(data.keys())
                values = ', '.join(['%s'] * len(data))
                sql = "INSERT INTO %s (%s) VALUES (%s)" % (table, columns, values)
                cursor.execute(sql, tuple(data.values()))
                conn.commit()
        except:
            conn.rollback()
            raise

该函数主要使用了cursor对象执行配置好的sql语句。其中，使用了字典的键值对，方便构造sql语句。

查询数据

下面是一个示例，从指定表中查询数据。

class MyDatabase:
    # ...

    def select_data(self, table, columns, where=None):
        conn = self.get_conn()
        try:
            with conn.cursor() as cursor:
                columns_str = ', '.join(columns)
                sql = "SELECT %s FROM %s" % (columns_str, table)
                if where:
                    sql += " WHERE " + where
                cursor.execute(sql)
                result = cursor.fetchall()
                return result
        except:
            raise

该函数同样使用了cursor对象执行配置好的sql语句。通过where参数可以实现拼接where语句，实现更加复杂的查询需求。查询返回的结果通过fetchall方法获取。

总结

以上示例代码只是基于pymysql的简单封装方式，开发者可以根据自己的需求进行更灵活的封装方式。