flectra/addons/website_sale_comparison/models/website_sale_comparison.py
2018-01-16 02:34:37 -08:00

32 lines
1011 B
Python

# -*- coding: utf-8 -*-
from collections import OrderedDict
from flectra import fields, models, _
class ProductAttributeCategory(models.Model):
_name = "product.attribute.category"
_description = "Product Attribute Category"
_order = 'sequence'
name = fields.Char("Category Name", required=True)
sequence = fields.Integer("Sequence", default=10)
class ProductAttribute(models.Model):
_inherit = 'product.attribute'
category_id = fields.Many2one('product.attribute.category', string="Category",
help="Set a category to regroup similar attributes under "
"the same section in the Comparison page of eCommerce")
class ProductTemplate(models.Model):
_inherit = 'product.template'
def get_variant_groups(self):
res = OrderedDict()
for var in self.attribute_line_ids:
res.setdefault(var.attribute_id.category_id.name or _('Uncategorized'), []).append(var)
return res