sale_order.py

# -*- coding: utf-8 -*-
#
#
# Copyright Camptocamp SA
#
#
from osv import osv, fields
from openerp.tools.translate import _


class SaleOrder(osv.osv):

    """Add text condition"""

    _inherit = "sale.order"
    _columns = {
        'text_condition1': fields.many2one('account.condition_text', 'Header'),
        'text_condition2': fields.many2one('account.condition_text', 'Footer'),
        'note1': fields.text('Header'),
        'note2': fields.text('Footer')}

    def set_condition(
        self, cursor, uid, inv_id, cond_id, field_name, partner_id
    ):
        cond_obj = self.pool.get('account.condition_text')
        return cond_obj.get_value(cursor, uid, cond_id, field_name, partner_id)

    def action_invoice_create(
        self, cursor, user, order_id, grouped=False,
        states=['confirmed', 'done', 'exception'], date_inv=False, context=None
    ):
        # function is design to return only one id
        invoice_obj = self.pool.get('account.invoice')
        inv_id = super(SaleOrder, self).action_invoice_create(
            cursor, user, order_id, grouped, states, date_inv, context)

        invoice = invoice_obj.browse(cursor, user, inv_id)
        if isinstance(order_id, list):
            if len(order_id) > 1:
                raise osv.except_osv(
                    _('action_invoice_create can only receive one id'),
                    _('action_invoice_create can only receive one id'))

            order_id = order_id[0]
        order = self.browse(cursor, user, order_id)
        inv_data = {'text_condition1': order.text_condition1.id,
                    'text_condition2': order.text_condition2.id,
                    'note1': order.note1,
                    'note2': order.note2}
        invoice.write(inv_data, context=context)
        return inv_id

SaleOrder()