pieces jointes

[auf_rh_dae.git] / project / dae / models.py

diff --git a/project/dae/models.py b/project/dae/models.py
index f6a0640..cef5947 100644 (file)
--- a/project/dae/models.py
+++ b/project/dae/models.py
@@ -1,10 +1,12 @@
 # -=- encoding: utf-8 -=-
 
+import os
+from django.core.files.storage import FileSystemStorage
 from django.db import models
 import reversion
 import datamaster_modeles.models as ref
 from rh_v1 import models as rh
-
+import settings
 
 STATUT_RESIDENCE_CHOICES = (
     ('local', 'Local'),
@@ -21,6 +23,19 @@ POSTE_STATUT_CHOICES = (
     ('DET', 'Détachement'),
 )
 
+
+storage_prive = FileSystemStorage(settings.PRIVE_MEDIA_ROOT, base_url=settings.PRIVE_MEDIA_URL)
+
+def poste_piece_dispatch(instance, filename):
+    path = "poste/%s/%s" % (instance.poste.id, filename)
+    return path
+
+
+class PostePiece(models.Model):
+    poste = models.ForeignKey("Poste")
+    nom = models.CharField(verbose_name="Nom", max_length=255)
+    fichier = models.FileField(verbose_name="Fichier", upload_to=poste_piece_dispatch, storage=storage_prive)
+
 class PosteManager(models.Manager):
     """
     Chargement de tous les objets FK existants sur chaque QuerySet.