[MERGE]: Merged lp:~openerp-dev/openobject-addons/trunk-improve-purchase-requisition...
[odoo/odoo.git] / addons / purchase_requisition / purchase_requisition.py
1 # -*- encoding: utf-8 -*-
2 ##############################################################################
3 #
4 #    OpenERP, Open Source Management Solution
5 #    Copyright (C) 2004-2010 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 from datetime import datetime
23 from dateutil.relativedelta import relativedelta
24 import time
25 import netsvc
26
27 from osv import fields,osv
28 from tools.translate import _
29 import decimal_precision as dp
30
31 class purchase_requisition(osv.osv):
32     _name = "purchase.requisition"
33     _description="Purchase Requisition"
34     _inherit = ['ir.needaction_mixin', 'mail.thread']
35     _columns = {
36         'name': fields.char('Requisition Reference', size=32,required=True),
37         'origin': fields.char('Source', size=32),
38         'date_start': fields.datetime('Requisition Date'),
39         'date_end': fields.datetime('Requisition Deadline'),
40         'user_id': fields.many2one('res.users', 'Responsible'),
41         'exclusive': fields.selection([('exclusive','Purchase Requisition (exclusive)'),('multiple','Multiple Requisitions')],'Requisition Type', required=True, help="Purchase Requisition (exclusive):  On the confirmation of a purchase order, it cancels the remaining purchase order.\nPurchase Requisition(Multiple):  It allows to have multiple purchase orders.On confirmation of a purchase order it does not cancel the remaining orders"""),
42         'description': fields.text('Description'),
43         'company_id': fields.many2one('res.company', 'Company', required=True),
44         'purchase_ids' : fields.one2many('purchase.order','requisition_id','Purchase Orders',states={'done': [('readonly', True)]}),
45         'line_ids' : fields.one2many('purchase.requisition.line','requisition_id','Products to Purchase',states={'done': [('readonly', True)]}),
46         'warehouse_id': fields.many2one('stock.warehouse', 'Warehouse'),        
47         'state': fields.selection([('draft','New'),('in_progress','Sent to Suppliers'),('cancel','Cancelled'),('done','Purchase Done')], 'Status', required=True)
48     }
49     _defaults = {
50         'date_start': time.strftime('%Y-%m-%d %H:%M:%S'),
51         'state': 'draft',
52         'exclusive': 'multiple',
53         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'purchase.requisition', context=c),
54         'user_id': lambda self, cr, uid, c: self.pool.get('res.users').browse(cr, uid, uid, c).id ,
55         'name': lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'purchase.order.requisition'),
56     }
57     
58     def copy(self, cr, uid, id, default=None, context=None):
59         if not default:
60             default = {}
61         default.update({
62             'state':'draft',
63             'purchase_ids':[],
64             'name': self.pool.get('ir.sequence').get(cr, uid, 'purchase.order.requisition'),
65         })
66         return super(purchase_requisition, self).copy(cr, uid, id, default, context)
67     def tender_cancel(self, cr, uid, ids, context=None):
68         purchase_order_obj = self.pool.get('purchase.order')
69         for purchase in self.browse(cr, uid, ids, context=context):
70             for purchase_id in purchase.purchase_ids:
71                 if str(purchase_id.state) in('draft','wait'):
72                     purchase_order_obj.action_cancel(cr,uid,[purchase_id.id])
73         self.write(cr, uid, ids, {'state': 'cancel'})
74         return True
75
76     def tender_in_progress(self, cr, uid, ids, context=None):
77         self.write(cr, uid, ids, {'state':'in_progress'} ,context=context)
78         return True
79
80     def tender_reset(self, cr, uid, ids, context=None):
81         self.write(cr, uid, ids, {'state': 'draft'})
82         return True
83
84     def tender_done(self, cr, uid, ids, context=None):
85         self.write(cr, uid, ids, {'state':'done', 'date_end':time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
86         return True
87
88     def _planned_date(self, requisition, delay=0.0):
89         company = requisition.company_id
90         date_planned = False
91         if requisition.date_start:
92             date_planned = datetime.strptime(requisition.date_start, '%Y-%m-%d %H:%M:%S') - relativedelta(days=company.po_lead)
93         else:
94             date_planned = datetime.today() - relativedelta(days=company.po_lead)
95         if delay:
96             date_planned -= relativedelta(days=delay)
97         return date_planned and date_planned.strftime('%Y-%m-%d %H:%M:%S') or False
98
99     def _seller_details(self, cr, uid, requisition_line, supplier, context=None):
100         product_uom = self.pool.get('product.uom')
101         pricelist = self.pool.get('product.pricelist')
102         supplier_info = self.pool.get("product.supplierinfo")
103         product = requisition_line.product_id
104         default_uom_po_id = product.uom_po_id.id
105         qty = product_uom._compute_qty(cr, uid, requisition_line.product_uom_id.id, requisition_line.product_qty, default_uom_po_id)
106         seller_delay = 0.0
107         seller_price = False
108         seller_qty = False
109         for product_supplier in product.seller_ids:
110             if supplier.id ==  product_supplier.name and qty >= product_supplier.qty:
111                 seller_delay = product_supplier.delay
112                 seller_qty = product_supplier.qty
113         supplier_pricelist = supplier.property_product_pricelist_purchase or False
114         seller_price = pricelist.price_get(cr, uid, [supplier_pricelist.id], product.id, qty, False, {'uom': default_uom_po_id})[supplier_pricelist.id]
115         if seller_qty:
116             qty = max(qty,seller_qty)
117         date_planned = self._planned_date(requisition_line.requisition_id, seller_delay)
118         return seller_price, qty, default_uom_po_id, date_planned
119
120     def make_purchase_order(self, cr, uid, ids, partner_id, context=None):
121         """
122         Create New RFQ for Supplier
123         """
124         if context is None:
125             context = {}
126         assert partner_id, 'Supplier should be specified'
127         purchase_order = self.pool.get('purchase.order')
128         purchase_order_line = self.pool.get('purchase.order.line')
129         res_partner = self.pool.get('res.partner')
130         fiscal_position = self.pool.get('account.fiscal.position')
131         supplier = res_partner.browse(cr, uid, partner_id, context=context)
132         supplier_pricelist = supplier.property_product_pricelist_purchase or False
133         res = {}
134         for requisition in self.browse(cr, uid, ids, context=context):
135             if supplier.id in filter(lambda x: x, [rfq.state <> 'cancel' and rfq.partner_id.id or None for rfq in requisition.purchase_ids]):
136                  raise osv.except_osv(_('Warning'), _('You have already one %s purchase order for this partner, you must cancel this purchase order to create a new quotation.') % rfq.state)
137             location_id = requisition.warehouse_id.lot_input_id.id
138             purchase_id = purchase_order.create(cr, uid, {
139                         'origin': requisition.name,
140                         'partner_id': supplier.id,
141                         'pricelist_id': supplier_pricelist.id,
142                         'location_id': location_id,
143                         'company_id': requisition.company_id.id,
144                         'fiscal_position': supplier.property_account_position and supplier.property_account_position.id or False,
145                         'requisition_id':requisition.id,
146                         'notes':requisition.description,
147                         'warehouse_id':requisition.warehouse_id.id ,
148             })
149             res[requisition.id] = purchase_id
150             for line in requisition.line_ids:
151                 product = line.product_id
152                 seller_price, qty, default_uom_po_id, date_planned = self._seller_details(cr, uid, line, supplier, context=context)
153                 taxes_ids = product.supplier_taxes_id
154                 taxes = fiscal_position.map_tax(cr, uid, supplier.property_account_position, taxes_ids)
155                 purchase_order_line.create(cr, uid, {
156                     'order_id': purchase_id,
157                     'name': product.partner_ref,
158                     'product_qty': qty,
159                     'product_id': product.id,
160                     'product_uom': default_uom_po_id,
161                     'price_unit': seller_price,
162                     'date_planned': date_planned,
163                     'notes': product.description_purchase,
164                     'taxes_id': [(6, 0, taxes)],
165                 }, context=context)
166                 
167         return res
168     
169     def get_needaction_user_ids(self, cr, uid, ids, context=None):
170         result = dict.fromkeys(ids, [])
171         for obj in self.browse(cr, uid, ids, context=context):
172             if (obj.state == 'draft'):
173                 result[obj.id] = [obj.user_id.id]
174         return result
175
176 purchase_requisition()
177
178 class purchase_requisition_line(osv.osv):
179
180     _name = "purchase.requisition.line"
181     _description="Purchase Requisition Line"
182     _rec_name = 'product_id'
183
184     _columns = {
185         'product_id': fields.many2one('product.product', 'Product' ),
186         'product_uom_id': fields.many2one('product.uom', 'Product Unit of Measure'),
187         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure')),
188         'requisition_id' : fields.many2one('purchase.requisition','Purchase Requisition', ondelete='cascade'),
189         'company_id': fields.related('requisition_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
190     }
191
192     def onchange_product_id(self, cr, uid, ids, product_id,product_uom_id, context=None):
193         """ Changes UoM and name if product_id changes.
194         @param name: Name of the field
195         @param product_id: Changed product_id
196         @return:  Dictionary of changed values
197         """
198         value = {'product_uom_id': ''}
199         if product_id:
200             prod = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
201             value = {'product_uom_id': prod.uom_id.id,'product_qty':1.0}
202         return {'value': value}
203
204     _defaults = {
205         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'purchase.requisition.line', context=c),
206     }
207 purchase_requisition_line()
208
209 class purchase_order(osv.osv):
210     _inherit = "purchase.order"
211     _columns = {
212         'requisition_id' : fields.many2one('purchase.requisition','Purchase Requisition')
213     }
214     def default_get(self, cr, uid, fields, context=None):
215         if not context:
216             context = {}
217         val_list = []
218         requisition_id = {}
219         product_obj = self.pool.get('product.product')
220         requisition_line_obj = self.pool.get('purchase.requisition.line')
221         purchase_order_line_obj = self.pool.get('purchase.order.line')
222         res = super(purchase_order, self).default_get(cr, uid, fields, context=context)
223         product_ids = context.get('product_ids', False)
224         if product_ids:
225             for id in product_ids:
226                 if  id[0] == 4:
227                     requisition_data = requisition_line_obj.browse(cr, uid, id[1], context=context)
228                     requisition_id.update({'requisition_id': requisition_data.requisition_id.id})
229                     line_vals={'product_id': requisition_data.product_id.id, 'product_qty': requisition_data.product_qty, 'name': requisition_data.product_id.name}
230                 elif id[0] == 0:
231                    product_data = id[2]
232                    name = product_obj.browse(cr, uid, product_data.get('product_id'), context=context).name
233                    line_vals = {'product_id': product_data.get('product_id'), 'product_qty': product_data.get('product_qty'), 'name': name}
234                 val_list.append((0,0,line_vals))
235         res.update({'order_line': val_list, 'requisition_id': requisition_id.get('requisition_id')})
236         return res
237         
238     def onchange_partner_id(self, cr, uid, ids, partner_id, requisition_id):
239         res = {}
240         res = super(purchase_order, self).onchange_partner_id(cr, uid, ids, partner_id)
241         res_partner = self.pool.get('res.partner')
242         value = {'partner_id': ''}
243         requisition_pool = self.pool.get('purchase.requisition')
244         supplier = res_partner.browse(cr, uid, partner_id)
245         if requisition_id: 
246             requisition = requisition_pool.browse(cr, uid, requisition_id)
247             if supplier.id in filter(lambda x: x, [rfq.state <> 'cancel' and rfq.partner_id.id or None for rfq in requisition.purchase_ids]):
248                  raise osv.except_osv(_('Warning'), _('You have already one %s purchase order for this partner, you must cancel this purchase order to create a new quotation.') % rfq.state)
249         return res
250         
251     def wkf_confirm_order(self, cr, uid, ids, context=None):
252         res = super(purchase_order, self).wkf_confirm_order(cr, uid, ids, context=context)
253         proc_obj = self.pool.get('procurement.order')
254         for po in self.browse(cr, uid, ids, context=context):
255             if po.requisition_id and (po.requisition_id.exclusive=='exclusive'):
256                 for order in po.requisition_id.purchase_ids:
257                     if order.id<>po.id:
258                         proc_ids = proc_obj.search(cr, uid, [('purchase_id', '=', order.id)])
259                         if proc_ids and po.state=='confirmed':
260                             proc_obj.write(cr, uid, proc_ids, {'purchase_id': po.id})
261                         wf_service = netsvc.LocalService("workflow")
262                         wf_service.trg_validate(uid, 'purchase.order', order.id, 'purchase_cancel', cr)
263                     po.requisition_id.tender_done(context=context)
264         return res
265
266 purchase_order()
267
268 class product_product(osv.osv):
269     _inherit = 'product.product'
270
271     _columns = {
272         'purchase_requisition': fields.boolean('Purchase Requisition', help="Check this box so that procurement generates  purchase requisition instead of only generating requests for quotation.")
273     }
274     _defaults = {
275         'purchase_requisition': False
276     }
277
278 product_product()
279
280 class procurement_order(osv.osv):
281
282     _inherit = 'procurement.order'
283     _columns = {
284         'requisition_id' : fields.many2one('purchase.requisition','Latest Requisition')
285     }
286     def make_po(self, cr, uid, ids, context=None):
287         sequence_obj = self.pool.get('ir.sequence')
288         res = super(procurement_order, self).make_po(cr, uid, ids, context=context)
289         for proc_id, po_id in res.items():
290             procurement = self.browse(cr, uid, proc_id, context=context)
291             requisition_id=False
292             if procurement.product_id.purchase_requisition:
293                 requisition_id=self.pool.get('purchase.requisition').create(cr, uid, {
294                     'name': sequence_obj.get(cr, uid, 'purchase.order.requisition'),
295                     'origin': procurement.origin,
296                     'date_end': procurement.date_planned,
297                     'warehouse_id':procurement.purchase_id and procurement.purchase_id.warehouse_id.id,
298                     'company_id':procurement.company_id.id,
299                     'line_ids': [(0,0,{
300                         'product_id': procurement.product_id.id,
301                         'product_uom_id': procurement.product_uom.id,
302                         'product_qty': procurement.product_qty
303
304                     })],
305                     'purchase_ids': [(6,0,[po_id])]
306                 })
307             self.write(cr,uid,proc_id,{'requisition_id':requisition_id})
308         return res
309
310 procurement_order()
311
312 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: