我该如何编写一个用sqlite3db填充的CSV文件

Question

对于如何使用给定用户的models.py中的信息填充以下csv函数，我有点困惑。有人能给我指正确的方向吗？我需要在一个罕见的separapy文件中处理信息吗？还是可以在我的视图中处理？

我的视图下载信息

def download(request):
    response = HttpResponse(mimetype='text/csv')
    response['Content-Disposition'] = 'attachment; filename=UserData.csv'
    writer = csv.writer(response)
    writer.writerow(['Date', 'HighBGL', 'LowBGL', 'Diet', 'Weight', 'Height', 'Etc'])
    writer.writerow(['Info pertaining to date 1'])
    writer.writerow(['info pertaining to date 2'])
    return response

其中一个模型谁的信息我有兴趣保存

class DailyVital(models.Model):
    user =  models.ForeignKey(User)
    entered_at = models.DateTimeField()
    high_BGL = models.IntegerField()
    low_BGL = models.IntegerField()
    height = models.IntegerField(blank = True, null = True)
    weight = models.IntegerField(blank = True, null = True)

Answer 1

首先需要查询django模型，比如：DailyVital.objects.all()或DailyVital.objects.filter(user=request.user)

然后，您可以手动将对象转换为元组，也可以使用Django QuerySet的values_list方法和字段名列表来返回元组而不是对象。类似于：

def download(request):
    response = HttpResponse(mimetype='text/csv')
    response['Content-Disposition'] = 'attachment; filename=UserData.csv'
    writer = csv.writer(response)
    writer.writerow(['Date', 'HighBGL', 'LowBGL', 'Weight', 'Height'])
    query = DailyVital.objects.filter(user=request.user)
    for row in query.values_list('entered_at', 'high_BGL', 'low_BGL', 'weight', 'height'):
       writer.writerow(row)
    return response

如果您在Django中不需要它，您也可以考虑sqlite3命令行程序的-csv选项。

Answer 2

一个简单的方法是将模型转换为列表列表。

首先你需要一个对象列表功能：

def object2list(obj, attr_list):
    " returns values (or None) for the object's attributes in attr_list"
    return [getattr(obj, attr, None) for attr in attr_list]

然后，您只需将其传递给具有列表理解的csvwriter（给定您查询的一些list_of_objects）

attr_list = ['date', 'high_BGL', 'low_BGL', 'diet', 'weight', 'height']
writer.writerows([object2list(obj, attr_list) for obj in list_of_objects])