diff --git a/pos_remove_pos_category/product.py b/pos_remove_pos_category/product.py index 99993081..7cee84f1 100644 --- a/pos_remove_pos_category/product.py +++ b/pos_remove_pos_category/product.py @@ -58,11 +58,15 @@ class ProductCategory(models.Model): @api.multi def _get_image(self): - return dict((rec.id, tools.image_get_resized_images(rec.image)) for rec in self) + return dict( + (rec.id, tools.image_get_resized_images(rec.image)) for rec in + self) @api.one def _set_image(self): - return self.write({'image': tools.image_resize_image_big(self.image_medium)}) + return self.write( + {'image': tools.image_resize_image_big(self.image_medium)}) + _auto_end_original = models.BaseModel._auto_end @@ -99,4 +103,5 @@ def _auto_end(self, cr, context=None): self._foreign_keys = foreign_keys return _auto_end_original(self, cr, context=context) + models.BaseModel._auto_end = _auto_end