django執行原始查詢sql,並返回Dict字典例子
很多時候執行sql語句,資料比django的model來的快,但並不想關心返回的欄位,例如你可以執行:select * from product這種sql,這裡個方法將會返回與資料庫列名相同的鍵值對 ,格式是這樣子的:
result = [{“id”:1,”name”:”product1”},{“id”:2,”name”:”product2”}]
當然你還可以
import json
json.dumps(result )
返回一串json資料,是不是很完美。。。
# coding:utf-8 from django.db import connection,transaction '''執行django原始sql語句 並返回一個數組物件''' def executeQuery(sql): cursor = connection.cursor() # 獲得一個遊標(cursor)物件 cursor.execute(sql) rawData = cursor.fetchall() print rawData col_names = [desc[0] for desc in cursor.description] print col_names result = [] for row in rawData: objDict = {} # 把每一行的資料遍歷出來放到Dict中 for index,value in enumerate(row): print index,col_names[index],value objDict[col_names[index]] = value result.append(objDict) return result
補充知識:重寫django的mysql驅動實現原生sql語句查詢返回字典型別資料
在使用django的時候,有些需求需要特別高的查詢效率,所以需要使用原生的sql語句查詢,但是查詢結果一般是一個元組巢狀元組。為了處理方便,需要從資料庫查詢後直接返回字典型別的資料。
這裡使用的方法是繼承django.db.backends.mysql驅動
首先在django專案下建立一個mysql資料夾,然後在這個資料夾下建立base.py。
base.py
from django.db.backends.mysql import base from django.db.backends.mysql import features from django.utils.functional import cached_property class DatabaseFeatures(features.DatabaseFeatures): @cached_property def is_sql_auto_is_null_enabled(self): with self.connection.cursor() as cursor: cursor.execute('SELECT @@SQL_AUTO_IS_NULL') result = cursor.fetchone() return result and result['@@SQL_AUTO_IS_NULL'] == 1 class DatabaseWrapper(base.DatabaseWrapper): features_class = DatabaseFeatures def create_cursor(self,name=None): cursor = self.connection.cursor(self.Database.cursors.DictCursor) return base.CursorWrapper(cursor) @cached_property def mysql_server_info(self): with self.temporary_connection() as cursor: cursor.execute('SELECT VERSION()') return cursor.fetchone()['VERSION()']
最後在django專案的settings.py檔案裡修改資料庫配置的資料庫引擎
DATABASES = { 'default': { 'ENGINE': 'Test.mysql',# 指定資料庫驅動為剛剛建立的mysql資料夾 'NAME': 'test',# 指定的資料庫名 'USER': 'root',# 資料庫登入的使用者名稱 'PASSWORD': '123456',# 登入資料庫的密碼 'HOST': '127.0.0.1','PORT': '3306',# 資料庫伺服器埠,mysql預設為3306 'DATABASE_OPTIONS': { 'connect_timeout': 60,'init_command': "SET sql_mode='STRICT_TRANS_TABLES'",'charset': 'utf8mb4',},} }
測試
from django.db import connections def f(): search_sql = "SELECT propertyphotoid,propertyid,alykey FROM lansi_architecture_data.propertyphoto limit 0,5" cursor = connections['default'].cursor() try: cursor.execute(search_sql) rows = cursor.fetchall() except Exception as e: print(e) rows = 1 print(rows)
輸出結果
[{'propertyphotoid': 27,'propertyid': 0,'alykey': '123456'},{'propertyphotoid': 28,'propertyid': 10837,'alykey': 'Property/6113/207504A1-AC65-4E3B-BE86-538B3807D364'},{'propertyphotoid': 29,'propertyid': 6113,'alykey': 'Property/6113/357A4EAE-750A-4F59-AF01-271B4225CFBD'},{'propertyphotoid': 31,'alykey': 'Property/6113/6DF1A2C1-F54C-4462-8363-619806A2F085'},{'propertyphotoid': 36,'alykey': 'Property/6113/572CB245-ABC0-4FD6-8353-729EBD5E5D46'}]
原始碼解析:
django.db.utils.ConnectionHandler的__getitem__方法
獲取連線物件的遊標是由DatabaseWrapper類的create_cursor返回的。所以只需要重寫create_cursor方法,就可以更改遊標返回的資料型別了。
django.db.backends.mysql.base.DatabaseWrapper類中的create_cursor方法如下:
def create_cursor(self,name=None): cursor = self.connection.cursor() return CursorWrapper(cursor)
到這裡,理論上已經完成了重寫目標,但是在測試的時候出錯了,在django.db.backends.mysql.features.DatabaseFeatures裡的is_sql_auto_is_null_enabled方法報出KeyError的錯誤。
@cached_property def is_sql_auto_is_null_enabled(self): with self.connection.cursor() as cursor: cursor.execute('SELECT @@SQL_AUTO_IS_NULL') result = cursor.fetchone() return result and result[0] == 1
原因是is_sql_auto_is_null_enabled方法使用了重寫後的遊標,cursor.execute('SELECT @@SQL_AUTO_IS_NULL')返回的結果不是元組,而是一個字典。所以result[0]會報出KeyError的錯誤。重寫is_sql_auto_is_null_enabled方法把result[0]改成result['@@SQL_AUTO_IS_NULL']就可以了.
最後還需要把DatabaseWrapper類裡的features_class賦值為重寫後的DatabaseFeatures類。
以上這篇django執行原始查詢sql,並返回Dict字典例子就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。