如何在Django API中使用xlsx将这些模型数据导出到Excel表中

时间:2018-10-31 04:04:09

标签: api django-rest-framework

"""when user call this api, automatically downloads user xlsx file with data"""## Heading ##

类DownloadUsersForHarshXlsx(APIView):     def get(self,request,* args,** kwargs):

    user_data = api_utils.get_all_user_details(request)
    workbook = xlsxwriter.Workbook('users.xlsx')
    worksheet = workbook.add_worksheet()
    row = 0
    col = 0
    for user in user_data:
        # import pdb; pdb.set_trace()
        worksheet.write(row, col, user.get('first_name'))
        worksheet.write(row, col + 1, user.get('email'))
        worksheet.write(row, col + 2, user.get('age'))
        worksheet.write(row, col + 3, user.get('id'))
        worksheet.write(row, col + 4, user.get('membership_id'))
        worksheet.write(row, col + 5, user.get('segment'))
        row += 1
    workbook.close()
    response = HttpResponse(content_type='application/vnd.ms-excel')
    response['Content-Disposition'] = 'attachment; filename="users.xlsx"'
    return response

0 个答案:

没有答案