JobOffer : check extension file

diff --git a/app/organization/job/views.py b/app/organization/job/views.py
index 5d3f4c2d1abc5cd1e407274f3eba3a66909a7fa3..a540c18172f4ab268995b062dfb8ff9cb356c3fa 100644 (file)
--- a/app/organization/job/views.py
+++ b/app/organization/job/views.py
@@ -1,4 +1,6 @@
 import os
+from django import forms
+from django.shortcuts import redirect
 from django.shortcuts import render
 from django.views.generic import DetailView, ListView, TemplateView
 from django.views.generic.edit import CreateView
@@ -8,6 +10,7 @@ from django.utils.translation import ugettext_lazy as _
 from organization.job.models import JobOffer, JobResponse
 from organization.job.forms import JobResponseForm
 
+extention = ['.pdf', '.PDF', '.doc', '.docx']
 
 class JobOfferDetailView(CreateView):
 
@@ -35,10 +38,11 @@ class JobOfferDetailView(CreateView):
 
     def form_valid(self, form):
         # check extension uploaded files
-        # name_cv, ext_cv = os.path.splitext(self.cleaned_data['curriculum_vitae'].name)
-        # name_cl, ext_cl = os.path.splitext(self.cleaned_data['cover_letter'].name)
-        # if ext_cv not in ['.pdf', '.PDF', '.doc', '.docx']:
-        #     raise forms.ValidationError("Only .pdf, .doc, .docx files allowed")
+        name_cv, ext_cv = os.path.splitext(form.cleaned_data['curriculum_vitae'].name)
+        name_cl, ext_cl = os.path.splitext(form.cleaned_data['cover_letter'].name)
+        if ext_cv not in extention or ext_cl not in extention :
+            messages.info(self.request, _("Only .pdf, .doc, .docx files allowed."))
+            return super(JobOfferDetailView, self).form_invalid(form)
         messages.info(self.request, _("You have successfully submitted your application."))
         return super(JobOfferDetailView, self).form_valid(form)