Skip to content

Commit 06fbf5c

Browse files
committed
Merge branch 'Profile-export-serializer' of https://github.com/Zdruzenie-STROM/webstrom-backend into Profile-export-serializer
2 parents 22c1706 + 2f4e40d commit 06fbf5c

File tree

1 file changed

+17
-0
lines changed

1 file changed

+17
-0
lines changed

competition/views.py

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,14 @@
11
import os
22
import zipfile
3+
import csv
34
from io import BytesIO
45
from operator import itemgetter
56

67
from django.core.files import File
78
from django.core.mail import send_mail
89
from django.http import FileResponse, HttpResponse
910
from django.template.loader import render_to_string
11+
from django.views.generic import View
1012
from rest_framework import exceptions, mixins, status, viewsets
1113
from rest_framework.decorators import action
1214
from rest_framework.exceptions import ValidationError
@@ -688,6 +690,21 @@ def participants(self, request, pk=None):
688690

689691
profiles = Profile.objects.only("user").filter(pk__in=participants_id)
690692
serializer = ProfileExportSerializer(profiles, many=True)
693+
694+
#if request == 'csv':
695+
696+
response = HttpResponse(content_type='text/csv')
697+
response['Content-Disposition'] = 'attachment; filename="export.csv"'
698+
699+
header = ProfileExportSerializer.Meta.fields
700+
701+
writer = csv.DictWriter(response, fieldnames=header)
702+
writer.writeheader()
703+
for row in serializer.data:
704+
writer.writerow(row)
705+
706+
return response
707+
691708
return Response(serializer.data)
692709

693710
def post(self, request, format_post):

0 commit comments

Comments
 (0)