数据库 | pymysql封装

"""
数据库操作
"""

from django.db import connection as conn


def update(sql, params=None):
    """
    增 删 改
    :param sql:
    :param params:
    :return:
    """

    with conn.cursor() as cursor:
        cursor.execute(sql, params=None)

        # 增加时返回lastrowid，删改时返回执行的行数
        return cursor.lastrowid if cursor.lastrowid else cursor.rowcount

def queryone(sql, params=None):
    """
    查询单条记录
    :param sql:
    :param params:
    :return:
    """
    with conn.cursor() as cursor:
        cursor.execute(sql, params)

        # 获取结果集
        data = cursor.fetchone()
        if data is None:
            return

        column = [columns[0] for columns in cursor.description]

        return dict(zip(column, data))


def querymany(sql, params=None):
    """
    查询多条记录
    :param sql:
    :param params:
    :return:
    """
    with conn.cursor() as cursor:
        cursor.execute(sql, params)

        # 获取结果集
        data = cursor.fetchone()

        if data is None:
            return []

        column = [columns[0] for columns in cursor.description]
        return [dict(zip(column, d)) for d in data]