stock_picking.py

# -*- coding: utf-8 -*-
##############################################################################
#
#    Copyright (C) 2013 Agile Business Group sagl (<http://www.agilebg.com>)
#    Author: Nicola Malcontenti <nicola.malcontenti@agilebg.com>
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as published
#    by the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
##############################################################################
from openerp.osv import fields, orm


class stock_move(orm.Model):
    _inherit = 'stock.move'

    def _get_product_customer_code(
            self, cr, uid, ids, name, args, context=None
    ):
        res = {}
        product_customer_code_obj = self.pool['product.customer.code']
        for move in self.browse(cr, uid, ids, context=context):
            res[move.id] = ''
            main_partner_id = (
                move.picking_id and
                move.picking_id.partner_id and
                move.picking_id.partner_id.commercial_partner_id and
                move.picking_id.partner_id.commercial_partner_id.id or
                False)
            product = move.product_id
            if product and main_partner_id:
                code_ids = product_customer_code_obj.search(
                    cr, uid, [
                        ('product_id', '=', product.id),
                        ('partner_id', '=', main_partner_id),
                    ], limit=1, context=context)
                if code_ids:
                    data = product_customer_code_obj.read(
                        cr, uid, code_ids[0],
                        ['product_code'], context=context
                    )
                    res[move.id] = data.get('product_code', '')
        return res

    _columns = {
        'product_customer_code': fields.function(
            _get_product_customer_code,
            string='Product Customer Code', type='char', size=64),
    }