From 4dd47534de59d7c6162a92a308dfa765ede9e6ea Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Fri, 3 May 2024 12:28:46 +0200 Subject: ✨ item deletion: better display of associated deleted items MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- archaeological_operations/models.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) (limited to 'archaeological_operations/models.py') diff --git a/archaeological_operations/models.py b/archaeological_operations/models.py index eeee9e8aa..83e60866d 100644 --- a/archaeological_operations/models.py +++ b/archaeological_operations/models.py @@ -1695,6 +1695,28 @@ class Operation( ) return dct + def get_deleted_data(self) -> dict: + """ + Return sub object list that will be deleted + :return: {"Sub object type": ["Sub object 1", "Sub object 2", ...]} + """ + data = {} + if self.administrative_act.count(): + lbl = str(_("Administrative act")) + data = {lbl: []} + for item in self.administrative_act.all(): + data[lbl].append(str(item)) + if self.context_record.count(): + lbl = str(_("Context record")) + data[lbl] = [] + for item in self.context_record.all(): + data[lbl].append(str(item)) + for key, value in item.get_deleted_data().items(): + if key not in data: + data[key] = [] + data[key] += value + return data + def archaeological_sites_list(self) -> list: return self.get_associated_main_item_list( "archaeological_sites", ArchaeologicalSite -- cgit v1.2.3