fixing bug when create a new lot

ereuse_devicehub/inventory/forms.py

@@ -45,27 +45,31 @@ class LotForm(FlaskForm):
 
     def __init__(self, *args, **kwargs):
         id = kwargs.pop('id', None)
-        self.lot = None
+        self.instance = None
         if id:
-            self.lot = Lot.query.filter(Lot.id == id).filter(
+            self.instance = Lot.query.filter(Lot.id == id).filter(
                 Lot.owner_id == g.user.id).one()
         super().__init__(*args, **kwargs)
-        if self.lot and not self.name.data:
+        if self.instance and not self.name.data:
-            self.name.data = self.lot.name
+            self.name.data = self.instance.name
 
     def save(self):
         name = self.name.data.strip()
-        if self.lot:
+        if self.instance:
-            if self.lot.name == name:
+            if self.instance.name == name:
-                return
+                return self.instance
-            self.lot.name = name
+            self.instance.name = name
         else:
-            self.lot = Lot(name=name)
+            self.instance = Lot(name=name)
 
-        db.session.add(self.lot)
+        db.session.add(self.instance)
         db.session.commit()
+        return self.instance
 
     def remove(self):
-        if self.lot and not self.lot.devices:
+        if self.instance and not self.instance.devices:
-            self.lot.delete()
+            self.instance.delete()
             db.session.commit()
+        return self.instance
+
+

ereuse_devicehub/inventory/views.py

@@ -73,8 +73,8 @@ class LotView(View):
             self.title = "Edit lot"
         form = LotForm(id=id)
         if form.validate_on_submit():
-            form.save()
+            lot = form.save()
-            next_url = url_for('inventory.devices.lotdevicelist', id=form.lot.id)
+            next_url = url_for('inventory.devices.lotdevicelist', id=lot.id)
             return flask.redirect(next_url)
 
         return flask.render_template(self.template_name, form=form, title=self.title)