PJ
[auf_rh_dae.git] / project / views.py
index c69b5a6..10efbef 100644 (file)
@@ -1,8 +1,51 @@
 # -*- encoding: utf-8 -*-
+
+import os
+from sendfile import sendfile
+
+from django.conf import settings
 from django.shortcuts import render_to_response
 from django.template import RequestContext
+from django.contrib.auth.decorators import login_required
+from django.shortcuts import redirect, render_to_response, get_object_or_404
+
+# pas de reference a DAE devrait etre refactorisé
 from project.dae.decorators import dae_groupe_requis
+from project.dae.decorators import redirect_interdiction
 
 @dae_groupe_requis
 def index(request):
     return render_to_response('index.html', {}, RequestContext(request))
+
+@login_required
+def piece(request, filename):
+    """Téléchargement d'une pièce jointe à un poste."""
+    # compatibilité avec DAE prod avant sécurité
+    try:
+        app, model, id, f = filename.split('/')
+    except:
+        path = os.path.join(settings.PRIVE_MEDIA_ROOT, filename)
+        return sendfile(request, path)
+
+    application = __import__(app).models
+    if model == 'contrat':
+        # TODO definir peut-être un controle d'accès
+        piece = get_object_or_404(application.Contrat, fichier=filename)
+        return sendfile(request, piece.fichier.path)
+    if model == 'employe':
+        # TODO definir peut-être un controle d'accès
+        piece = get_object_or_404(application.EmployePiece, fichier=filename)
+        return sendfile(request, piece.fichier.path)
+    if model == 'poste':
+        piece = get_object_or_404(application.PostePiece, fichier=filename)
+        if application.Poste.objects.ma_region_ou_service(request.user).filter(id=piece.poste_id).exists():
+            return sendfile(request, piece.fichier.path)
+        else:
+            return redirect_interdiction(request)
+    if model == 'dossier':
+        piece = get_object_or_404(application.DossierPiece, fichier=filename)
+        if application.Dossier.objects.ma_region_ou_service(request.user).filter(id=piece.dossier_id).exists():
+            return sendfile(request, piece.fichier.path)
+        else:
+            return redirect_interdiction(request)
+