diff --git a/website_product_retail/models/product_template.py b/website_product_retail/models/product_template.py index 00dbf98..e8b1944 100644 --- a/website_product_retail/models/product_template.py +++ b/website_product_retail/models/product_template.py @@ -7,19 +7,19 @@ class ProductTemplate(models.Model): @api.model def create(self, vals): vals = self.set_distribution_fields(vals) - return super(ProductTemplate, self).create(vals) + return super(ProductTemplate, self).create(vals) def write(self, vals): vals = self.set_distribution_fields(vals) return super().write(vals) def set_distribution_fields(self, vals): - if 'in_distribution' in vals: - if self.env.user.company_id.web_cat_in_distribution_libros: - category_id = self.env.user.company_id.web_cat_in_distribution_libros.id - elif 'in_distribution_music' in vals: + if 'in_distribution_music' in vals: if self.env.user.company_id.web_cat_in_distribution_musica: category_id = self.env.user.company_id.web_cat_in_distribution_musica.id + elif 'in_distribution' in vals: + if self.env.user.company_id.web_cat_in_distribution_libros: + category_id = self.env.user.company_id.web_cat_in_distribution_libros.id else: return vals if vals.get('in_distribution', False) or vals.get('in_distribution_music', False): @@ -30,4 +30,4 @@ class ProductTemplate(models.Model): else: vals['public_categ_ids'] = [(3,category_id)] vals['website_ribbon_id'] = False - return vals \ No newline at end of file + return vals