poste import + liens dans RH
[auf_rh_dae.git] / project / dae / exporter.py
index 08f4e07..5b64c0f 100644 (file)
@@ -90,9 +90,17 @@ class SuperCopier(object):
     
         return copy
 
+    def copy(self, obj):
+        return self.duplicate(obj)
+
 
 class PosteCopier(SuperCopier):
 
+    def copy(self, obj):
+        copy = super(PosteCopier, self).copy(obj)
+        dae.ImportPoste(dae=obj, rh=copy).save()
+        return copy
+
     def clean_rh_dossiers(self, source, copy, parent, field, value):
         self.out(SKIP, 1)
         return []
@@ -140,6 +148,11 @@ class PosteCopier(SuperCopier):
 
 class DossierCopier(SuperCopier):
 
+    def copy(self, obj):
+        copy = super(DossierCopier, self).copy(obj)
+        dae.ImportDossier(dae=obj, rh=copy).save()
+        return copy
+
     def clean_rh_contrats(self, source, copy, parent, field, value):
         self.out(SKIP, 1)
         return []
@@ -179,11 +192,11 @@ class DossierCopier(SuperCopier):
             return
 
         if dae.ImportPoste.objects.filter(dae=value).exists():
-            poste = dae.ImportPoste.get(dae=value).rh
+            poste = dae.ImportPoste.objects.get(dae=value).rh
         else:
             copier = PosteCopier(verbosity=self.verbosity,
                     dry_run=self.dry_run)
-            poste = copier.duplicate(value)
+            poste = copier.copy(value)
         copy.poste_id = poste.id
 
     def clean_fichier(self, source, copy, parent, field, value):