c4patino/csmb-housewars
Archived
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Updated csv magic button

Browse source
This commit is contained in:
Ceferino Patino 2022-12-12 18:30:20 -06:00
commit 6c7c5ad188
1 changed files with 7 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

13
housewars/admin/UserEntry.py
View file

@ -72,19 +72,20 @@ class UserEntryAdmin(admin.ModelAdmin):
@admin.action(description='Export selected to csv')
def export_to_csv(self, request, queryset):
# Augment the queryset with extra data
queryset = queryset
# Augment the querset with extra data
queryset = queryset.annotate(
a1_room=F('activity1__room_number')).annotate(a2_room=F('activity1__room_number'))
# Initialize csv object writers
response = HttpResponse(content_type='text/csv', headers={
'Content-Disposition': 'attachment; filename="export.csv"'},)
writer = csv.writer(response, delimiter=";")
# Select the headers that are to be unpacked
headers = ['first_name', 'last_name', 'email',
'grade', 'house', 'activity1', 'activity2']
# Select headers that are to be unpacked
headers = ['first_name', 'last_name', 'grade', 'house',
'activity1', 'a1_room', 'activity2', 'a2_room']
# Unpack queryset and write to file
# Unpack queryset data
data = unpack_values(queryset, headers)
# Write data into tables