D-django自定义导出csv文件

最新推荐文章于 2022-02-19 21:49:31 发布

转载最新推荐文章于 2022-02-19 21:49:31 发布 · 147 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/bk770466199/p/7909790.html

文章标签：

#python

本文介绍了一个Python脚本，该脚本用于从数据库中获取用户信息并将其导出为CSV文件。它演示了如何使用Python的csv模块来创建CSV响应，并如何处理不同类型的字段值，包括日期和枚举类型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import csv
from datetime import timedelta

def download_csv(commend=None):
    response = HttpResponse(content_type='text/csv')
    response['Content-Disposition'] = 'attachment;filename=download.csv'
    response.write('\xEF\xBB\xBF')
    
    writer = csv.writer(response)
    headers = [u"姓名",u"年龄",u"性别"]
    headers = [item.encode('utf8') for item in headers]
    writer.writerow(headers)
　　list_user = User.object.filter(**commend).order_by("-datetime_created")
　　queryset = list_user.values_list("user__username","age","sex")
　　for idx,row in enumerate(queryset):
　　　　values = []
　　　　cnt = 0
　　　　for field in row:
    　　　　cnt += 1
    　　　　value = ''
　　　　　　if cnt == 2:
　　　　　　　　if int(field) == 0:
　　　　　　　　　　value = u'男'
　　　　　　　　elif int(field) == 1:
　　　　　　　　　　value = u'女'
　　　　　　elif field:
    　　　　　　if type(field) is datetime.datetime or type(field) is datetime.date:
        　　　　　　field = field + timedelta(hours=8)
        　　　　　　value = field.strftime("%Y-%m-%d")
    　　　　　　else:
        　　　　　　value = field　
　　　　　　　values.append(unicode(value).encode('utf-8'))
　　　　writer.writerow(values)
　　return response