Files
stock-rma/rma_purchase/wizards/rma_add_purchase.py
Jordi Ballester 20ecbee99c init branch
2020-12-24 13:34:58 +05:30

115 lines
4.7 KiB
Python

# -*- coding: utf-8 -*-
# © 2017 Eficent Business and IT Consulting Services S.L.
# License LGPL-3.0 or later (https://www.gnu.org/licenses/lgpl.html)
import time
from openerp import models, fields, exceptions, api, _
from openerp.exceptions import ValidationError
from openerp.tools import DEFAULT_SERVER_DATETIME_FORMAT as DT_FORMAT
import openerp.addons.decimal_precision as dp
class RmaAddPurchase(models.TransientModel):
_name = 'rma_add_purchase'
_description = 'Wizard to add rma lines'
@api.model
def default_get(self, fields):
res = super(RmaAddPurchase, self).default_get(fields)
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['purchase_id'] = False
res['purchase_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)
purchase_id = fields.Many2one(comodel_name='purchase.order', string='Order')
purchase_line_ids = fields.Many2many('purchase.order.line',
'rma_add_purchase_add_line_rel',
'purchase_line_id', 'rma_add_purchase_id',
readonly=False,
string='Purcahse Order Lines')
def _prepare_rma_line_from_po_line(self, line):
operation = line.product_id.rma_operation_id and \
line.product_id.rma_operation_id.id or False
if not operation:
operation = line.product_id.categ_id.rma_operation_id and \
line.product_id.categ_id.rma_operation_id.id or False
data = {
'purchase_order_line_id': line.id,
'product_id': line.product_id.id,
'origin': line.order_id.name,
'uom_id': line.product_uom.id,
'operation_id': operation,
'product_qty': line.product_qty,
'price_unit': line.currency_id.compute(
line.price_unit, line.currency_id, round=False),
'rma_id': self.rma_id.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")
data.update(
{'in_route_id': operation.in_route_id.id or route,
'out_route_id': operation.out_route_id.id or route,
'receipt_policy': operation.receipt_policy,
'location_id': operation.location_id.id or
self.env.ref('stock.stock_location_stock').id,
'operation_id': operation.id,
'refund_policy': operation.refund_policy,
'delivery_policy': operation.delivery_policy
})
return data
@api.model
def _get_rma_data(self):
data = {
'date_rma': fields.Datetime.now(),
'delivery_address_id': self.purchase_id.partner_id.id,
'invoice_address_id': self.purchase_id.partner_id.id
}
return data
@api.model
def _get_existing_purchase_lines(self):
existing_purchase_lines = []
for rma_line in self.rma_id.rma_line_ids:
existing_purchase_lines.append(rma_line.purchase_order_line_id)
return existing_purchase_lines
@api.multi
def add_lines(self):
rma_line_obj = self.env['rma.order.line']
existing_purchase_lines = self._get_existing_purchase_lines()
for line in self.purchase_line_ids:
# Load a PO line only once
if line not in existing_purchase_lines:
data = self._prepare_rma_line_from_po_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'}