Django 提供了兩種方式來(lái)執(zhí)行原生 SQL 代碼。

一種是使用 raw() 函數(shù)，一種是 使用 connection.cursor()。

但是官方還是推薦在使用原生 SQL 之前，盡量的先去探索一下 QuerySet 提供的各種 API。

目前而言，官方文檔提供的各種 API 是能夠滿足絕大多數(shù)的應(yīng)用場(chǎng)景的。

以下是本篇筆記的目錄：

• raw()
• connection.cursor()
• 多數(shù)據(jù)庫(kù)操作

1、raw()

這個(gè)方法可以用來(lái)操作原生 SQL，然后返回 model 實(shí)例：

我們以 Blog 作為示例，使用的代碼如下：

for blog in Blog.objects.raw("select * from blog_blog"):
    print(blog)

上面這段代碼的作用效果跟 Blog.objects.all() 獲取的結(jié)果是一樣的，但是在一些操作上又不太一樣，比如 all().count() 可以獲取到 總數(shù)，但是 raw() 就不能進(jìn)行這種操作。

需要注意的是，raw() 并不會(huì)去檢測(cè)輸入的 SQL 代碼，即使我們使用 Blog 這個(gè)model 來(lái)查詢 Entry 的數(shù)據(jù)，也能夠返回結(jié)果，不過(guò)返回的都是 Entry 表的屬性：

for blog in Blog.objects.raw("select * from blog_entry"):
    print(blog.__dict__)  # __dict__ 輸出的都是 blog_entry 的字段

也就是說(shuō) 在 Blog.objects.raw() 中，真正起作用的只有 raw() 這一個(gè)函數(shù)，前面的 Blog 不過(guò)是一個(gè)架子，或者途徑，是為了引出 raw() 這個(gè)函數(shù)。

自定義返回字段

在 raw() 函數(shù)的 SQL 代碼里，我們可以自定義選中的字段，如果需要使用沒(méi)有選中的字段，那么系統(tǒng)會(huì)再次訪問(wèn)數(shù)據(jù)庫(kù)來(lái)獲取，這個(gè)操作過(guò)程就跟前面介紹的 defer() 函數(shù)一樣。

for item in Blog.objects.raw("select id from blog_blog"):
    print(item.__dict__)
    print(item.id, item.name)
    print(item.__dict__)

{'_state': <django.db.models.base.ModelState object at 0x7fbd4165b6a0>, 'id': 2}
2 hunter
{'_state': <django.db.models.base.ModelState object at 0x7fbd4165b6a0>, 'id': 2, 'name': 'hunter'}

可以看到，返回的結(jié)果中先輸出的數(shù)據(jù)只有 id，后面，當(dāng)我們?cè)L問(wèn) name 字段的時(shí)候，又去獲取了 name 字段的數(shù)據(jù)。

自定義字段必須包含主鍵

當(dāng)我們自定義字段返回的時(shí)候，必須是要包含主鍵字段的，否則在我們獲取信息的時(shí)候會(huì)報(bào)錯(cuò)，比如下面的操作：

for blog in Blog.objects.raw("select name from blog_blog"):
    print(blog.__dict__)

在 print(blog.dict) 的時(shí)候就會(huì)報(bào)錯(cuò)，數(shù)據(jù)中沒(méi)有主鍵信息

自定義返回新字段

可以跟 QuerySet 的 annotate 操作一樣，自定義新字段返回，獲取的時(shí)候可以直接根據(jù)屬性值返回，比如：

entry = Entry.objects.raw("select *, date_format(pub_date, '%%Y-%%m') as date_1 from blog_entry")[0]
print(entry.date_1)

傳遞變量

給輸入的 SQL 語(yǔ)句傳遞變量：

name = "python"
Blog.objects.raw("select * from blog_blog where name = '%s'", [name])

2、connection.cursor()

Django 推出了一種更加直接執(zhí)行 SQL 的方式，用到的模塊是 django.db.connction，用到的 cursor 和 pymysql 的庫(kù)是一樣的用法，官方給出的示例如下：

from django.db import connection


