1. 程式人生 > >flask_sqlalchemy 查詢結果轉dict 終極解決方案

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

這個帖子,說queryresult物件加入了json屬性,欣喜萬分,搞了一晚上也沒找到這個方法。

  原文連結: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