手把手教你用Django执行原生SQL

摘要.

击上方 “Python 爬虫与数据挖掘”，进行关注

回复 “书籍” 即可获赠 Python 从入门到进阶共 10 本电子书

今

日

鸡

汤

秦时明月汉时关，万里长征人未还。

# 前言

Hey，各位小伙伴，这次怎么来玩一下, 如何使用 Django 执行原生 SQL。

我们都知道，Python 在 web 界的扛把子——Django，可谓是集大成为统一，各种各样的插件、forms 组件、model 模型、Admin 后台等等，后面我会专门出文章娓娓道来，反正就是一个字，NB。

本次就来学一下, 如何在 Django 执行原生语句。

# 起因

在使用 Django 时, 一般情况下, 我们使用 Django 自带的 model 查询是没有问题的, 基本能满足 80% 的问题

但是, 但是, 那 20% 就不要了吗??? 肯定不行哎, 小孩才做选择

在 Django 执行原生 SQL 有以下三种方式

extra
raw
django connection

一般情况下, 就以上三种方式

# 表结构

文件:django_project/app01/models

class Book(models.Model):
    title = models.CharField(verbose_name="书名", max_length=32)
    describe = models.TextField(verbose_name="描述")
    author = models.CharField(verbose_name="作者", max_length=32)
    publisher = models.CharField(verbose_name="出版社", max_length=32)
    publisher_date = models.DateField(verbose_name="publisher")

就是一个很简单的图书表

通过 admin 录入一些数据测试使用

extra 方式

强烈建议, 不用学, 没毛用

# raw 方式

这个相比较 extra, 还是比较有用的,

语法如下

models.表名.objecs.raw(sql)
models.表名.objecs.raw(sql,[参数1,参数2])

注: 如果没有参数, 就只写 sql 语句, 如果由参数, 后面需要用列表, 如图所示

举例

返回的仍然一个个的 Book 对象

# 真正的原生 sql 方式

上述的，其实还是和 django 的 model 有些绑定。但是我就是说，我就是想要原生 sql，不要跟任何绑定。

这里说一下，千万不要在 django 使用 pymysql 执行原生 sql，会发生一些奇怪的问题。一定要导入from django.db import connection执行 sql。代码如下：

from django.db import connection
def book_list(request):
    
    cursor = connection.cursor()
    print(type(cursor))
    cursor.execute("select * from app01_book where id=%s", [1, ])
    raw = cursor.fetchall()
    print(raw)

返回内容如下图所示：

可以看到，返回的是列表里面套一个个的数组。我就在想，有没有什么办法能将查询出来的 sql，直接返回成字典呢？答案是当然可以！

# 执行原生 sql 并且返回成 dict

我将执行原生 sql 并且直接返回成字典的方式封装成了两个函数

一个是查询多个，代码如下所示：

def query_all_dict(sql, params=None):
    '''
    查询所有结果返回字典类型数据
    :param sql:
    :param params:
    :return:
    '''
    with connection.cursor() as cursor:
        if params:
            cursor.execute(sql, params=params)
        else:
            cursor.execute(sql)
        col_names = [desc[0] for desc in cursor.description]
        row = cursor.fetchall()
        rowList = []
        for list in row:
            tMap = dict(zip(col_names, list))
            rowList.append(tMap)
        return rowList

一个是查询一个，代码如下所示：

def query_one_dict(sql, params=None):
    """
    查询一个结果返回字典类型数据
    :param sql:
    :param params:
    :return:
    """
    with connection.cursor() as cursor:
        if params:
            cursor.execute(sql, params=params)
        else:
            cursor.execute(sql)
        col_names = [desc[0] for desc in cursor.description]
        row = cursor.fetchone()
        tMap = dict(zip(col_names, row))
        return tMap