从零开始搭建django前后端分离项目 系列五（实战之excel流式导出）

@csrf_exempt
def exportData(request):
    format = request.GET.get(‘format‘)
    pk = request.GET.get(‘pk‘)
    export_sql=SqlModel.objects.get(pk=pk).export_sql
    if format==‘csv‘:
        response = StreamingHttpResponse((row for row in FileHandle.csv_stream_response_generator(export_sql)),content_type="text/csv;charset=utf-8")
        response[‘Content-Disposition‘] = ‘attachment; filename="query_result.csv"‘
        return response

def csv_stream_response_generator(export_sql):
    db = MySQLdb.connect("10.39.211.198", "root", "password", "busycell", charset=‘utf8‘)
    chunk_size = 30000
    offset = 0
    yield codecs.BOM_UTF8
    while True:
        isHeader = False
        print(offset)
        if offset == 0:
            isHeader = True
        sql=export_sql + " limit {1} offset {0}".format(offset, chunk_size)
        df = pd.read_sql(sql, db)
        f = StringIO()
        df.to_csv(f, index=False, header=isHeader, encoding="utf_8_sig")
        yield f.getvalue()
        offset += chunk_size
        if df.shape[0] < chunk_size:
            break