From 49062d99440cfc5baf6e6f297258173c2a297cf3 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Mon, 20 Jun 2022 10:40:00 +0200
Subject: [PATCH] add export devices lots

---
 ereuse_devicehub/inventory/views.py      | 48 +++++++++++++++++++-----
 ereuse_devicehub/resources/lot/models.py | 10 +++++
 2 files changed, 49 insertions(+), 9 deletions(-)

diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py
index 646f5a63..a1ed20fa 100644
--- a/ereuse_devicehub/inventory/views.py
+++ b/ereuse_devicehub/inventory/views.py
@@ -478,6 +478,7 @@ class ExportsView(View):
             'devices': self.devices_list,
             'certificates': self.erasure,
             'lots': self.lots_export,
+            'devices_lots': self.devices_lots_export,
         }
 
         if export_id not in export_ids:
@@ -613,14 +614,6 @@ class ExportsView(View):
             receiver_note = lot.transfer and lot.transfer.receiver_note or ''
             wb_devs = 0
             placeholders = 0
-            type_transfer = ''
-            if lot.transfer:
-                if lot.transfer.user_from == g.user:
-                    type_transfer = 'Outgoing'
-                if lot.transfer.user_to == g.user:
-                    type_transfer = 'Incoming'
-            else:
-                type_transfer = 'Temporary'
 
             for dev in lot.devices:
                 snapshots = [e for e in dev.actions if e.type == 'Snapshot']
@@ -634,7 +627,7 @@ class ExportsView(View):
             row = [
                 lot.id,
                 lot.name,
-                type_transfer,
+                lot.type_transfer(),
                 lot.transfer and (lot.transfer.closed and 'Closed' or 'Open') or '',
                 lot.transfer and lot.transfer.code or '',
                 lot.transfer and lot.transfer.date or '',
@@ -657,6 +650,43 @@ class ExportsView(View):
 
         return self.response_csv(data, "lots_export.csv")
 
+    def devices_lots_export(self):
+        data = StringIO()
+        cw = csv.writer(data, delimiter=';', lineterminator="\n", quotechar='"')
+        head = [
+            'DHID',
+            'Lot Id',
+            'Lot Name',
+            'Lot Type',
+            'Transfer Status',
+            'Transfer Code',
+            'Transfer Date',
+            'Transfer Creation Date',
+            'Transfer Update Date'
+        ]
+        cw.writerow(head)
+
+        for dev in self.find_devices():
+            for lot in dev.lots:
+                type_transfer = lot.type_transfer()
+                if type_transfer in ['Temporary', '']:
+                    continue
+
+                row = [
+                    dev.devicehub_id,
+                    lot.id,
+                    lot.name,
+                    type_transfer,
+                    lot.transfer and (lot.transfer.closed and 'Closed' or 'Open') or '',
+                    lot.transfer and lot.transfer.code or '',
+                    lot.transfer and lot.transfer.date or '',
+                    lot.transfer and lot.transfer.created or '',
+                    lot.transfer and lot.transfer.updated or '',
+                ]
+                cw.writerow(row)
+
+        return self.response_csv(data, "Devices_Incoming_and_Outgoing_Lots_Spreadsheet.csv")
+
 
 class SnapshotListView(GenericMixin):
     template_name = 'inventory/snapshots_list.html'
diff --git a/ereuse_devicehub/resources/lot/models.py b/ereuse_devicehub/resources/lot/models.py
index ef6577a8..c2309224 100644
--- a/ereuse_devicehub/resources/lot/models.py
+++ b/ereuse_devicehub/resources/lot/models.py
@@ -151,6 +151,16 @@ class Lot(Thing):
         """Gets the lots that are not under any other lot."""
         return cls.query.join(cls.paths).filter(db.func.nlevel(Path.path) == 1)
 
+    def type_transfer(self):
+        # Used in reports lots_export.csv
+        if not self.transfer:
+            return 'Temporary'
+        if self.transfer.user_from == g.user:
+            return 'Outgoing'
+        if self.transfer.user_to == g.user:
+            return 'Incoming'
+        return ''
+
     def add_children(self, *children):
         """Add children lots to this lot.