product_pricelist_prices/models/product_pricelist.py

# -*- coding: utf-8 -*-
# Copyright© 2016 ICTSTUDIO <http://www.ictstudio.eu>
# License: AGPL-3.0 or later (http://www.gnu.org/licenses/agpl)

import logging

from openerp import models, fields, api, _

_logger = logging.getLogger(__name__)

class product_pricelist(models.Model):
    _inherit = 'product.pricelist'
    
    show_on_products = fields.Boolean(string="Display on product page")
    product_price = fields.Float(
            compute="_get_product_price",
            inverse="_set_product_price",
            string="Price"
    )
    product_price_manual = fields.Boolean(
            compute="_get_price_manual",
            # inverse="_set_price_manual",
            string="Manual Price",

    )
    product_id = fields.Integer(
            #comodel_name="product.product",
            compute="_get_product_price",
            string="product_id"
    )

    @api.model
    def _get_product_id(self):
        product_id = False
        if self.env.context.get('product_template_id', False):
            product_tmpl = self.env['product.template'].browse(
                    [self.env.context.get('product_template_id')]
            )
            product_id = False
            if product_tmpl:
                product_tmpl_id = product_tmpl[0]
                if product_tmpl_id and product_tmpl_id.product_variant_ids:
                    product = product_tmpl_id.product_variant_ids[0]
                    if product:
                        product_id = product.id
        elif self.env.context.get('product_id', False):
            product_id = self.env.context.get('product_id')

        return product_id

    @api.one
    def _get_price_manual(self):
        product_id = self._get_product_id()

        version = self.get_active_pricelist_version()
        if version:
            if self._context.get('product_id') and \
                    self.env['product.pricelist.item'].search(
                            [
                                ('price_version_id', '=', version.id),
                                ('product_id', '=', self._context.get('product_id'))
                            ]
                    ):
                self.product_price_manual = True
            if self._context.get('product_template_id') and \
                    self.env['product.pricelist.item'].search(
                            [
                                ('price_version_id', '=', version.id),
                                ('product_tmpl_id', '=', self._context.get('product_template_id'))
                            ]
                    ):
                self.product_price_manual = True

    @api.one
    def _get_product_price(self):
        product_id = self._get_product_id()
        if product_id:
            self.product_price = self.price_get(
                    product_id, 1).get(self.id, 0.0
            )
            self.product_id = product_id

    def _set_product_price(self):
        # Real change takes place in price_set after inverse
        # method of pricelists object on product_template
        _logger.debug("Set Price: %s", self.product_price)



    @api.one
    def remove_price_manual(self):
        # Real change takes place in price_set after inverse
        # method of pricelists object on product_template
        if self._context.get('product_template_id'):
            self.price_remove(self._context.get('product_template_id'))
        if self._context.get('product_id'):
            product = self.env['product.product'].browse(self._context.get('product_id'))
            self.price_remove(product.product_tmpl_id.id)


    def price_set(self, product_template, new_price):
        """
        Set the price of the product in current pricelist
        if different from price through  pricerules
        :param product_template: product_template object
        :param new_price: new price
        :return:
        """
        if new_price:

            version = self.get_active_pricelist_version()
            if not version:
                return False
            items = self.env['product.pricelist.item'].search(
                    [
                        ('product_tmpl_id','=', product_template.id),
                        ('price_version_id', '=', version.id)
                    ]
            )
            product_price_type_ids = self.env['product.price.type'].search(
                    [
                        ('field', '=', 'list_price')
                    ]
            )
            if not items:
                self.env['product.pricelist.item'].create(
                        {
                            'base': product_price_type_ids and product_price_type_ids[0].id,
                            'sequence': 1,
                            'name': product_template.name,
                            'product_tmpl_id': product_template.id,
                            'price_version_id': version.id,
                            'price_surcharge': new_price,
                            'price_discount': -1
                        }
                )
            else:
                for item in items:
                    item.write(
                            {
                                'base': product_price_type_ids and product_price_type_ids[0].id,
                                'sequence': 1,
                                'name': product_template.name,
                                'product_tmpl_id': product_template.id,
                                'price_version_id': version.id,
                                'price_surcharge': new_price,
                                'price_discount': -1
                            }
                    )
            return True

    def price_remove(self, product_template_id):

        version = self.get_active_pricelist_version()
        if not version:
            return False

        items = self.env['product.pricelist.item'].search(
                    [
                        ('product_tmpl_id', '=', product_template_id),
                        ('price_version_id', '=', version.id)
                    ]
        )
        _logger.debug("Items: %s", items)
        for item in items:
            _logger.debug("Remove Item: %s", item)
            item.unlink()

        return True



    def get_active_pricelist_version(self):
        date = self.env.context.get('date') or fields.Date.context_today(self)

        version = False
        for v in self.version_id:
            if ((v.date_start is False) or (v.date_start <= date)) and ((v.date_end is False) or (v.date_end >= date)):
                version = v
                break

        return version