addons/delivery/delivery.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 import time
  23 from osv import fields,osv
  24 from tools.translate import _
  25
  26 class delivery_carrier(osv.osv):
  27     _name = "delivery.carrier"
  28     _description = "Carrier"
  29
  30     def name_get(self, cr, uid, ids, context=None):
  31         if not len(ids):
  32             return []
  33         if context is None:
  34             context = {}
  35         order_id = context.get('order_id',False)
  36         if not order_id:
  37             res = super(delivery_carrier, self).name_get(cr, uid, ids, context=context)
  38         else:
  39             order = self.pool.get('sale.order').browse(cr, uid, order_id, context=context)
  40             currency = order.pricelist_id.currency_id.name or ''
  41             res = [(r['id'], r['name']+' ('+(str(r['price']))+' '+currency+')') for r in self.read(cr, uid, ids, ['name', 'price'], context)]
  42         return res
  43     def get_price(self, cr, uid, ids, field_name, arg=None, context=None):
  44         res={}
  45         if context is None:
  46             context = {}
  47         sale_obj=self.pool.get('sale.order')
  48         grid_obj=self.pool.get('delivery.grid')
  49         for carrier in self.browse(cr, uid, ids, context=context):
  50             order_id=context.get('order_id',False)
  51             price=False
  52             if order_id:
  53               order = sale_obj.browse(cr, uid, order_id, context=context)
  54               carrier_grid=self.grid_get(cr,uid,[carrier.id],order.partner_shipping_id.id,context)
  55               if carrier_grid:
  56                   price=grid_obj.get_price(cr, uid, carrier_grid, order, time.strftime('%Y-%m-%d'), context)
  57               else:
  58                   price = 0.0
  59             res[carrier.id]=price
  60         return res
  61     _columns = {
  62         'name': fields.char('Carrier', size=64, required=True),
  63         'partner_id': fields.many2one('res.partner', 'Carrier Partner', required=True),
  64         'product_id': fields.many2one('product.product', 'Delivery Product', required=True),
  65         'grids_id': fields.one2many('delivery.grid', 'carrier_id', 'Delivery Grids'),
  66         'price' : fields.function(get_price, method=True,string='Price'),
  67         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the delivery carrier without removing it.")
  68     }
  69     _defaults = {
  70         'active': lambda *args:1
  71     }
  72     def grid_get(self, cr, uid, ids, contact_id, context=None):
  73         contact = self.pool.get('res.partner.address').browse(cr, uid, contact_id, context=context)
  74         for carrier in self.browse(cr, uid, ids, context=context):
  75             for grid in carrier.grids_id:
  76                 get_id = lambda x: x.id
  77                 country_ids = map(get_id, grid.country_ids)
  78                 state_ids = map(get_id, grid.state_ids)
  79                 if country_ids and not contact.country_id.id in country_ids:
  80                     continue
  81                 if state_ids and not contact.state_id.id in state_ids:
  82                     continue
  83                 if grid.zip_from and (contact.zip or '')< grid.zip_from:
  84                     continue
  85                 if grid.zip_to and (contact.zip or '')> grid.zip_to:
  86                     continue
  87                 return grid.id
  88         return False
  89 delivery_carrier()
  90
  91 class delivery_grid(osv.osv):
  92     _name = "delivery.grid"
  93     _description = "Delivery Grid"
  94     _columns = {
  95         'name': fields.char('Grid Name', size=64, required=True),
  96         'sequence': fields.integer('Sequence', size=64, required=True, help="Gives the sequence order when displaying a list of delivery grid."),
  97         'carrier_id': fields.many2one('delivery.carrier', 'Carrier', required=True, ondelete='cascade'),
  98         'country_ids': fields.many2many('res.country', 'delivery_grid_country_rel', 'grid_id', 'country_id', 'Countries'),
  99         'state_ids': fields.many2many('res.country.state', 'delivery_grid_state_rel', 'grid_id', 'state_id', 'States'),
 100         'zip_from': fields.char('Start Zip', size=12),
 101         'zip_to': fields.char('To Zip', size=12),
 102         'line_ids': fields.one2many('delivery.grid.line', 'grid_id', 'Grid Line'),
 103         'active': fields.boolean('Active', help="If the active field is set to False, it will allow you to hide the delivery grid without removing it."),
 104     }
 105     _defaults = {
 106         'active': lambda *a: 1,
 107         'sequence': lambda *a: 1,
 108     }
 109     _order = 'sequence'
 110
 111     def get_price(self, cr, uid, id, order, dt, context=None):
 112         total = 0
 113         weight = 0
 114         volume = 0
 115         for line in order.order_line:
 116             if not line.product_id:
 117                 continue
 118             total += line.price_subtotal or 0.0
 119             weight += (line.product_id.weight or 0.0) * line.product_uom_qty
 120             volume += (line.product_id.volume or 0.0) * line.product_uom_qty
 121
 122
 123         return self.get_price_from_picking(cr, uid, id, total,weight, volume, context=context)
 124
 125     def get_price_from_picking(self, cr, uid, id, total, weight, volume, context=None):
 126         grid = self.browse(cr, uid, id, context=context)
 127         price = 0.0
 128         ok = False
 129
 130         for line in grid.line_ids:
 131             price_dict = {'price': total, 'volume':volume, 'weight': weight, 'wv':volume*weight}
 132             test = eval(line.type+line.operator+str(line.max_value), price_dict)
 133             if test:
 134                 if line.price_type=='variable':
 135                     price = line.list_price * price_dict[line.variable_factor]
 136                 else:
 137                     price = line.list_price
 138                 ok = True
 139                 break
 140         if not ok:
 141             raise osv.except_osv(_('No price available !'), _('No line matched this order in the choosed delivery grids !'))
 142
 143         return price
 144
 145
 146 delivery_grid()
 147
 148 class delivery_grid_line(osv.osv):
 149     _name = "delivery.grid.line"
 150     _description = "Delivery Grid Line"
 151     _columns = {
 152         'name': fields.char('Name', size=32, required=True),
 153         'grid_id': fields.many2one('delivery.grid', 'Grid',required=True),
 154         'type': fields.selection([('weight','Weight'),('volume','Volume'),('wv','Weight * Volume'), ('price','Price')], 'Variable', required=True),
 155         'operator': fields.selection([('==','='),('<=','<='),('>=','>=')], 'Operator', required=True),
 156         'max_value': fields.float('Maximum Value', required=True),
 157         'price_type': fields.selection([('fixed','Fixed'),('variable','Variable')], 'Price Type', required=True),
 158         'variable_factor': fields.selection([('weight','Weight'),('volume','Volume'),('wv','Weight * Volume'), ('price','Price')], 'Variable Factor', required=True),
 159         'list_price': fields.float('Sale Price', required=True),
 160         'standard_price': fields.float('Cost Price', required=True),
 161     }
 162     _defaults = {
 163         'type': lambda *args: 'weight',
 164         'operator': lambda *args: '<=',
 165         'price_type': lambda *args: 'fixed',
 166         'variable_factor': lambda *args: 'weight',
 167     }
 168     _order = 'list_price'
 169
 170
 171 delivery_grid_line()
 172
 173
 174 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 175