SRA import: preventive by default - auto aliment raw fields on file save

1 files changed, 12 insertions, 0 deletions

diff --git a/archaeological_files/models.py b/archaeological_files/models.py
index 9bcae4280..8ac5cddde 100644
--- a/archaeological_files/models.py
+++ b/archaeological_files/models.py
@@ -330,6 +330,18 @@ class File(BaseHistorizedItem, OwnPerms, ValueGetter, ShortMenuItem,
         returned = super(File, self).save(*args, **kwargs)
         if self.main_town and not self.main_town in list(self.towns.all()):
             self.towns.add(self.main_town)
+        if self.responsible_town_planning_service \
+           and not self.raw_town_planning_service:
+            lbl = unicode(self.responsible_town_planning_service)
+            if lbl:
+                self.raw_town_planning_service = lbl
+                self.save
+        if self.general_contractor \
+           and not self.raw_general_contractor:
+            lbl = unicode(self.general_contractor)
+            if lbl:
+                self.raw_general_contractor = lbl
+                self.save()
         self.update_delay_date()
         self.update_short_menu_class()
         return returned