From 5e461416226801e1aa8178717fd80b804733fd91 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Wed, 16 Feb 2022 14:03:01 +0100
Subject: [PATCH] fix save trade action

---
 ereuse_devicehub/inventory/forms.py | 45 +++++++++++++++++++++++++++--
 ereuse_devicehub/inventory/views.py |  4 +++
 2 files changed, 47 insertions(+), 2 deletions(-)

diff --git a/ereuse_devicehub/inventory/forms.py b/ereuse_devicehub/inventory/forms.py
index f5bbd9ff..0326ee1e 100644
--- a/ereuse_devicehub/inventory/forms.py
+++ b/ereuse_devicehub/inventory/forms.py
@@ -31,6 +31,7 @@ from ereuse_devicehub.resources.device.sync import Sync
 from ereuse_devicehub.resources.enums import Severity, SnapshotSoftware
 from ereuse_devicehub.resources.lot.models import Lot
 from ereuse_devicehub.resources.tag.model import Tag
+from ereuse_devicehub.resources.user.models import User
 from ereuse_devicehub.resources.user.exceptions import InsufficientPermission
 
 
@@ -500,6 +501,7 @@ class NewActionForm(FlaskForm):
         if not is_valid:
             return False
 
+        self._devices = OrderedSet()
         if self.devices.data:
             devices = set(self.devices.data.split(","))
             self._devices = OrderedSet(Device.query.filter(Device.id.in_(devices)).filter(
@@ -638,14 +640,17 @@ class TradeForm(NewActionForm):
         super().__init__(*args, **kwargs)
         self.supplier.render_kw['data-email'] = g.user.email
         self.receiver.render_kw['data-email'] = g.user.email
+        self._lot = Lot.query.filter(Lot.id==self.lot.data).one()
 
     def validate(self, extra_validators=None):
-        is_valid = super().validate(extra_validators)
+        # is_valid = super().validate(extra_validators)
+        # import pdb; pdb.set_trace()
+        is_valid = True
 
         if not self.confirm.data and not self.code.data:
             self.code.errors = ["If you don't want confirm, you need a code"]
             is_valid = False
-
+# 
         if self.confirm.data and not (self.receiver.data or self.supplier.data):
             errors = ["If you want confirm, you need a email"]
             if not self.receiver.data:
@@ -656,4 +661,40 @@ class TradeForm(NewActionForm):
 
             is_valid = False
 
+        if self.confirm.data and is_valid:
+            user_to = User.query.filter_by(email=self.receiver.data).all() or [g.user]
+            user_from = User.query.filter_by(email=self.supplier.data).all() or [g.user]
+            if user_to == user_from:
+                is_valid = False
+            else:
+                self.user_to = user_to[0]
+                self.user_from = user_from[0]
+
+        self._devices = OrderedSet()
+        if self.devices.data:
+            devices = set(self.devices.data.split(","))
+            self._devices = OrderedSet(Device.query.filter(Device.id.in_(devices)).filter(
+                Device.owner_id == g.user.id).all())
+
         return is_valid
+
+    def save(self):
+        # import pdb; pdb.set_trace()
+        Model = db.Model._decl_class_registry.data[self.type.data]()
+        self.instance = Model()
+        self.instance.devices = self._devices
+        self.instance.severity = Severity[self.severity.data]
+        self.instance.user_from = self.user_from
+        self.instance.user_to = self.user_to
+        self.instance.lot_id = self._lot.id
+        self.instance.code = self.code.data
+        self.instance.confirm = self.confirm.data
+        self.instance.date = self.date.data
+        self.instance.name = self.name.data
+        self.instance.description = self.description.data
+
+        db.session.add(self.instance)
+        db.session.commit()
+
+
+        return self.instance
diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py
index 1e072eb8..94e9e6ac 100644
--- a/ereuse_devicehub/inventory/views.py
+++ b/ereuse_devicehub/inventory/views.py
@@ -313,6 +313,7 @@ class NewActionView(View):
         self.form = self.form_class()
 
         if self.form.validate_on_submit():
+            instance = self.form.save()
             messages.success('Action "{}" created successfully!'.format(self.form.type.data))
 
             next_url = self.get_next_url()
@@ -335,6 +336,7 @@ class NewAllocateView(NewActionView, DeviceListMix):
         self.form = self.form_class()
 
         if self.form.validate_on_submit():
+            instance = self.form.save()
             messages.success('Action "{}" created successfully!'.format(self.form.type.data))
 
             next_url = self.get_next_url()
@@ -354,6 +356,7 @@ class NewDataWipeView(NewActionView, DeviceListMix):
         self.form = self.form_class()
 
         if self.form.validate_on_submit():
+            instance = self.form.save()
             messages.success('Action "{}" created successfully!'.format(self.form.type.data))
 
             next_url = self.get_next_url()
@@ -373,6 +376,7 @@ class NewTradeView(NewActionView, DeviceListMix):
         self.form = self.form_class()
 
         if self.form.validate_on_submit():
+            instance = self.form.save()
             messages.success('Action "{}" created successfully!'.format(self.form.type.data))
 
             next_url = self.get_next_url()