Django怎么使用原生SQL查詢數據庫

這篇文章主要介紹“Django怎么使用原生SQL查詢數據庫”，在日常操作中，相信很多人在Django怎么使用原生SQL查詢數據庫問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Django怎么使用原生SQL查詢數據庫”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

Django 提供了兩種方式來執行原生 SQL 代碼。

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

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

目前而言，官方文檔提供的各種 API 是能夠滿足絕大多數的應用場景的。

1、raw()

這個方法可以用來操作原生 SQL，然后返回 model 實例：

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

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

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

需要注意的是，raw() 并不會去檢測輸入的 SQL 代碼，即使我們使用 Blog 這個model 來查詢 Entry 的數據，也能夠返回結果，不過返回的都是 Entry 表的屬性：

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

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

自定義返回字段

在 raw() 函數的 SQL 代碼里，我們可以自定義選中的字段，如果需要使用沒有選中的字段，那么系統會再次訪問數據庫來獲取，這個操作過程就跟前面介紹的 defer() 函數一樣。

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'}

可以看到，返回的結果中先輸出的數據只有 id，后面，當我們訪問 name 字段的時候，又去獲取了 name 字段的數據。

自定義字段必須包含主鍵

當我們自定義字段返回的時候，必須是要包含主鍵字段的，否則在我們獲取信息的時候會報錯，比如下面的操作：

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

在 print(blog.dict) 的時候就會報錯，數據中沒有主鍵信息

自定義返回新字段

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

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

傳遞變量

給輸入的 SQL 語句傳遞變量：

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

2、connection.cursor()

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

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

需要注意，如果有參數傳入 SQL 中，需要對一些符號進行轉義之后才可以使用，比如：

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

其中，第二條語句的 % 需要進行轉義

適配fetch的數據

通過 fetchone 和 fetchall() 返回的數據是只有 value 值的，沒有對應的字段 key，如果可以適當的犧牲性能和內存，來換取獲取數據的便利和準確性，官方提供了這樣一種方式：

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()
    ]

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

介紹一下 cursor.description，這個返回的是一個元組數據，里面的元素也是一個元組，元組的第一個元素就是我們 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)

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

在我使用的過程中，我們使用的是上下文管理器的方式來獲取游標：

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

所以，使用完之后，不需要手動關閉，他與下面的用法效果是一致的：

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

但推薦的還是上下文管理器的方式，更優雅。

3、多數據庫操作

如果系統用到了多個數據庫，那么在使用 cursor 的時候，就需要使用到 django.db.connections 模塊：

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

到此，關于“Django怎么使用原生SQL查詢數據庫”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！