addons/website_sale_delivery/models/sale_order.py

   1 # -*- coding: utf-8 -*-
   2
   3 from openerp.osv import orm, fields
   4 from openerp import SUPERUSER_ID
   5 from openerp.addons import decimal_precision
   6
   7
   8 class delivery_carrier(orm.Model):
   9     _inherit = 'delivery.carrier'
  10     _columns = {
  11         'website_published': fields.boolean('Available in the website', copy=False),
  12         'website_description': fields.text('Description for the website'),
  13     }
  14     _defaults = {
  15         'website_published': True
  16     }
  17
  18
  19 class SaleOrder(orm.Model):
  20     _inherit = 'sale.order'
  21
  22     def _amount_all_wrapper(self, cr, uid, ids, field_name, arg, context=None):
  23         """ Wrapper because of direct method passing as parameter for function fields """
  24         return self._amount_all(cr, uid, ids, field_name, arg, context=context)
  25
  26     def _amount_all(self, cr, uid, ids, field_name, arg, context=None):
  27         res = super(SaleOrder, self)._amount_all(cr, uid, ids, field_name, arg, context=context)
  28         currency_pool = self.pool.get('res.currency')
  29         for order in self.browse(cr, uid, ids, context=context):
  30             line_amount = sum([line.price_subtotal for line in order.order_line if line.is_delivery])
  31             currency = order.pricelist_id.currency_id
  32             res[order.id]['amount_delivery'] = currency_pool.round(cr, uid, currency, line_amount)
  33         return res
  34
  35     def _get_order(self, cr, uid, ids, context=None):
  36         result = {}
  37         for line in self.pool.get('sale.order.line').browse(cr, uid, ids, context=context):
  38             result[line.order_id.id] = True
  39         return result.keys()
  40
  41     _columns = {
  42         'amount_delivery': fields.function(
  43             _amount_all_wrapper, type='float', digits_compute=decimal_precision.get_precision('Account'),
  44             string='Delivery Amount',
  45             store={
  46                 'sale.order': (lambda self, cr, uid, ids, c={}: ids, ['order_line'], 10),
  47                 'sale.order.line': (_get_order, ['price_unit', 'tax_id', 'discount', 'product_uom_qty'], 10),
  48             },
  49             multi='sums', help="The amount without tax.", track_visibility='always'
  50         ),
  51         'website_order_line': fields.one2many(
  52             'sale.order.line', 'order_id',
  53             string='Order Lines displayed on Website', readonly=True,
  54             domain=[('is_delivery', '=', False)],
  55             help='Order Lines to be displayed on the website. They should not be used for computation purpose.',
  56         ),
  57     }
  58
  59     def _check_carrier_quotation(self, cr, uid, order, force_carrier_id=None, context=None):
  60         carrier_obj = self.pool.get('delivery.carrier')
  61
  62         # check to add or remove carrier_id
  63         if not order:
  64             return False
  65         if all(line.product_id.type == "service" for line in order.website_order_line):
  66             order.write({'carrier_id': None}, context=context)
  67             self.pool['sale.order']._delivery_unset(cr, SUPERUSER_ID, [order.id], context=context)
  68             return True
  69         else:
  70             carrier_id = force_carrier_id or order.carrier_id.id
  71             carrier_ids = self._get_delivery_methods(cr, uid, order, context=context)
  72             if carrier_id:
  73                 if carrier_id not in carrier_ids:
  74                     carrier_id = False
  75                 else:
  76                     carrier_ids.remove(carrier_id)
  77                     carrier_ids.insert(0, carrier_id)
  78             if force_carrier_id or not carrier_id or not carrier_id in carrier_ids:
  79                 for delivery_id in carrier_ids:
  80                     grid_id = carrier_obj.grid_get(cr, SUPERUSER_ID, [delivery_id], order.partner_shipping_id.id)
  81                     if grid_id:
  82                         carrier_id = delivery_id
  83                         break
  84                 order.write({'carrier_id': carrier_id}, context=context)
  85             if carrier_id:
  86                 order.delivery_set(context=context)
  87             else:
  88                 order._delivery_unset(context=context)
  89
  90         return bool(carrier_id)
  91
  92     def _get_delivery_methods(self, cr, uid, order, context=None):
  93         carrier_obj = self.pool.get('delivery.carrier')
  94         delivery_ids = carrier_obj.search(cr, uid, [('website_published','=',True)], context=context)
  95         # Following loop is done to avoid displaying delivery methods who are not available for this order
  96         # This can surely be done in a more efficient way, but at the moment, it mimics the way it's
  97         # done in delivery_set method of sale.py, from delivery module
  98         for delivery_id in carrier_obj.browse(cr, SUPERUSER_ID, delivery_ids, context=dict(context, order_id=order.id)):
  99             if not delivery_id.available:
 100                 delivery_ids.remove(delivery_id.id)
 101         return delivery_ids
 102
 103     def _get_errors(self, cr, uid, order, context=None):
 104         errors = super(SaleOrder, self)._get_errors(cr, uid, order, context=context)
 105         if not self._get_delivery_methods(cr, uid, order, context=context):
 106             errors.append(('No delivery method available', 'There is no available delivery method for your order'))
 107         return errors
 108
 109     def _get_website_data(self, cr, uid, order, context=None):
 110         """ Override to add delivery-related website data. """
 111         values = super(SaleOrder, self)._get_website_data(cr, uid, order, context=context)
 112         # We need a delivery only if we have stockable products
 113         has_stockable_products = False
 114         for line in order.order_line:
 115             if line.product_id.type in ('consu', 'product'):
 116                 has_stockable_products = True
 117         if not has_stockable_products:
 118             return values
 119
 120         delivery_ctx = dict(context, order_id=order.id)
 121         DeliveryCarrier = self.pool.get('delivery.carrier')
 122         delivery_ids = self._get_delivery_methods(cr, uid, order, context=context)
 123
 124         values['deliveries'] = DeliveryCarrier.browse(cr, SUPERUSER_ID, delivery_ids, context=delivery_ctx)
 125         return values