def my_custom_sql(self):
    with connection.cursor() as cursor:
        cursor.execute("UPDATE bar SET foo = 1 WHERE baz = %s", [self.baz])
        cursor.execute("SELECT foo FROM bar WHERE baz = %s", [self.baz])
        row = cursor.fetchone()

    return row

需要注意，如果有參數(shù)傳入 SQL 中，需要對(duì)一些符號(hào)進(jìn)行轉(zhuǎn)義之后才可以使用，比如：

cursor.execute("SELECT foo FROM bar WHERE baz = '30%'")
cursor.execute("SELECT foo FROM bar WHERE baz = '30%%' AND id = %s", [self.id])

其中，第二條語(yǔ)句的 % 需要進(jìn)行轉(zhuǎn)義

適配fetch的數(shù)據(jù)

通過(guò) fetchone 和 fetchall() 返回的數(shù)據(jù)是只有 value 值的，沒(méi)有對(duì)應(yīng)的字段 key，如果可以適當(dāng)?shù)臓奚阅芎蛢?nèi)存，來(lái)?yè)Q取獲取數(shù)據(jù)的便利和準(zhǔn)確性，官方提供了這樣一種方式：

def dictfetchall(cursor):
    "Return all rows from a cursor as a dict"
    columns = [col[0] for col in cursor.description]
    return [
        dict(zip(columns, row))
        for row in cursor.fetchall()
    ]

在我們執(zhí)行完 cursor.execute(sql) 之后，把 cursor 作為參數(shù)傳入 dictfetchall 函數(shù)中，就可以返回一個(gè)字典列表

介紹一下 cursor.description，這個(gè)返回的是一個(gè)元組數(shù)據(jù)，里面的元素也是一個(gè)元組，元組的第一個(gè)元素就是我們 select 的字段名。

所以 columns = [col[0] for col in cursor.description] 這一行代碼獲取的是指定的所有字段名

使用示例：

def dictfetchall(cursor):
    "Return all rows from a cursor as a dict"
    columns = [col[0] for col in cursor.description]
    return [
        dict(zip(columns, row))
        for row in cursor.fetchall()
    ]

with connection.cursor() as cursor:
	sql = "select id, name from blog_blog"
	cursor.execute(sql)
	results = dictfetchall(cursor)

print(results)

# 返回的結(jié)果：
# [{'id': 20, 'name': 'name_3'}, {'id': 21, 'name': 'name_4'}, {'id': 1, 'name': 'name_5'}]

在我使用的過(guò)程中，我們使用的是上下文管理器的方式來(lái)獲取游標(biāo)：

with connection.cursor() as cursor:
	cursor.execute()

所以，使用完之后，不需要手動(dòng)關(guān)閉，他與下面的用法效果是一致的：

c = connection.cursor()
try:
    c.execute(...)
finally:
    c.close()

但推薦的還是上下文管理器的方式，更優(yōu)雅。

3、多數(shù)據(jù)庫(kù)操作

如果系統(tǒng)用到了多個(gè)數(shù)據(jù)庫(kù)，那么在使用 cursor 的時(shí)候，就需要使用到 django.db.connections 模塊：

from django.db import connections
with connections['my_db_alias'].cursor() as cursor:
    pass

本文詳細(xì)介紹了在Django中使用原生SQL進(jìn)行數(shù)據(jù)庫(kù)查詢的方法，包括優(yōu)點(diǎn)和使用方式。通過(guò)使用原生SQL，我們可以更靈活、更高效地進(jìn)行數(shù)據(jù)庫(kù)操作，特別是在涉及到復(fù)雜的查詢或性能要求較高的情況下。但需要注意的是，使用原生SQL查詢可能會(huì)帶來(lái)安全問(wèn)題，需要謹(jǐn)慎使用，并且要遵守Django框架的開(kāi)發(fā)規(guī)范。

到此這篇關(guān)于Django使用原生SQL查詢數(shù)據(jù)庫(kù)詳解的文章就介紹到這了,更多相關(guān)Django使用原生SQL查詢數(shù)據(jù)庫(kù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論