Using Django Rest Framework to ease csv generation

models.py

from django.db import models

class SomeModel(models.Model):
    
    email = models.EmailField()
    first_name = models.CharField(max_length=50)
    last_name = models.CharField(max_length=50)

serializers.py

from rest_framework import serializers

from .models import SomeModel


class SomeModelSerializer(serializers.ModelSerializer):
    
    class Meta:
        model = SomeModel
        fields = (
            'email',
            'first_name',
            'last_name',
        )

views.py

import csv

from django.views.generic import View
from django.http import HttpResponse

from .serializers import SomeModelSerializer
from .models import SomeModel


class SomeModelCSVExportView(View):

    serializer_class = SomeModelSerializer
    
    def get_serializer(self, queryset, many=True):
        return self.serializer_class(
            queryset,
            many=many,
        )

    def get(self, request, *args, **kwargs):
        response = HttpResponse(content_type='text/csv')
        response['Content-Disposition'] = 'attachment; filename="export.csv"'
        
        serializer = self.get_serializer(
            SomeModel.objects.all(),
            many=True
        )
        header = SomeModelSerializer.Meta.fields
        
        writer = csv.DictWriter(response, fieldnames=header)
        writer.writeheader()
        for row in serializer.data:
            writer.writerow(row)
        
        return response

this just helped me to solve some issues I was having