From 2c2480a1410c3445572936a70159efda5cc854f4 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Wed, 19 Aug 2015 15:12:43 +0200 Subject: Archaeological files: change planning service management --- archaeological_files/models.py | 15 +++++++++++++-- 1 file changed, 13 insertions(+), 2 deletions(-) (limited to 'archaeological_files/models.py') 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: -- cgit v1.2.3