From 1c82fcfa3081909e29f725b98ed5a72829e0ecdb Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Fri, 10 Mar 2023 18:34:09 +0100
Subject: [PATCH] add customer details for default if there are more than one
 incoming lot

---
 ereuse_devicehub/inventory/views.py         | 22 +++++++++++++++++++--
 ereuse_devicehub/resources/device/models.py | 12 +++++++++++
 2 files changed, 32 insertions(+), 2 deletions(-)

diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py
index 89d0bdac..57bab6d7 100644
--- a/ereuse_devicehub/inventory/views.py
+++ b/ereuse_devicehub/inventory/views.py
@@ -1058,9 +1058,10 @@ class ExportsView(View):
                     erasures.append(device.privacy)
         return erasures
 
-    def get_costum_details(self):
+    def get_costum_details(self, erasures):
         my_data = None
         customer_details = None
+        lot = None
         if hasattr(g.user, 'sanitization_entity'):
             my_data = g.user.sanitization_entity
 
@@ -1071,6 +1072,23 @@ class ExportsView(View):
                 customer_details = lot.transfer.customer_details
         except Exception:
             pass
+
+        if lot or not erasures:
+            return my_data, customer_details
+
+        init = erasures[0].device.get_set_lots()
+        for e in erasures:
+            init = init.intersection(e.device.get_set_lots())
+
+        if len(init) != 1:
+            return my_data, customer_details
+
+        lot = init.pop()
+        try:
+            customer_details = lot.transfer.customer_details
+        except Exception:
+            pass
+
         return my_data, customer_details
 
     def get_server_erasure_hosts(self, erasures):
@@ -1093,7 +1111,7 @@ class ExportsView(View):
                 erasures[0].snapshot.version,
             )
 
-        my_data, customer_details = self.get_costum_details()
+        my_data, customer_details = self.get_costum_details(erasures)
 
         a, b = self.get_server_erasure_hosts(erasures)
         erasures_host, erasures_on_server = a, b
diff --git a/ereuse_devicehub/resources/device/models.py b/ereuse_devicehub/resources/device/models.py
index 168c4371..1e3d4b12 100644
--- a/ereuse_devicehub/resources/device/models.py
+++ b/ereuse_devicehub/resources/device/models.py
@@ -1021,6 +1021,18 @@ class Device(Thing):
 
         return
 
+    def get_set_lots(self):
+        if self.lots:
+            return set(self.lots)
+
+        if hasattr(self, "parent") and self.parent and self.parent.lots:
+            return set(self.parent.lots)
+
+        if self.binding:
+            return self.binding.device.get_set_lots()
+
+        return set()
+
     def __lt__(self, other):
         return self.id < other.id