Archaeological files: change planning service management

1 files changed, 13 insertions, 2 deletions

diff --git a/archaeological_files/models.py b/archaeological_files/models.py
index f79de4f98..a7b23564d 100644
--- a/archaeological_files/models.py
+++ b/archaeological_files/models.py
@@ -359,13 +359,23 @@ class File(BaseHistorizedItem, OwnPerms, ValueGetter, ShortMenuItem,
 
     def update_planning_service(self):
         if not self.responsible_town_planning_service or \
-           self.responsible_town_planning_service.attached_to \
-           == self.planning_service:
+           not self.responsible_town_planning_service.attached_to or \
+           self.planning_service:
             return False
         self.planning_service = \
             self.responsible_town_planning_service.attached_to
         return True
 
+    def update_resp_planning_service(self):
+        if not self.responsible_town_planning_service or \
+           self.responsible_town_planning_service.attached_to or \
+           not self.planning_service:
+            return False
+        self.responsible_town_planning_service.attached_to = \
+            self.planning_service
+        self.responsible_town_planning_service.save()
+        return True
+
     def update_raw_general_contractor(self):
         if (self.raw_general_contractor and not
            self.general_contractor) or \
@@ -395,6 +405,7 @@ class File(BaseHistorizedItem, OwnPerms, ValueGetter, ShortMenuItem,
             self.towns.add(self.main_town)
         updated = self.update_raw_town_planning_service()
         updated += self.update_planning_service()
+        self.update_resp_planning_service()
         updated += self.update_raw_general_contractor()
         updated += self.update_corpo_general_contractor()
         if updated: