django csv download of queryset

Question

Here is my view..

class SingleNewsView(ListView):
    model = News
    form_class = SearchForm
    template_name = "single_news.html"



    def get(self, request, pk, **kwargs):
        self.pk = pk

        self.pub_from = request.GET.get('pub_date_from')
        self.pub_to = request.GET.get('pub_date_to')
        self.crawlers = request.GET.get('crawler')

        print self.crawlers


        return super(SingleNewsView,self).get(request,pk, **kwargs)

    def get_context_data(self, **kwargs):

        context = super(SingleNewsView,self).get_context_data(**kwargs)
        context["form"] = SearchForm#(self.request.GET)
        context["something"] = News.objects.filter(category_id=self.pk).filter(published_date__range=(self.pub_from,self.pub_to), crawler=self.crawlers)

        return context

Here on the same page where this view is displayed I want the "Download CSV" option and want to download the csv of query returned by context["something"]. How can I do that ?? Any help..

I have also written view to download csv but confused..

def CSVView(request):
    response = HttpResponse(content_type="text/csv")
    response["Content-Disposition"] = 'attachment; filename=somefilename.csv"'

    writer = csv.writer(response)
    writer.writerow([News.objects.filter(what goes here ???])

    return response

As above I have filtered context["something"]. I want the csv of that context. There I have applied the filter — Aaeronn, Jun 20 '14 at 11:07

AlvaroAV · Answer 1 · 2014-06-20T11:34:50.877

1

This is how I create a csv file, fill it and send it to template for download:

def my view():
...    ...    ...
output_file = open('path/new_file_name', 'wb')
writer = csv.writer(output_file, delimiter=';')  # Delimiter is not needed
# This is to show a "title" per column:
headers = ['column1', 'column2', 'column3', 'column4']  
writer.writerow(headers)
for item in item_list:  # Then add the info you need to the file from the objects
    writer.writerow('write what you need for each item')

response = StreamingHttpResponse(writer, content_type="text/csv")
response['Content-Disposition'] = 'attachment; filename="FILENAME.csv"'
return response

When you use StreamingHttpResponse and create the response like I show you, when you do the return response, automatically it should start to download the file you send back

edited Jun 20 '14 at 11:34

answered Jun 20 '14 at 11:28

AlvaroAV

10,335
12
60
91

This is not what he asked. – Burhan Khalid Jun 20 '14 at 11:39
Isn't he trying to créate a csv file and download it ?? – AlvaroAV Jun 20 '14 at 11:41
He is trying to download vía csv the content of something, im explaining how to generate the file and send it to the template for downloading – AlvaroAV Jun 20 '14 at 11:43
You are not sending anything to the template here and the question is how to capture the context in a class view, to generate the filtered list of objects which will then be converted into a CSV. – Burhan Khalid Jun 20 '14 at 11:50
Im not sending anything to the template ? Does not "return response" send the file to the template ? – AlvaroAV Jun 20 '14 at 12:08
Im showing how to generate csv file, add content to the file, and download that generated csv file – AlvaroAV Jun 20 '14 at 12:09

mariodev · Accepted Answer · 2014-06-20T12:27:23.643

Recently I have found django-queryset-csv library to be very useful. With this, your code comes down to:

from djqscsv import render_to_csv_response


class SingleNewsView(ListView):
    model = News
    form_class = SearchForm
    template_name = "single_news.html"

    def get(self, request, pk, **kwargs):
        self.pk = pk

        self.pub_from = request.GET.get('pub_date_from')
        self.pub_to = request.GET.get('pub_date_to')
        self.crawlers = request.GET.get('crawler')

        self.format = request.GET.get('format', False)

        # check for format query key in url (my/url/?format=csv)
        if self.format == 'csv':
            qs = News.objects.filter(category_id=self.pk)\
                             .filter(published_date__range=(self.pub_from,self.pub_to), crawler=self.crawlers)\
                             .values('id', 'name')
            return render_to_csv_response(qs)

        print self.crawlers

        return super(SingleNewsView,self).get(request,pk, **kwargs)

    def get_context_data(self, **kwargs):

        context = super(SingleNewsView,self).get_context_data(**kwargs)
        context["form"] = SearchForm#(self.request.GET)
        context["something"] = News.objects.filter(category_id=self.pk).filter(published_date__range=(self.pub_from,self.pub_to), crawler=self.crawlers)

        return context

Of course you need to use your own column names that you want to output, instead of id, name..

Now since I don't see how you render the download csv button, I can't tell you how to add format key to it's url..

I m just putting a hyperlink to "Download CSV". How can I send the format then ??? — Aaeronn, Jun 23 '14 at 04:15

django csv download of queryset

2 Answers2

Linked