Files
stock-rma/rma_account/wizards/rma_add_invoice.py

115 lines
4.9 KiB
Python

# © 2017 Eficent Business and IT Consulting Services S.L.
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html)
from odoo import _, api, fields, models
from odoo.exceptions import ValidationError
class RmaAddInvoice(models.TransientModel):
_name = 'rma_add_invoice'
_description = 'Wizard to add rma lines'
@api.model
def default_get(self, fields_list):
res = super(RmaAddInvoice, self).default_get(fields_list)
rma_obj = self.env['rma.order']
rma_id = self.env.context['active_ids'] or []
active_model = self.env.context['active_model']
if not rma_id:
return res
assert active_model == 'rma.order', 'Bad context propagation'
rma = rma_obj.browse(rma_id)
res['rma_id'] = rma.id
res['partner_id'] = rma.partner_id.id
res['invoice_line_ids'] = False
return res
rma_id = fields.Many2one('rma.order', string='RMA Order', readonly=True,
ondelete='cascade')
partner_id = fields.Many2one(comodel_name='res.partner', string='Partner',
readonly=True)
invoice_line_ids = fields.Many2many('account.invoice.line',
'rma_add_invoice_add_line_rel',
'invoice_line_id',
'rma_add_invoice_id',
string='Invoice Lines')
def _prepare_rma_line_from_inv_line(self, line):
if self.env.context.get('customer'):
operation = line.product_id.rma_customer_operation_id or \
line.product_id.categ_id.rma_customer_operation_id
else:
operation = line.product_id.rma_supplier_operation_id or \
line.product_id.categ_id.rma_supplier_operation_id
if not operation:
operation = self.env['rma.operation'].search(
[('type', '=', self.rma_id.type)], limit=1)
if not operation:
raise ValidationError(_("Please define an operation first"))
if not operation.in_route_id or not operation.out_route_id:
route = self.env['stock.location.route'].search(
[('rma_selectable', '=', True)], limit=1)
if not route:
raise ValidationError(_("Please define an rma route"))
if not operation.in_warehouse_id or not operation.out_warehouse_id:
warehouse = self.env['stock.warehouse'].search(
[('company_id', '=', self.rma_id.company_id.id),
('lot_rma_id', '!=', False)], limit=1)
if not warehouse:
raise ValidationError(_("Please define a warehouse with a"
" default rma location"))
data = {
'partner_id': self.partner_id.id,
'invoice_line_id': line.id,
'product_id': line.product_id.id,
'origin': line.invoice_id.number,
'uom_id': line.uom_id.id,
'operation_id': operation.id,
'product_qty': line.quantity,
'price_unit': line.invoice_id.currency_id.compute(
line.price_unit, line.currency_id, round=False),
'delivery_address_id': line.invoice_id.partner_id.id,
'invoice_address_id': line.invoice_id.partner_id.id,
'rma_id': self.rma_id.id,
'receipt_policy': operation.receipt_policy,
'refund_policy': operation.refund_policy,
'delivery_policy': operation.delivery_policy,
'in_warehouse_id': operation.in_warehouse_id.id or warehouse.id,
'out_warehouse_id': operation.out_warehouse_id.id or warehouse.id,
'in_route_id': operation.in_route_id.id or route.id,
'out_route_id': operation.out_route_id.id or route.id,
'location_id': (operation.location_id.id or
operation.in_warehouse_id.lot_rma_id.id or
warehouse.lot_rma_id.id),
}
return data
@api.model
def _get_rma_data(self):
data = {
'date_rma': fields.Datetime.now(),
}
return data
@api.model
def _get_existing_invoice_lines(self):
existing_invoice_lines = []
for rma_line in self.rma_id.rma_line_ids:
existing_invoice_lines.append(rma_line.invoice_line_id)
return existing_invoice_lines
@api.multi
def add_lines(self):
rma_line_obj = self.env['rma.order.line']
existing_invoice_lines = self._get_existing_invoice_lines()
for line in self.invoice_line_ids:
# Load a PO line only once
if line not in existing_invoice_lines:
data = self._prepare_rma_line_from_inv_line(line)
rma_line_obj.create(data)
rma = self.rma_id
data_rma = self._get_rma_data()
rma.write(data_rma)
return {'type': 'ir.actions.act_window_close'}