From d3dc5c898689046ad9b08b9cd64e8bea8d2fcc17 Mon Sep 17 00:00:00 2001
From: Cayo Puigdefabregas <cayo@ribaguifi.com>
Date: Thu, 30 Dec 2021 21:35:54 +0100
Subject: [PATCH] view for create an edit the a lot

---
 ereuse_devicehub/inventory/forms.py           | 35 ++++++++++++++++++
 ereuse_devicehub/inventory/views.py           | 37 +++++++++++++------
 .../templates/ereuse_devicehub/base_site.html |  2 +-
 3 files changed, 61 insertions(+), 13 deletions(-)

diff --git a/ereuse_devicehub/inventory/forms.py b/ereuse_devicehub/inventory/forms.py
index 6ababca8..6ee56861 100644
--- a/ereuse_devicehub/inventory/forms.py
+++ b/ereuse_devicehub/inventory/forms.py
@@ -1,7 +1,9 @@
 from flask_wtf import FlaskForm
 from wtforms import StringField, validators
 from flask_login import current_user
+from flask import g
 
+from ereuse_devicehub.db import db
 from ereuse_devicehub.resources.device.models import Device
 from ereuse_devicehub.resources.lot.models import Lot
 
@@ -27,3 +29,36 @@ class LotDeviceAddForm(FlaskForm):
             return False
 
         return True
+
+    def save(self):
+        self.lot.devices.update(self.devices)
+        g.user = current_user
+        db.session.add(self.lot)
+        db.session.commit()
+
+
+class LotForm(FlaskForm):
+    name = StringField(u'Name', [validators.length(min=1)])
+
+    def __init__(self, *args, **kwargs):
+        id = kwargs.pop('id', None)
+        self.lot = None
+        if id:
+            self.lot = Lot.query.filter(Lot.id == id).filter(
+                Lot.owner_id == current_user.id).one()
+        super().__init__(*args, **kwargs)
+        if self.lot and not self.name.data:
+            self.name.data = self.lot.name
+
+    def save(self):
+        name = self.name.data.strip()
+        if self.lot:
+            if self.lot.name == name:
+                return
+            self.lot.name = name
+        else:
+            self.lot = Lot(name=name)
+
+        g.user = current_user
+        db.session.add(self.lot)
+        db.session.commit()
diff --git a/ereuse_devicehub/inventory/views.py b/ereuse_devicehub/inventory/views.py
index ec2add39..e0b9d356 100644
--- a/ereuse_devicehub/inventory/views.py
+++ b/ereuse_devicehub/inventory/views.py
@@ -1,12 +1,11 @@
 import flask
 from flask.views import View
-from flask import Blueprint, url_for, g
+from flask import Blueprint, url_for
 from flask_login import login_required, current_user
 
-from ereuse_devicehub.db import db
 from ereuse_devicehub.resources.lot.models import Lot
 from ereuse_devicehub.resources.device.models import Device
-from ereuse_devicehub.inventory.forms import LotDeviceAddForm
+from ereuse_devicehub.inventory.forms import LotDeviceAddForm, LotForm
 
 devices = Blueprint('inventory.devices', __name__, url_prefix='/inventory')
 
@@ -34,7 +33,6 @@ class LotDeviceListView(View):
 
     def dispatch_request(self, id):
         # TODO @cayop adding filter
-        # import pdb; pdb.set_trace()
         lots = Lot.query.filter(Lot.owner_id == current_user.id)
         lot = lots.filter(Lot.id == id).one()
         filter_types = ['Desktop', 'Laptop', 'Server']
@@ -55,17 +53,32 @@ class LotDeviceAddView(View):
     def dispatch_request(self):
         form = LotDeviceAddForm()
         if form.validate_on_submit():
-            lot = form.lot
-            devices = form.devices
-            lot.devices.update(devices)
-            g.user = current_user
-            db.session.add(lot)
-            db.session().final_flush()
+            form.save()
 
-            next_url = url_for('inventory.devices.lotdevicelist', id=lot.id)
+            next_url = url_for('inventory.devices.lotdevicelist', id=form.lot.id)
             return flask.redirect(next_url)
 
 
+class LotView(View):
+    methods = ['GET', 'POST']
+    decorators = [login_required]
+    template_name = 'inventory/lot.html'
+    title = "Add a new lot"
+
+    def dispatch_request(self, id=None):
+        if id:
+            self.title = "Edit lot"
+        form = LotForm(id=id)
+        if form.validate_on_submit():
+            form.save()
+            next_url = url_for('inventory.devices.lotdevicelist', id=form.lot.id)
+            return flask.redirect(next_url)
+
+        return flask.render_template(self.template_name, form=form, title=self.title)
+
+
 devices.add_url_rule('/device/', view_func=DeviceListView.as_view('devicelist'))
 devices.add_url_rule('/lot/<string:id>/device/', view_func=LotDeviceListView.as_view('lotdevicelist'))
-devices.add_url_rule('/lot/devices/add', view_func=LotDeviceAddView.as_view('lot_devices_add'))
+devices.add_url_rule('/lot/devices/add/', view_func=LotDeviceAddView.as_view('lot_devices_add'))
+devices.add_url_rule('/lot/add', view_func=LotView.as_view('lot_add'))
+devices.add_url_rule('/lot/<string:id>/', view_func=LotView.as_view('lot_edit'))
diff --git a/ereuse_devicehub/templates/ereuse_devicehub/base_site.html b/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
index 9db77874..07a82e0c 100644
--- a/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
+++ b/ereuse_devicehub/templates/ereuse_devicehub/base_site.html
@@ -290,7 +290,7 @@
       </a>
       <ul id="temporal-lots-nav" class="nav-content collapse " data-bs-parent="#sidebar-nav">
         <li>
-          <a href="#TODO">
+          <a href="{{ url_for('inventory.devices.lot_add')}}">
             <i class="bi bi-plus" style="font-size: larger;"></i><span>New temporal lot</span>
           </a>
         </li>