flask_sqlalchemy 查詢結果轉dict 終極解決方案
之前為了學習Python,試著拿Flask作框架搞小網站,感覺還不錯,基本就拋棄了PHP。前段時間做了一個微信小程式,想著yii框架拿來寫幾十個小介面是不是浪費了,就繼續用flask寫api了,哪想到填坑無數啊。
Python的ORM框架就屬Sqlalchemy牛逼,網上資料也多,想著和yii裡面應該差不多,就拿來用了。第二天萬萬沒想到,php裡面簡單的一句asArray就能解決的問題,flask_sqlalchemy居然沒有解決方案,查詢的結果物件無法直接JSON序列化。這期間從南到北地找,大部分解決方案都是做一個JSON.dumps的Encoder方法,來轉化restult物件,無意中看見https://www.cnblogs.com/wancy86/p/6421792.html
原文連結:https://www.cnblogs.com/eating-gourd/p/9997751.html
咳咳,正文:
網上的方法主要問題在於只能處理result物件或model物件之一,當查詢某個表全部欄位時,如
1 db.session.query(User).filter().all()
其返回User這個類的物件列表,而查詢某些欄位或者多表連線時,如:
1 db.session.query(User.UserID,User.UserName).filter().all()
其返回result物件的列表,這兩種情況下,物件的屬性不同,導致很多情況下只能適應一種返回。今天趁閒著沒事,把兩種情況的查詢結果轉dict作了一下整理,封裝為一個queryToDict函式,並同時支援all()返回的列表和first()返回的單個物件結果:
from datetime import datetime as cdatetime #有時候會返回datatime型別 from datetime import date,time from flask_sqlalchemy import Model from sqlalchemy.orm.query import Query from sqlalchemy import DateTime,Numeric,Date,Time #有時又是DateTime def queryToDict(models): if(isinstance(models,list)): if(isinstance(models[0],Model)): lst = [] for model in models: gen = model_to_dict(model) dit = dict((g[0],g[1]) for g in gen) lst.append(dit) return lst else: res = result_to_dict(models) return res else: if (isinstance(models, Model)): gen = model_to_dict(models) dit = dict((g[0],g[1]) for g in gen) return dit else: res = dict(zip(models.keys(), models)) find_datetime(res) return res #當結果為result物件列表時,result有key()方法 def result_to_dict(results): res = [dict(zip(r.keys(), r)) for r in results] #這裡r為一個字典,物件傳遞直接改變字典屬性 for r in res: find_datetime(r) return res def model_to_dict(model): #這段來自於參考資源 for col in model.__table__.columns: if isinstance(col.type, DateTime): value = convert_datetime(getattr(model, col.name)) elif isinstance(col.type, Numeric): value = float(getattr(model, col.name)) else: value = getattr(model, col.name) yield (col.name, value) def find_datetime(value): for v in value: if (isinstance(value[v], cdatetime)): value[v] = convert_datetime(value[v]) #這裡原理類似,修改的字典物件,不用返回即可修改 def convert_datetime(value): if value: if(isinstance(value,(cdatetime,DateTime))): return value.strftime("%Y-%m-%d %H:%M:%S") elif(isinstance(value,(date,Date))): return value.strftime("%Y-%m-%d") elif(isinstance(value,(Time,time))): return value.strftime("%H:%M:%S") else: return ""
dit = dict((g[0],g[1]) for g in gen)相關程式碼也是之前查詢資料獲得,現在找不到出處了,作者可以聯絡我。
滾去學雅思了,程式碼寫得較快,歡迎指出bug
參考資源:
[1] https://stackoverflow.com/questions/5022066/how-to-serialize-sqlalchemy-result-to-json
[2] https://segmentfault.com/q/1010000007459402/a-1020000007460322