Improve default value in imports

1 files changed, 15 insertions, 1 deletions

diff --git a/archaeological_files/data_importer.py b/archaeological_files/data_importer.py
index 51809980f..dd6322901 100644
--- a/archaeological_files/data_importer.py
+++ b/archaeological_files/data_importer.py
@@ -41,7 +41,21 @@ class ImportClosingFormater(ImportFormater):
         obj.end_date = open_date + datetime.timedelta(30)
         obj.save()
 
-class FileImporterSraPdL(Importer):
+class FilePostProcessing(object):
+    # erroneaous date to be easily identified on post-processing
+    DEFAULT_YEAR = 1756
+    EXTRA_DEFAULTS = {tuple():{'year':DEFAULT_YEAR}}
+
+    def post_processing(self, item, data):
+        if item.year == self.DEFAULT_YEAR:
+            if item.creation_date:
+                item.year = item.creation_date.year
+            elif item.reception_date:
+                item.year = item.reception_date.year
+        item.save()
+        return item
+
+class FileImporterSraPdL(FilePostProcessing, Importer):
     DESC = u"Exports dossiers SRA PdL : importeur Filemaker dossiers"
     LINE_FORMAT = []
     OBJECT_CLS = models.File