Django中如何直接執(zhí)行SQL語(yǔ)句

更新時(shí)間：2023年08月24日 09:55:34 作者：程序員學(xué)編程

這篇文章主要介紹了Django中如何直接執(zhí)行SQL語(yǔ)句問(wèn)題,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

Django直接執(zhí)行SQL語(yǔ)句

今天在django views.py看到同事寫(xiě)的代碼里面有段關(guān)于數(shù)據(jù)庫(kù)查詢(xún)的語(yǔ)句。

因?yàn)樯婕岸鄠€(gè)表的查詢(xún)，所以django 的models的查詢(xún)無(wú)法滿(mǎn)足需求，所以直接執(zhí)行了SQL語(yǔ)句。

他是按照下面的方法實(shí)現(xiàn)的。

    try:
        connection = MySQLdb.connect(host=SQL_IP,
                                     user=SQL_USER,
                                     passwd=SQL_PASSWD,
                                     db=SQL_DB, charset='utf8')
        cursor = connection.cursor()
        cmd_query = query_data(request)
        data_sort = sort_data(request)
        cmd = "set session group_concat_max_len = 8000;"
        cursor.execute(cmd)
        ......

看到這段代碼，感覺(jué)應(yīng)該重復(fù)造了輪子，數(shù)據(jù)庫(kù)的鏈接應(yīng)該交由django這個(gè)框架處理的。于是查看了下官方文檔。

django.db.connection 就是上面的同事寫(xiě)的那個(gè)connection了。

后面的方法同他的一樣執(zhí)行就是了。即執(zhí)行

connection.cursor()
cursor.execute(sql, [params])
cursor.fetchone() or cursor.fetchall()

下面是一個(gè)簡(jiǎn)單的例子。

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

如果你項(xiàng)目中有多個(gè)數(shù)據(jù)庫(kù)的話(huà)，django.db.connection也能很方便的選取數(shù)據(jù)庫(kù)。

django.db.connections 是一個(gè)類(lèi)字典的結(jié)構(gòu)，允許你使用數(shù)據(jù)庫(kù)別名獲取connection。

from django.db import connections
with connections['my_db_alias'].cursor() as cursor:
    # Your code here...

使用with作為上下文管理器：

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

和下面這段代碼等效

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

注意的問(wèn)題：

如果你做了更新或者插入操作需要在代碼中使用：transaction.commit_unless_managed() 來(lái)提交數(shù)據(jù)。

或者使用事務(wù)裝飾器(例如 commit_on_success)來(lái)修飾視圖和提供事務(wù)控制數(shù)據(jù)提交。

這樣就不用在代碼中調(diào)用transaction.commit_unless_managed()。

但是，如果你不手動(dòng)提交修改，你需要使用 transaction.set_dirty() 將事務(wù)標(biāo)識(shí)為已臟。

使用 Django ORM 對(duì)數(shù)據(jù)庫(kù)進(jìn)行修改時(shí)，Django 會(huì)自動(dòng)調(diào)用 set_dirty() 。

但如果你使用了原始 SQL ，Django 就無(wú)法獲得你的 SQL 是否修改了數(shù)據(jù)。

只有手動(dòng)調(diào)用 set_dirty() 才能確保 Django 知曉哪些修改必須被提交。

from django.db.transaction import commit_on_success
@commit_on_success
def my_custom_sql_view(request, value):
    from django.db import connection, transaction
    cursor = connection.cursor()
    # Data modifying operation
    cursor.execute("UPDATE bar SET foo = 1 WHERE baz = %s", [value])
    # Since we modified data, mark the transaction as dirty
    transaction.set_dirty()
    # Data retrieval operation. This doesn't dirty the transaction,
    # so no call to set_dirty() is required.
    cursor.execute("SELECT foo FROM bar WHERE baz = %s", [value])
    row = cursor.fetchone()
    #transaction.commit_unless_managed()
    return render_to_response('template.html', {'row': row})

如果不使用

參考文獻(xiàn)：1、https://docs.djangoproject.com/en/2.0/topics/db/sql/

Django底層實(shí)現(xiàn)sql語(yǔ)句

在django中我們可以自己手寫(xiě)底層sql語(yǔ)句，那么如何寫(xiě)呢？

#導(dǎo)入原生sql模塊
from django.db import connection
# 獲取當(dāng)前用戶(hù)關(guān)注課程列表
def get_myflow(request):
    # 獲取用戶(hù)id
    uid = request.GET.get('id')
    # 建立游標(biāo)對(duì)象
    cursor = connection.cursor()
    # 鏈表查詢(xún)
    cursor.execute('select a.username,c.title from user a left join flows b on a.id=b.uid left join course c on b.cid=c.id where a.id = %s'% uid)
    # 獲取結(jié)果集
    result = dictfetchall(cursor)
    print(result)
    # 返回結(jié)果
    return HttpResponse(json.dumps(result),content_type='application/json')