3b8ea1c06c7423992d02ee6bcf08150a3a554cd0
[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 from openerp import netsvc
26
27 from openerp.osv import fields,osv
28 from openerp.tools.translate import _
29 import openerp.addons.decimal_precision as dp
30
31 class purchase_requisition(osv.osv):
32     _name = "purchase.requisition"
33     _description="Purchase Requisition"
34     _inherit = ['mail.thread', 'ir.needaction_mixin']
35     _columns = {
36         'name': fields.char('Requisition Reference', size=32,required=True),
37         'origin': fields.char('Source Document', 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')],
48             'Status', track_visibility='onchange', required=True)
49     }
50     _defaults = {
51         'date_start': lambda *args: time.strftime('%Y-%m-%d %H:%M:%S'),
52         'state': 'draft',
53         'exclusive': 'multiple',
54         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'purchase.requisition', context=c),
55         'user_id': lambda self, cr, uid, c: self.pool.get('res.users').browse(cr, uid, uid, c).id ,
56         'name': lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'purchase.order.requisition'),
57     }
58
59     def copy(self, cr, uid, id, default=None, context=None):
60         if not default:
61             default = {}
62         default.update({
63             'state':'draft',
64             'purchase_ids':[],
65             'name': self.pool.get('ir.sequence').get(cr, uid, 'purchase.order.requisition'),
66         })
67         return super(purchase_requisition, self).copy(cr, uid, id, default, context)
68     
69     def tender_cancel(self, cr, uid, ids, context=None):
70         purchase_order_obj = self.pool.get('purchase.order')
71         for purchase in self.browse(cr, uid, ids, context=context):
72             for purchase_id in purchase.purchase_ids:
73                 if str(purchase_id.state) in('draft'):
74                     purchase_order_obj.action_cancel(cr,uid,[purchase_id.id])
75         procurement_ids = self.pool['procurement.order'].search(cr, uid, [('requisition_id', 'in', ids)], context=context)
76         self.pool['procurement.order'].write(cr, uid, procurement_ids, {'state': 'cancel'}, context=context)
77         return self.write(cr, uid, ids, {'state': 'cancel'})
78
79     def tender_in_progress(self, cr, uid, ids, context=None):
80         return self.write(cr, uid, ids, {'state':'in_progress'} ,context=context)
81
82     def tender_reset(self, cr, uid, ids, context=None):
83         return self.write(cr, uid, ids, {'state': 'draft'})
84
85     def tender_done(self, cr, uid, ids, context=None):
86         procurement_ids = self.pool['procurement.order'].search(cr, uid, [('requisition_id', 'in', ids)], context=context)
87         self.pool['procurement.order'].write(cr, uid, procurement_ids, {'state': 'done'}, context=context)
88         return self.write(cr, uid, ids, {'state':'done', 'date_end':time.strftime('%Y-%m-%d %H:%M:%S')}, context=context)
89
90     def _planned_date(self, requisition, delay=0.0):
91         company = requisition.company_id
92         date_planned = False
93         if requisition.date_start:
94             date_planned = datetime.strptime(requisition.date_start, '%Y-%m-%d %H:%M:%S') - relativedelta(days=company.po_lead)
95         else:
96             date_planned = datetime.today() - relativedelta(days=company.po_lead)
97         if delay:
98             date_planned -= relativedelta(days=delay)
99         return date_planned and date_planned.strftime('%Y-%m-%d %H:%M:%S') or False
100
101     def _seller_details(self, cr, uid, requisition_line, supplier, context=None):
102         product_uom = self.pool.get('product.uom')
103         pricelist = self.pool.get('product.pricelist')
104         supplier_info = self.pool.get("product.supplierinfo")
105         product = requisition_line.product_id
106         default_uom_po_id = product.uom_po_id.id
107         qty = product_uom._compute_qty(cr, uid, requisition_line.product_uom_id.id, requisition_line.product_qty, default_uom_po_id)
108         seller_delay = 0.0
109         seller_price = False
110         seller_qty = False
111         for product_supplier in product.seller_ids:
112             if supplier.id ==  product_supplier.name and qty >= product_supplier.qty:
113                 seller_delay = product_supplier.delay
114                 seller_qty = product_supplier.qty
115         supplier_pricelist = supplier.property_product_pricelist_purchase or False
116         seller_price = pricelist.price_get(cr, uid, [supplier_pricelist.id], product.id, qty, supplier.id, {'uom': default_uom_po_id})[supplier_pricelist.id]
117         if seller_qty:
118             qty = max(qty,seller_qty)
119         date_planned = self._planned_date(requisition_line.requisition_id, seller_delay)
120         return seller_price, qty, default_uom_po_id, date_planned
121
122     def make_purchase_order(self, cr, uid, ids, partner_id, context=None):
123         """
124         Create New RFQ for Supplier
125         """
126         if context is None:
127             context = {}
128         assert partner_id, 'Supplier should be specified'
129         purchase_order = self.pool.get('purchase.order')
130         purchase_order_line = self.pool.get('purchase.order.line')
131         res_partner = self.pool.get('res.partner')
132         fiscal_position = self.pool.get('account.fiscal.position')
133         supplier = res_partner.browse(cr, uid, partner_id, context=context)
134         supplier_pricelist = supplier.property_product_pricelist_purchase or False
135         res = {}
136         for requisition in self.browse(cr, uid, ids, context=context):
137             if supplier.id in filter(lambda x: x, [rfq.state <> 'cancel' and rfq.partner_id.id or None for rfq in requisition.purchase_ids]):
138                  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)
139             location_id = requisition.warehouse_id.lot_input_id.id
140             purchase_id = purchase_order.create(cr, uid, {
141                         'origin': requisition.name,
142                         'partner_id': supplier.id,
143                         'pricelist_id': supplier_pricelist.id,
144                         'location_id': location_id,
145                         'company_id': requisition.company_id.id,
146                         'fiscal_position': supplier.property_account_position and supplier.property_account_position.id or False,
147                         'requisition_id':requisition.id,
148                         'notes':requisition.description,
149                         'warehouse_id':requisition.warehouse_id.id ,
150             })
151             res[requisition.id] = purchase_id
152             for line in requisition.line_ids:
153                 product = line.product_id
154                 seller_price, qty, default_uom_po_id, date_planned = self._seller_details(cr, uid, line, supplier, context=context)
155                 taxes_ids = product.supplier_taxes_id
156                 taxes = fiscal_position.map_tax(cr, uid, supplier.property_account_position, taxes_ids)
157                 purchase_order_line.create(cr, uid, {
158                     'order_id': purchase_id,
159                     'name': product.partner_ref,
160                     'product_qty': qty,
161                     'product_id': product.id,
162                     'product_uom': default_uom_po_id,
163                     'price_unit': seller_price,
164                     'date_planned': date_planned,
165                     'taxes_id': [(6, 0, taxes)],
166                 }, context=context)
167                 
168         return res
169
170
171 class purchase_requisition_line(osv.osv):
172
173     _name = "purchase.requisition.line"
174     _description="Purchase Requisition Line"
175     _rec_name = 'product_id'
176
177     _columns = {
178         'product_id': fields.many2one('product.product', 'Product' ),
179         'product_uom_id': fields.many2one('product.uom', 'Product Unit of Measure'),
180         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure')),
181         'requisition_id' : fields.many2one('purchase.requisition','Purchase Requisition', ondelete='cascade'),
182         'company_id': fields.related('requisition_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
183     }
184
185     def onchange_product_id(self, cr, uid, ids, product_id, product_uom_id, context=None):
186         """ Changes UoM and name if product_id changes.
187         @param name: Name of the field
188         @param product_id: Changed product_id
189         @return:  Dictionary of changed values
190         """
191         value = {'product_uom_id': ''}
192         if product_id:
193             prod = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
194             value = {'product_uom_id': prod.uom_id.id,'product_qty':1.0}
195         return {'value': value}
196
197     _defaults = {
198         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'purchase.requisition.line', context=c),
199     }
200 purchase_requisition_line()
201
202 class purchase_order(osv.osv):
203     _inherit = "purchase.order"
204     _columns = {
205         'requisition_id' : fields.many2one('purchase.requisition','Purchase Requisition')
206     }
207
208     def wkf_confirm_order(self, cr, uid, ids, context=None):
209         res = super(purchase_order, self).wkf_confirm_order(cr, uid, ids, context=context)
210         proc_obj = self.pool.get('procurement.order')
211         for po in self.browse(cr, uid, ids, context=context):
212             if po.requisition_id and (po.requisition_id.exclusive=='exclusive'):
213                 for order in po.requisition_id.purchase_ids:
214                     if order.id != po.id:
215                         proc_ids = proc_obj.search(cr, uid, [('purchase_id', '=', order.id)])
216                         if proc_ids and po.state=='confirmed':
217                             proc_obj.write(cr, uid, proc_ids, {'purchase_id': po.id})
218                         wf_service = netsvc.LocalService("workflow")
219                         wf_service.trg_validate(uid, 'purchase.order', order.id, 'purchase_cancel', cr)
220                     po.requisition_id.tender_done(context=context)
221         return res
222
223 purchase_order()
224
225 class product_product(osv.osv):
226     _inherit = 'product.product'
227
228     _columns = {
229         'purchase_requisition': fields.boolean('Purchase Requisition', help="Check this box to generates purchase requisition instead of generating requests for quotation from procurement.")
230     }
231     _defaults = {
232         'purchase_requisition': False
233     }
234
235 product_product()
236
237 class procurement_order(osv.osv):
238     _inherit = 'procurement.order'
239     _columns = {
240         'requisition_id': fields.many2one('purchase.requisition', 'Latest Requisition')
241     }
242
243     def _get_warehouse(self, procurement, user_company):
244         """
245             Return the warehouse containing the procurment stock location (or one of it ancestors)
246             If none match, returns then first warehouse of the company
247         """
248         # NOTE This method is a copy of the one on the procurement.order defined in purchase
249         #      module. It's been copied to ensure it been always available, even if module
250         #      purchase is not up to date.
251         #      Do not forget to update both version in case of modification.
252         company_id = (procurement.company_id or user_company).id
253         domains = [
254             [
255                 '&', ('company_id', '=', company_id),
256                 '|', '&', ('lot_stock_id.parent_left', '<', procurement.location_id.parent_left),
257                           ('lot_stock_id.parent_right', '>', procurement.location_id.parent_right),
258                      ('lot_stock_id', '=', procurement.location_id.id)
259             ],
260             [('company_id', '=', company_id)]
261         ]
262
263         cr, uid = procurement._cr, procurement._uid
264         context = procurement._context
265         Warehouse = self.pool['stock.warehouse']
266         for domain in domains:
267             ids = Warehouse.search(cr, uid, domain, context=context)
268             if ids:
269                 return ids[0]
270         return False
271
272     def make_po(self, cr, uid, ids, context=None):
273         res = {}
274         requisition_obj = self.pool.get('purchase.requisition')
275         non_requisition = []
276         for procurement in self.browse(cr, uid, ids, context=context):
277             if procurement.product_id.purchase_requisition:
278                 user_company = self.pool['res.users'].browse(cr, uid, uid, context=context).company_id
279                 req = requisition_obj.create(cr, uid, {
280                     'origin': procurement.origin,
281                     'date_end': procurement.date_planned,
282                     'warehouse_id': self._get_warehouse(procurement, user_company),
283                     'company_id': procurement.company_id.id,
284                     'line_ids': [(0, 0, {
285                         'product_id': procurement.product_id.id,
286                         'product_uom_id': procurement.product_uom.id,
287                         'product_qty': procurement.product_qty
288
289                     })],
290                 })
291                 procurement.write({
292                     'state': 'running',
293                     'requisition_id': req
294                 })
295                 res[procurement.id] = 0
296             else:
297                 non_requisition.append(procurement.id)
298
299         if non_requisition:
300             res.update(super(procurement_order, self).make_po(cr, uid, non_requisition, context=context))
301
302         return res
303
304 procurement_order()
305
306 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: