diff options
Diffstat (limited to 'archaeological_files/models.py')
| -rw-r--r-- | archaeological_files/models.py | 15 |
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: |