claudep/epcstages
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add contacts infos in stages export

Browse source
This commit is contained in:
Claude Paroz 2013-02-07 09:53:12 +01:00
parent 9a87fd6c81
commit 4c12886e73
1 changed files with 18 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

23
stages/views.py
View file

@ -10,7 +10,8 @@ from django.shortcuts import get_object_or_404
from django.views.generic import DetailView, TemplateView, ListView
from .forms import PeriodForm
from .models import Section, Student, Corporation, Period, Training, Referent, Availability
from .models import (Section, Student, Corporation, CorpContact, Period,
Training, Referent, Availability)
def school_year_start():
@ -137,12 +138,13 @@ def stages_export(request):
('Prénom', 'student__first_name'), ('Nom', 'student__last_name'),
('Classe', 'student__klass__name'), ('Filière', 'student__klass__section__name'),
('Début', 'availability__period__start_date'), ('Fin', 'availability__period__end_date'),
('Prénom référent', 'referent__first_name'), ('Nom référent', 'referent__last_name'),
('Institution', 'availability__corporation__name'),
('Rue Inst.', 'availability__corporation__street'),
('NPA Inst.', 'availability__corporation__pcode'),
('Ville Inst.', 'availability__corporation__city'),
('Domaine', 'availability__domain__name'),
('Prénom référent', 'referent__first_name'), ('Nom référent', 'referent__last_name')
('Civilité contact', None), ('Prénom contact', None), ('Nom contact', None),
]
period_filter = request.GET.get('filter')
@ -151,6 +153,11 @@ def stages_export(request):
else:
query = Training.objects.all()
contacts = {}
for contact in CorpContact.objects.all().select_related('corporation').order_by('corporation'):
if contact.corporation.name not in contacts or contact.is_main is True:
contacts[contact.corporation.name] = contact
wb = Workbook()
ws = wb.get_active_sheet()
ws.title = 'Stages'
@ -159,9 +166,15 @@ def stages_export(request):
ws.cell(row=0, column=col_idx).value = header
ws.cell(row=0, column=col_idx).style.font.bold = True
# Data
for row_idx, tr in enumerate(query.values_list(*[f[1] for f in export_fields]), start=1):
for col_idx, field in enumerate(tr):
ws.cell(row=row_idx, column=col_idx).value = field
query_keys = [f[1] for f in export_fields if f[1] is not None]
for row_idx, tr in enumerate(query.values(*query_keys), start=1):
for col_idx, field in enumerate(query_keys):
ws.cell(row=row_idx, column=col_idx).value = tr[field]
contact = contacts.get(tr['availability__corporation__name'])
if contact:
ws.cell(row=row_idx, column=col_idx+1).value = contact.title
ws.cell(row=row_idx, column=col_idx+2).value = contact.first_name
ws.cell(row=row_idx, column=col_idx+3).value = contact.last_name
response = HttpResponse(save_virtual_workbook(wb), mimetype='application/ms-excel')
response['Content-Disposition'] = 'attachment; filename=%s%s.xlsx' % (