优化从mongdb取数据python代码，用于出fusioncharts图

原始代码

 

def ioBandwidth(request):
    beginDate = request.GET.get("beginDate")
    collection = getCollection()
    data = collection.find().sort("InsertDate",pymongo.ASCENDING)
    mscombi2d.categories = {"category": [{"label":time.strftime("%H:%M",time.strptime(d["InsertDate"],"%Y-%m-%d %H:%M:%S"))} for d in data]}
    data.rewind()#因为开始是用find返回的，所以是游标，取完后要回退到开始才能重新取
    temp = data.clone()#因为后面有两次，只能先把游标拷贝，以方便后面的取值

    mscombi2d.dataset = [
            {"seriesname": u"入带宽", "parentyaxis": "P", "color": "5a885e","renderas":"Area",
             "data": [{"value": (d["InBandwidth"])} for d in data]
        },
            {"seriesname": u"出带宽", "parentyaxis": "S", "renderas": "Line", "color": "dca657", "anchorbgcolor": "dca657",
             "data": [{"value": (d["OutBandwidth"])} for d in temp]
        }
    ]
    res = json.dumps(mscombi2d.to_dict(), ensure_ascii=False, sort_keys=True, indent=4)

    return HttpResponse(res)

 

修改后代码

 

 

def ioBandwidth(request):
    beginDate = request.GET.get("beginDate")
    collection = getCollection()
    data = [d for d in collection.find()]#把游标内数据存入list
    mscombi2d = MSCombi2D(u"大节点带宽",beginDate)
    mscombi2d.categories = {"category": []}
    mscombi2d.dataset = [
            {"seriesname": u"入带宽", "parentyaxis": "P", "color": "5a885e","renderas":"Area",
             "data": []
        },
            {"seriesname": u"出带宽", "parentyaxis": "S", "renderas": "Line", "color": "dca657", "anchorbgcolor": "dca657",
             "data":[]
        }
    ]
    labels = []#时间标签
    inband = []#入带宽
    outband = []#出带宽
    for d in data:
        labels.append({"label":time.strftime("%H:%M",time.strptime(d["InsertDate"],"%Y-%m-%d %H:%M:%S"))})
        inband.append({"value":"%s"%(d["InBandwidth"])})
        outband.append({"value":"%s"%(d["OutBandwidth"])})

    mscombi2d.dataset[0]["data"] = inband
    mscombi2d.dataset[1]["data"] = outband
    
    res = json.dumps(mscombi2d.to_dict(), ensure_ascii=False, sort_keys=True, indent=4)

    return HttpResponse(res)

 

 

1）从游标取出数据存入list，不需要再rewind和clone了

2）只用了一次循环，不像前面循环了三次。

3）查找用索引，而不要用sort