stock-rma/rma_repair/wizards/rma_order_line_make_repair.py

# Copyright 2017 Eficent Business and IT Consulting Services S.L.
# License LGPL-3.0 or later (http://www.gnu.org/licenses/lgpl-3.0).

import odoo.addons.decimal_precision as dp
from odoo import _, api, fields, models
from odoo.exceptions import ValidationError


class RmaLineMakeRepair(models.TransientModel):
    _name = "rma.order.line.make.repair"
    _description = "Make Repair Order from RMA Line"

    item_ids = fields.One2many(
        comodel_name='rma.order.line.make.repair.item',
        inverse_name='wiz_id', string='Items')

    @api.model
    def _prepare_item(self, line):
        if line.product_id.refurbish_product_id:
            to_refurbish = True
            refurbish_product_id = line.product_id.refurbish_product_id.id
        else:
            to_refurbish = refurbish_product_id = False
        return {
            'line_id': line.id,
            'product_id': line.product_id.id,
            'product_qty': line.qty_to_repair,
            'rma_id': line.rma_id.id,
            'out_route_id': line.out_route_id.id,
            'product_uom_id': line.uom_id.id,
            'partner_id': line.partner_id.id,
            'to_refurbish': to_refurbish,
            'refurbish_product_id': refurbish_product_id,
            'location_id': line.location_id.id,
            'location_dest_id': line.location_id.id,
            'invoice_method': 'after_repair',
        }

    @api.model
    def default_get(self, fields):
        res = super(RmaLineMakeRepair, self).default_get(
            fields)
        rma_line_obj = self.env['rma.order.line']
        rma_line_ids = self.env.context['active_ids'] or []
        active_model = self.env.context['active_model']

        if not rma_line_ids:
            return res
        assert active_model == 'rma.order.line', 'Bad context propagation'
        items = []
        lines = rma_line_obj.browse(rma_line_ids)
        for line in lines:
            items.append([0, 0, self._prepare_item(line)])
        res['item_ids'] = items
        return res

    @api.multi
    def make_repair_order(self):
        res = []
        repair_obj = self.env['repair.order']
        for item in self.item_ids:
            rma_line = item.line_id
            data = item._prepare_repair_order(rma_line)
            repair = repair_obj.create(data)
            res.append(repair.id)
        return {
            'domain': [('id', 'in', res)],
            'name': _('Repairs'),
            'view_type': 'form',
            'view_mode': 'tree,form',
            'res_model': 'repair.order',
            'view_id': False,
            'context': False,
            'type': 'ir.actions.act_window'
        }


class RmaLineMakeRepairItem(models.TransientModel):
    _name = "rma.order.line.make.repair.item"
    _description = "RMA Line Make Repair Item"

    @api.constrains('product_qty')
    def _check_prodcut_qty(self):
        for rec in self:
            if rec.product_qty <= 0.0:
                raise ValidationError(_('Quantity must be positive.'))

    @api.onchange('to_refurbish')
    def _onchange_to_refurbish(self):
        if self.to_refurbish:
            self.refurbish_product_id = self.product_id.refurbish_product_id
        else:
            self.refurbish_product_id = False

    wiz_id = fields.Many2one(
        comodel_name='rma.order.line.make.repair', string='Wizard',
        ondelete='cascade',
    )
    line_id = fields.Many2one(
        comodel_name='rma.order.line', string='RMA',
        required=True,
    )
    rma_id = fields.Many2one(
        comodel_name='rma.order', related='line_id.rma_id',
        string='RMA Order',
    )
    product_id = fields.Many2one(
        comodel_name='product.product', string='Product', readonly=True,
    )
    product_qty = fields.Float(
        string='Quantity to repair', digits=dp.get_precision('Product UoS'),
    )
    product_uom_id = fields.Many2one(
        comodel_name='uom.uom', string='UoM', readonly=True,
    )
    out_route_id = fields.Many2one(
        comodel_name='stock.location.route', string='Outbound Route',
        domain=[('rma_selectable', '=', True)],
    )
    partner_id = fields.Many2one(
        comodel_name='res.partner', string='Customer', required=False,
        domain=[('customer', '=', True)], readonly=True,
    )
    location_id = fields.Many2one(
        comodel_name="stock.location", string="Location", required=True,
    )
    location_dest_id = fields.Many2one(
        comodel_name="stock.location", string="Destination location",
        required=True,
    )
    to_refurbish = fields.Boolean(string="To Refurbish?")
    refurbish_product_id = fields.Many2one(
        comodel_name="product.product", string="Refurbished Product",
    )
    invoice_method = fields.Selection(
        string="Invoice Method", selection=[
            ("none", "No Invoice"),
            ("b4repair", "Before Repair"),
            ("after_repair", "After Repair")],
        required=True,
        help="Selecting 'Before Repair' or 'After Repair' will allow you "
             "to generate invoice before or after the repair is done "
             "respectively. 'No invoice' means you don't want to generate "
             "invoice for this repair order.",
    )

    @api.model
    def _prepare_repair_order(self, rma_line):
        location_dest = (self.location_dest_id if not self.to_refurbish else
                         rma_line.product_id.property_stock_refurbish)
        refurbish_location_dest_id = (self.location_dest_id.id if
                                      self.to_refurbish else False)
        return {
            'product_id': rma_line.product_id.id,
            'partner_id': rma_line.partner_id.id,
            'product_qty': self.product_qty,
            'rma_line_id': rma_line.id,
            'product_uom': rma_line.product_id.uom_po_id.id,
            'company_id': rma_line.company_id.id,
            'location_id': self.location_id.id,
            'location_dest_id': location_dest.id,
            'refurbish_location_dest_id': refurbish_location_dest_id,
            'refurbish_product_id': self.refurbish_product_id.id,
            'to_refurbish': self.to_refurbish,
            'invoice_method': self.invoice_method,
            'partner_invoice_id': rma_line.invoice_address_id.id,
        }