95 lines
4.1 KiB
Python
95 lines
4.1 KiB
Python
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
||
|
|
|
||
|
|
from odoo import _, api, fields, models
|
||
|
|
from odoo.exceptions import UserError
|
||
|
|
|
||
|
|
|
||
|
|
class ChooseDeliveryCarrier(models.TransientModel):
|
||
|
|
_name = 'choose.delivery.carrier'
|
||
|
|
_description = 'Delivery Carrier Selection Wizard'
|
||
|
|
|
||
|
|
def _get_default_weight_uom(self):
|
||
|
|
return self.env['product.template']._get_weight_uom_name_from_ir_config_parameter()
|
||
|
|
|
||
|
|
order_id = fields.Many2one('sale.order', required=True, ondelete="cascade")
|
||
|
|
partner_id = fields.Many2one('res.partner', related='order_id.partner_id', required=True)
|
||
|
|
carrier_id = fields.Many2one(
|
||
|
|
'delivery.carrier',
|
||
|
|
string="Shipping Method",
|
||
|
|
required=True,
|
||
|
|
)
|
||
|
|
delivery_type = fields.Selection(related='carrier_id.delivery_type')
|
||
|
|
delivery_price = fields.Float()
|
||
|
|
display_price = fields.Float(string='Cost', readonly=True)
|
||
|
|
currency_id = fields.Many2one('res.currency', related='order_id.currency_id')
|
||
|
|
company_id = fields.Many2one('res.company', related='order_id.company_id')
|
||
|
|
available_carrier_ids = fields.Many2many("delivery.carrier", compute='_compute_available_carrier', string="Available Carriers")
|
||
|
|
invoicing_message = fields.Text(compute='_compute_invoicing_message')
|
||
|
|
delivery_message = fields.Text(readonly=True)
|
||
|
|
total_weight = fields.Float(string='Total Order Weight', related='order_id.shipping_weight', readonly=False)
|
||
|
|
weight_uom_name = fields.Char(readonly=True, default=_get_default_weight_uom)
|
||
|
|
|
||
|
|
@api.onchange('carrier_id', 'total_weight')
|
||
|
|
def _onchange_carrier_id(self):
|
||
|
|
self.delivery_message = False
|
||
|
|
if self.delivery_type in ('fixed', 'base_on_rule'):
|
||
|
|
vals = self._get_shipment_rate()
|
||
|
|
if vals.get('error_message'):
|
||
|
|
return {'error': vals['error_message']}
|
||
|
|
else:
|
||
|
|
self.display_price = 0
|
||
|
|
self.delivery_price = 0
|
||
|
|
|
||
|
|
@api.onchange('order_id')
|
||
|
|
def _onchange_order_id(self):
|
||
|
|
# fixed and base_on_rule delivery price will computed on each carrier change so no need to recompute here
|
||
|
|
if self.carrier_id and self.order_id.delivery_set and self.delivery_type not in ('fixed', 'base_on_rule'):
|
||
|
|
vals = self._get_shipment_rate()
|
||
|
|
if vals.get('error_message'):
|
||
|
|
warning = {
|
||
|
|
'title': _("%(carrier)s Error", carrier=self.carrier_id.name),
|
||
|
|
'message': vals['error_message'],
|
||
|
|
'type': 'notification',
|
||
|
|
}
|
||
|
|
return {'warning': warning}
|
||
|
|
|
||
|
|
@api.depends('carrier_id')
|
||
|
|
def _compute_invoicing_message(self):
|
||
|
|
self.ensure_one()
|
||
|
|
self.invoicing_message = ""
|
||
|
|
|
||
|
|
@api.depends('partner_id')
|
||
|
|
def _compute_available_carrier(self):
|
||
|
|
for rec in self:
|
||
|
|
carriers = self.env['delivery.carrier'].search(self.env['delivery.carrier']._check_company_domain(rec.order_id.company_id))
|
||
|
|
rec.available_carrier_ids = carriers.available_carriers(rec.order_id.partner_shipping_id) if rec.partner_id else carriers
|
||
|
|
|
||
|
|
def _get_shipment_rate(self):
|
||
|
|
vals = self.carrier_id.with_context(order_weight=self.total_weight).rate_shipment(self.order_id)
|
||
|
|
if vals.get('success'):
|
||
|
|
self.delivery_message = vals.get('warning_message', False)
|
||
|
|
self.delivery_price = vals['price']
|
||
|
|
self.display_price = vals['carrier_price']
|
||
|
|
return {}
|
||
|
|
return {'error_message': vals['error_message']}
|
||
|
|
|
||
|
|
def update_price(self):
|
||
|
|
vals = self._get_shipment_rate()
|
||
|
|
if vals.get('error_message'):
|
||
|
|
raise UserError(vals.get('error_message'))
|
||
|
|
return {
|
||
|
|
'name': _('Add a shipping method'),
|
||
|
|
'type': 'ir.actions.act_window',
|
||
|
|
'view_mode': 'form',
|
||
|
|
'res_model': 'choose.delivery.carrier',
|
||
|
|
'res_id': self.id,
|
||
|
|
'target': 'new',
|
||
|
|
}
|
||
|
|
|
||
|
|
def button_confirm(self):
|
||
|
|
self.order_id.set_delivery_line(self.carrier_id, self.delivery_price)
|
||
|
|
self.order_id.write({
|
||
|
|
'recompute_delivery_price': False,
|
||
|
|
'delivery_message': self.delivery_message,
|
||
|
|
})
|