addons/sale/stock.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 # Copyright (c) 2004-2008 TINY SPRL. (http://tiny.be) All Rights Reserved.
   5 #
   6 # $Id$
   7 #
   8 # WARNING: This program as such is intended to be used by professional
   9 # programmers who take the whole responsability of assessing all potential
  10 # consequences resulting from its eventual inadequacies and bugs
  11 # End users who are looking for a ready-to-use solution with commercial
  12 # garantees and support are strongly adviced to contract a Free Software
  13 # Service Company
  14 #
  15 # This program is Free Software; you can redistribute it and/or
  16 # modify it under the terms of the GNU General Public License
  17 # as published by the Free Software Foundation; either version 2
  18 # of the License, or (at your option) any later version.
  19 #
  20 # This program is distributed in the hope that it will be useful,
  21 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 # GNU General Public License for more details.
  24 #
  25 # You should have received a copy of the GNU General Public License
  26 # along with this program; if not, write to the Free Software
  27 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  28 #
  29 ##############################################################################
  30
  31 from osv import osv, fields
  32
  33 class stock_move(osv.osv):
  34     _inherit = 'stock.move'
  35     _columns = {
  36         'sale_line_id': fields.many2one('sale.order.line', 'Sale Order Line', ondelete='set null', select=True, readonly=True),
  37     }
  38     _defaults = {
  39         'sale_line_id': lambda *a:False
  40     }
  41 stock_move()
  42
  43 class stock_picking(osv.osv):
  44     _inherit = 'stock.picking'
  45     _columns = {
  46         'sale_id': fields.many2one('sale.order', 'Sale Order', ondelete='set null', select=True, readonly=True),
  47     }
  48     _defaults = {
  49         'sale_id': lambda *a: False
  50     }
  51
  52     def _get_address_invoice(self, cursor, user, picking):
  53         res = {}
  54         if picking.sale_id:
  55             res['contact'] = picking.sale_id.partner_order_id.id
  56             res['invoice'] = picking.sale_id.partner_invoice_id.id
  57             return res
  58         return super(stock_picking, self)._get_address_invoice(cursor,
  59                 user, picking)
  60
  61     def _get_comment_invoice(self, cursor, user, picking):
  62         if picking.sale_id and picking.sale_id.note:
  63             if picking.note:
  64                 return picking.note + '\n' + picking.sale_id.note
  65             else:
  66                 return picking.sale_id.note
  67         return super(stock_picking, self)._get_comment_invoice(cursor, user,
  68                 picking)
  69
  70     def _get_price_unit_invoice(self, cursor, user, move_line, type):
  71         if move_line.sale_line_id:
  72             return move_line.sale_line_id.price_unit
  73         return super(stock_picking, self)._get_price_unit_invoice(cursor,
  74                 user, move_line, type)
  75
  76     def _get_discount_invoice(self, cursor, user, move_line):
  77         if move_line.sale_line_id:
  78             return move_line.sale_line_id.discount
  79         return super(stock_picking, self)._get_discount_invoice(cursor, user,
  80                 move_line)
  81
  82     def _get_taxes_invoice(self, cursor, user, move_line, type):
  83         if move_line.sale_line_id:
  84             return [x.id for x in move_line.sale_line_id.tax_id]
  85         return super(stock_picking, self)._get_taxes_invoice(cursor, user,
  86                 move_line, type)
  87
  88     def _get_account_analytic_invoice(self, cursor, user, picking, move_line):
  89         if picking.sale_id:
  90             return picking.sale_id.project_id.id
  91         return super(stock_picking, self)._get_account_analytic_invoice(cursor,
  92                 user, picking, move_line)
  93
  94     def _invoice_line_hook(self, cursor, user, move_line, invoice_line_id):
  95         sale_line_obj = self.pool.get('sale.order.line')
  96         if move_line.sale_line_id:
  97             sale_line_obj.write(cursor, user, [move_line.sale_line_id.id], {
  98                 'invoice_lines': [(6, 0, [invoice_line_id])],
  99                 })
 100         return super(stock_picking, self)._invoice_line_hook(cursor, user,
 101                 move_line, invoice_line_id)
 102
 103     def _invoice_hook(self, cursor, user, picking, invoice_id):
 104         sale_obj = self.pool.get('sale.order')
 105         if picking.sale_id:
 106             sale_obj.write(cursor, user, [picking.sale_id.id], {
 107                 'invoice_ids': [(4, invoice_id)],
 108                 })
 109         return super(stock_picking, self)._invoice_hook(cursor, user,
 110                 picking, invoice_id)
 111
 112 stock_picking()
 113
 114
 115
 116 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
 117