obook/product_sequence/models/product_product.py

66 lines
2.6 KiB
Python

# Copyright 2004 Tiny SPRL
# Copyright 2016 Sodexis
# Copyright 2018 Eficent Business and IT Consulting Services S.L.
# (http://www.eficent.com)
# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
from odoo import _, api, fields, models
class ProductProduct(models.Model):
_inherit = 'product.product'
default_code = fields.Char(
required=True,
default='/',
track_visibility='onchange',
help="Set to '/' and save if you want a new internal reference "
"to be proposed."
)
@api.model
def create(self, vals):
if 'default_code' not in vals or vals['default_code'] == '/':
categ_id = vals.get("categ_id")
template_id = vals.get("product_tmpl_id")
category = self.env['product.category']
if categ_id:
# Created as a product.product
category = category.browse(categ_id)
elif template_id:
# Created from a product.template
template = self.env["product.template"].browse(template_id)
category = template.categ_id
sequence = self.env["ir.sequence"].get_category_sequence_id(category)
vals['default_code'] = sequence.next_by_id()
return super().create(vals)
@api.multi
def write(self, vals):
"""To assign a new internal reference, just write '/' on the field.
Note this is up to the user, if the product category is changed,
she/he will need to write '/' on the internal reference to force the
re-assignment."""
if vals.get('default_code', '') == '/':
product_category_obj = self.env['product.category']
for product in self:
category_id = vals.get('categ_id', product.categ_id.id)
category = product_category_obj.browse(category_id)
sequence = self.env["ir.sequence"].get_category_sequence_id(category)
ref = sequence.next_by_id()
vals['default_code'] = ref
if len(product.product_tmpl_id.product_variant_ids) == 1:
product.product_tmpl_id.write({'default_code': ref})
super(ProductProduct, product).write(vals)
return True
return super().write(vals)
@api.multi
def copy(self, default=None):
if default is None:
default = {}
if self.default_code and 'default_code' not in default:
default.update({
'default_code': self.default_code + _('-copy'),
})
return super().copy(default)