From 07aa770aaa816f02cb2108c34e5225a9eb25b797 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Sun, 10 Jan 2016 19:47:44 +0100 Subject: Fix automatic association of general contractor corporation --- archaeological_files/models.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) (limited to 'archaeological_files/models.py') diff --git a/archaeological_files/models.py b/archaeological_files/models.py index d64d20a02..b21415d16 100644 --- a/archaeological_files/models.py +++ b/archaeological_files/models.py @@ -475,8 +475,13 @@ class File(BaseHistorizedItem, OwnPerms, ValueGetter, ShortMenuItem, self.general_contractor.attached_to \ == self.corporation_general_contractor: return False - self.corporation_general_contractor = \ - self.general_contractor.attached_to + if self.general_contractor.attached_to: + self.corporation_general_contractor = \ + self.general_contractor.attached_to + else: + self.general_contractor.attached_to = \ + self.corporation_general_contractor + self.general_contractor.save() return True def save(self, *args, **kwargs): -- cgit v1.2.3