addons/sale_layout/sale_layout.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2008 Tiny SPRL (<http://tiny.be>). All Rights Reserved
   6 #    $Id$
   7 #
   8 #    This program is free software: you can redistribute it and/or modify
   9 #    it under the terms of the GNU Affero General Public License as published by
  10 #    the Free Software Foundation, either version 3 of the License, or
  11 #    (at your option) any later version.
  12 #
  13 #    This program is distributed in the hope that it will be useful,
  14 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  15 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16 #    GNU Affero General Public License for more details.
  17 #
  18 #    You should have received a copy of the GNU Affero General Public License
  19 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 #
  21 ##############################################################################
  22
  23 from osv import fields, osv
  24 import decimal_precision as dp
  25
  26 class sale_order_line(osv.osv):
  27
  28     def _amount_line(self, cr, uid, ids, field_name, arg, context=None):
  29         res = {}
  30         for line in self.browse(cr, uid, ids, context=context):
  31             if line.layout_type == 'article':
  32                 return super(sale_order_line, self)._amount_line(cr, uid, ids, field_name, arg, context)
  33         return res
  34
  35     def invoice_line_create(self, cr, uid, ids, context=None):
  36         new_ids = []
  37         list_seq = []
  38         for line in self.browse(cr, uid, ids, context=context):
  39             if line.layout_type == 'article':
  40                 new_ids.append(line.id)
  41                 list_seq.append(line.sequence)
  42         invoice_line_ids = super(sale_order_line, self).invoice_line_create(cr, uid, new_ids, context)
  43         pool_inv_line = self.pool.get('account.invoice.line')
  44         seq = 0
  45         for obj_inv_line in pool_inv_line.browse(cr, uid, invoice_line_ids, context=context):
  46             pool_inv_line.write(cr, uid, [obj_inv_line.id], {'sequence': list_seq[seq]}, context=context)
  47             seq += 1
  48         return invoice_line_ids
  49
  50     def onchange_sale_order_line_view(self, cr, uid, id, type, context={}, *args):
  51         temp = {
  52             'value': {
  53                 'name': '',
  54                 'product_id': False,
  55                 'uos_id': False,
  56                 'account_id': False,
  57                 'price_unit': 0.0,
  58                 'price_subtotal': 0.0,
  59                 'quantity': 0,
  60                 'discount': 0.0,
  61                 'invoice_line_tax_id': False,
  62                 'account_analytic_id': False,
  63                 'product_uom_qty': 0.0,
  64             },
  65         }
  66         if type == 'line':
  67             temp['value']['name'] = '___'
  68         if type == 'break':
  69             temp['value']['name'] = '·····Page Break·····'
  70         if type == 'subtotal':
  71             temp['value']['name'] = 'Sub Total'
  72         return temp
  73
  74     def create(self, cr, user, vals, context=None):
  75         if vals.has_key('layout_type'):
  76             if vals['layout_type'] == 'line':
  77                 vals['name'] = '___'
  78             if vals['layout_type'] == 'break':
  79                 vals['name'] = '·····Page Break·····'
  80             if vals['layout_type'] != 'article':
  81                 vals['product_uom_qty']= 0
  82         return super(sale_order_line, self).create(cr, user, vals, context)
  83
  84     def write(self, cr, user, ids, vals, context=None):
  85         if vals.has_key('layout_type'):
  86             if vals['layout_type'] == 'line':
  87                 vals['name'] = '___'
  88             if vals['layout_type'] == 'break':
  89                 vals['name'] = '·····Page Break·····'
  90         return super(sale_order_line, self).write(cr, user, ids, vals, context)
  91
  92     def copy(self, cr, uid, id, default=None, context=None):
  93         if default is None:
  94             default = {}
  95         default['layout_type'] = self.browse(cr, uid, id, context=context).layout_type
  96         return super(sale_order_line, self).copy(cr, uid, id, default, context)
  97
  98     _order = "order_id, sequence asc"
  99     _description = "Sales Order line"
 100     _inherit = "sale.order.line"
 101     _columns = {
 102         'layout_type': fields.selection([
 103                 ('article', 'Product'),
 104                 ('title', 'Title'),
 105                 ('text', 'Note'),
 106                 ('subtotal', 'Sub Total'),
 107                 ('line', 'Separator Line'),
 108                 ('break', 'Page Break'),]
 109             ,'Layout Type', select=True, required=True),
 110         'sequence': fields.integer('Layout Sequence'),
 111         'price_unit': fields.float('Unit Price', required=True, digits_compute= dp.get_precision('Sale Price'), readonly=True, states={'draft': [('readonly', False)]}),
 112         'product_uom_qty': fields.float('Quantity (UoM)', digits=(16,2)),
 113         'product_uom': fields.many2one('product.uom', 'Product UoM'),
 114     }
 115
 116     _defaults = {
 117         'layout_type': 'article',
 118     }
 119
 120 sale_order_line()
 121
 122 class one2many_mod2(fields.one2many):
 123     def get(self, cr, obj, ids, name, user=None, offset=0, context=None, values=None):
 124         if not values:
 125             values = {}
 126         res = {}
 127         for id in ids:
 128             res[id] = []
 129         ids2 = obj.pool.get(self._obj).search(cr, user, [(self._fields_id, 'in', ids), ('layout_type', '=', 'article')], limit=self._limit)
 130         for r in obj.pool.get(self._obj)._read_flat(cr, user, ids2, [self._fields_id], context=context, load='_classic_write'):
 131             res[r[self._fields_id]].append( r['id'] )
 132         return res
 133
 134
 135 class sale_order(osv.osv):
 136
 137     def copy(self, cr, uid, id, default=None, context=None):
 138         if default is None:
 139             default = {}
 140         default['order_line'] = False
 141         return super(sale_order, self).copy(cr, uid, id, default, context)
 142
 143     _inherit = "sale.order"
 144     _columns = {
 145         'abstract_line_ids': fields.one2many('sale.order.line', 'order_id', 'Order Lines', readonly=True, states={'draft': [('readonly', False)]}),
 146         'order_line': one2many_mod2('sale.order.line', 'order_id', 'Order Lines', readonly=True, states={'draft': [('readonly', False)]}),
 147     }
 148
 149 sale_order()
 150
 151 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: