[FIX] account_payment: the amount_residual field is not searchable so it cannot be...
[odoo/odoo.git] / addons / purchase / purchase.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 import pytz
24 from openerp import SUPERUSER_ID
25 from datetime import datetime
26 from dateutil.relativedelta import relativedelta
27 from operator import attrgetter
28
29 from openerp.osv import fields, osv
30 from openerp.tools.translate import _
31 import openerp.addons.decimal_precision as dp
32 from openerp.osv.orm import browse_record, browse_null
33 from openerp.tools import DEFAULT_SERVER_DATE_FORMAT, DEFAULT_SERVER_DATETIME_FORMAT, DATETIME_FORMATS_MAP
34
35 class purchase_order(osv.osv):
36
37     def _amount_all(self, cr, uid, ids, field_name, arg, context=None):
38         res = {}
39         cur_obj=self.pool.get('res.currency')
40         for order in self.browse(cr, uid, ids, context=context):
41             res[order.id] = {
42                 'amount_untaxed': 0.0,
43                 'amount_tax': 0.0,
44                 'amount_total': 0.0,
45             }
46             val = val1 = 0.0
47             cur = order.pricelist_id.currency_id
48             for line in order.order_line:
49                val1 += line.price_subtotal
50                for c in self.pool.get('account.tax').compute_all(cr, uid, line.taxes_id, line.price_unit, line.product_qty, line.product_id, order.partner_id)['taxes']:
51                     val += c.get('amount', 0.0)
52             res[order.id]['amount_tax']=cur_obj.round(cr, uid, cur, val)
53             res[order.id]['amount_untaxed']=cur_obj.round(cr, uid, cur, val1)
54             res[order.id]['amount_total']=res[order.id]['amount_untaxed'] + res[order.id]['amount_tax']
55         return res
56
57     def _set_minimum_planned_date(self, cr, uid, ids, name, value, arg, context=None):
58         if not value: return False
59         if type(ids)!=type([]):
60             ids=[ids]
61         for po in self.browse(cr, uid, ids, context=context):
62             if po.order_line:
63                 cr.execute("""update purchase_order_line set
64                         date_planned=%s
65                     where
66                         order_id=%s and
67                         (date_planned=%s or date_planned<%s)""", (value,po.id,po.minimum_planned_date,value))
68             cr.execute("""update purchase_order set
69                     minimum_planned_date=%s where id=%s""", (value, po.id))
70         return True
71
72     def _minimum_planned_date(self, cr, uid, ids, field_name, arg, context=None):
73         res={}
74         purchase_obj=self.browse(cr, uid, ids, context=context)
75         for purchase in purchase_obj:
76             res[purchase.id] = False
77             if purchase.order_line:
78                 min_date=purchase.order_line[0].date_planned
79                 for line in purchase.order_line:
80                     if line.date_planned < min_date:
81                         min_date=line.date_planned
82                 res[purchase.id]=min_date
83         return res
84
85
86     def _invoiced_rate(self, cursor, user, ids, name, arg, context=None):
87         res = {}
88         for purchase in self.browse(cursor, user, ids, context=context):
89             tot = 0.0
90             for invoice in purchase.invoice_ids:
91                 if invoice.state not in ('draft','cancel'):
92                     tot += invoice.amount_untaxed
93             if purchase.amount_untaxed:
94                 res[purchase.id] = tot * 100.0 / purchase.amount_untaxed
95             else:
96                 res[purchase.id] = 0.0
97         return res
98
99     def _shipped_rate(self, cr, uid, ids, name, arg, context=None):
100         if not ids: return {}
101         res = {}
102         for id in ids:
103             res[id] = [0.0,0.0]
104         cr.execute('''SELECT
105                 p.purchase_id,sum(m.product_qty), m.state
106             FROM
107                 stock_move m
108             LEFT JOIN
109                 stock_picking p on (p.id=m.picking_id)
110             WHERE
111                 p.purchase_id IN %s GROUP BY m.state, p.purchase_id''',(tuple(ids),))
112         for oid,nbr,state in cr.fetchall():
113             if state=='cancel':
114                 continue
115             if state=='done':
116                 res[oid][0] += nbr or 0.0
117                 res[oid][1] += nbr or 0.0
118             else:
119                 res[oid][1] += nbr or 0.0
120         for r in res:
121             if not res[r][1]:
122                 res[r] = 0.0
123             else:
124                 res[r] = 100.0 * res[r][0] / res[r][1]
125         return res
126
127     def _get_order(self, cr, uid, ids, context=None):
128         result = {}
129         for line in self.pool.get('purchase.order.line').browse(cr, uid, ids, context=context):
130             result[line.order_id.id] = True
131         return result.keys()
132
133     def _invoiced(self, cursor, user, ids, name, arg, context=None):
134         res = {}
135         for purchase in self.browse(cursor, user, ids, context=context):
136             invoiced = False
137             if purchase.invoiced_rate == 100.00:
138                 invoiced = True
139             res[purchase.id] = invoiced
140         return res
141     
142     def _get_journal(self, cr, uid, context=None):
143         if context is None:
144             context = {}
145         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
146         company_id = context.get('company_id', user.company_id.id)
147         journal_obj = self.pool.get('account.journal')
148         res = journal_obj.search(cr, uid, [('type', '=', 'purchase'),
149                                             ('company_id', '=', company_id)],
150                                                 limit=1)
151         return res and res[0] or False  
152
153     STATE_SELECTION = [
154         ('draft', 'Draft PO'),
155         ('sent', 'RFQ Sent'),
156         ('confirmed', 'Waiting Approval'),
157         ('approved', 'Purchase Order'),
158         ('except_picking', 'Shipping Exception'),
159         ('except_invoice', 'Invoice Exception'),
160         ('done', 'Done'),
161         ('cancel', 'Cancelled')
162     ]
163     _track = {
164         'state': {
165             'purchase.mt_rfq_confirmed': lambda self, cr, uid, obj, ctx=None: obj.state == 'confirmed',
166             'purchase.mt_rfq_approved': lambda self, cr, uid, obj, ctx=None: obj.state == 'approved',
167             'purchase.mt_rfq_done': lambda self, cr, uid, obj, ctx=None: obj.state == 'done',
168         },
169     }
170     _columns = {
171         'name': fields.char('Order Reference', size=64, required=True, select=True, help="Unique number of the purchase order, computed automatically when the purchase order is created."),
172         'origin': fields.char('Source Document', size=64,
173             help="Reference of the document that generated this purchase order request; a sales order or an internal procurement request."
174         ),
175         'partner_ref': fields.char('Supplier Reference', states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)],'done':[('readonly',True)]}, size=64,
176             help="Reference of the sales order or quotation sent by your supplier. It's mainly used to do the matching when you receive the products as this reference is usually written on the delivery order sent by your supplier."),
177         'date_order':fields.date('Order Date', required=True, states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)]}, select=True, help="Date on which this document has been created."),
178         'date_approve':fields.date('Date Approved', readonly=1, select=True, help="Date on which purchase order has been approved"),
179         'partner_id':fields.many2one('res.partner', 'Supplier', required=True, states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)],'done':[('readonly',True)]},
180             change_default=True, track_visibility='always'),
181         'dest_address_id':fields.many2one('res.partner', 'Customer Address (Direct Delivery)',
182             states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)],'done':[('readonly',True)]},
183             help="Put an address if you want to deliver directly from the supplier to the customer. " \
184                 "Otherwise, keep empty to deliver to your own company."
185         ),
186         'warehouse_id': fields.many2one('stock.warehouse', 'Destination Warehouse'),
187         'location_id': fields.many2one('stock.location', 'Destination', required=True, domain=[('usage','<>','view')], states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)],'done':[('readonly',True)]} ),
188         'pricelist_id':fields.many2one('product.pricelist', 'Pricelist', required=True, states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)],'done':[('readonly',True)]}, help="The pricelist sets the currency used for this purchase order. It also computes the supplier price for the selected products/quantities."),
189         'currency_id': fields.many2one('res.currency','Currency', readonly=True, required=True,states={'draft': [('readonly', False)],'sent': [('readonly', False)]}),
190         'state': fields.selection(STATE_SELECTION, 'Status', readonly=True, help="The status of the purchase order or the quotation request. A quotation is a purchase order in a 'Draft' status. Then the order has to be confirmed by the user, the status switch to 'Confirmed'. Then the supplier must confirm the order to change the status to 'Approved'. When the purchase order is paid and received, the status becomes 'Done'. If a cancel action occurs in the invoice or in the reception of goods, the status becomes in exception.", select=True),
191         'order_line': fields.one2many('purchase.order.line', 'order_id', 'Order Lines', states={'approved':[('readonly',True)],'done':[('readonly',True)]}),
192         'validator' : fields.many2one('res.users', 'Validated by', readonly=True),
193         'notes': fields.text('Terms and Conditions'),
194         'invoice_ids': fields.many2many('account.invoice', 'purchase_invoice_rel', 'purchase_id', 'invoice_id', 'Invoices', help="Invoices generated for a purchase order"),
195         'picking_ids': fields.one2many('stock.picking.in', 'purchase_id', 'Picking List', readonly=True, help="This is the list of incoming shipments that have been generated for this purchase order."),
196         'shipped':fields.boolean('Received', readonly=True, select=True, help="It indicates that a picking has been done"),
197         'shipped_rate': fields.function(_shipped_rate, string='Received Ratio', type='float'),
198         'invoiced': fields.function(_invoiced, string='Invoice Received', type='boolean', help="It indicates that an invoice has been paid"),
199         'invoiced_rate': fields.function(_invoiced_rate, string='Invoiced', type='float'),
200         'invoice_method': fields.selection([('manual','Based on Purchase Order lines'),('order','Based on generated draft invoice'),('picking','Based on incoming shipments')], 'Invoicing Control', required=True,
201             readonly=True, states={'draft':[('readonly',False)], 'sent':[('readonly',False)]},
202             help="Based on Purchase Order lines: place individual lines in 'Invoice Control / On Purchase Order lines' from where you can selectively create an invoice.\n" \
203                 "Based on generated invoice: create a draft invoice you can validate later.\n" \
204                 "Based on incoming shipments: let you create an invoice when receptions are validated."
205         ),
206         'minimum_planned_date':fields.function(_minimum_planned_date, fnct_inv=_set_minimum_planned_date, string='Expected Date', type='date', select=True, help="This is computed as the minimum scheduled date of all purchase order lines' products.",
207             store = {
208                 'purchase.order.line': (_get_order, ['date_planned'], 10),
209             }
210         ),
211         'amount_untaxed': fields.function(_amount_all, digits_compute= dp.get_precision('Account'), string='Untaxed Amount',
212             store={
213                 'purchase.order.line': (_get_order, None, 10),
214             }, multi="sums", help="The amount without tax", track_visibility='always'),
215         'amount_tax': fields.function(_amount_all, digits_compute= dp.get_precision('Account'), string='Taxes',
216             store={
217                 'purchase.order.line': (_get_order, None, 10),
218             }, multi="sums", help="The tax amount"),
219         'amount_total': fields.function(_amount_all, digits_compute= dp.get_precision('Account'), string='Total',
220             store={
221                 'purchase.order.line': (_get_order, None, 10),
222             }, multi="sums",help="The total amount"),
223         'fiscal_position': fields.many2one('account.fiscal.position', 'Fiscal Position'),
224         'payment_term_id': fields.many2one('account.payment.term', 'Payment Term'),
225         'product_id': fields.related('order_line','product_id', type='many2one', relation='product.product', string='Product'),
226         'create_uid':  fields.many2one('res.users', 'Responsible'),
227         'company_id': fields.many2one('res.company','Company',required=True,select=1, states={'confirmed':[('readonly',True)], 'approved':[('readonly',True)]}),
228         'journal_id': fields.many2one('account.journal', 'Journal'),
229     }
230     _defaults = {
231         'date_order': fields.date.context_today,
232         'state': 'draft',
233         'name': lambda obj, cr, uid, context: '/',
234         'shipped': 0,
235         'invoice_method': 'order',
236         'invoiced': 0,
237         'pricelist_id': lambda self, cr, uid, context: context.get('partner_id', False) and self.pool.get('res.partner').browse(cr, uid, context['partner_id']).property_product_pricelist_purchase.id,
238         'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'purchase.order', context=c),
239         'journal_id': _get_journal,
240         'currency_id': lambda self, cr, uid, context: self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.currency_id.id
241     }
242     _sql_constraints = [
243         ('name_uniq', 'unique(name, company_id)', 'Order Reference must be unique per Company!'),
244     ]
245     _name = "purchase.order"
246     _inherit = ['mail.thread', 'ir.needaction_mixin']
247     _description = "Purchase Order"
248     _order = "name desc"
249
250     def create(self, cr, uid, vals, context=None):
251         if vals.get('name','/')=='/':
252             vals['name'] = self.pool.get('ir.sequence').get(cr, uid, 'purchase.order') or '/'
253         if context is None:
254             context = {}
255         context.update({'mail_create_nolog': True})
256         order =  super(purchase_order, self).create(cr, uid, vals, context=context)
257         self.message_post(cr, uid, [order], body=_("RFQ created"), context=context)
258         return order
259
260     def unlink(self, cr, uid, ids, context=None):
261         purchase_orders = self.read(cr, uid, ids, ['state'], context=context)
262         unlink_ids = []
263         for s in purchase_orders:
264             if s['state'] in ['draft','cancel']:
265                 unlink_ids.append(s['id'])
266             else:
267                 raise osv.except_osv(_('Invalid Action!'), _('In order to delete a purchase order, you must cancel it first.'))
268
269         # automatically sending subflow.delete upon deletion
270         self.signal_purchase_cancel(cr, uid, unlink_ids)
271
272         return super(purchase_order, self).unlink(cr, uid, unlink_ids, context=context)
273
274     def button_dummy(self, cr, uid, ids, context=None):
275         return True
276
277     def onchange_pricelist(self, cr, uid, ids, pricelist_id, context=None):
278         if not pricelist_id:
279             return {}
280         return {'value': {'currency_id': self.pool.get('product.pricelist').browse(cr, uid, pricelist_id, context=context).currency_id.id}}
281
282     def onchange_dest_address_id(self, cr, uid, ids, address_id):
283         if not address_id:
284             return {}
285         address = self.pool.get('res.partner')
286         values = {'warehouse_id': False}
287         supplier = address.browse(cr, uid, address_id)
288         if supplier:
289             location_id = supplier.property_stock_customer.id
290             values.update({'location_id': location_id})
291         return {'value':values}
292
293     def onchange_warehouse_id(self, cr, uid, ids, warehouse_id):
294         if not warehouse_id:
295             return {}
296         warehouse = self.pool.get('stock.warehouse').browse(cr, uid, warehouse_id)
297         return {'value':{'location_id': warehouse.lot_input_id.id, 'dest_address_id': False}}
298
299     def onchange_partner_id(self, cr, uid, ids, partner_id):
300         partner = self.pool.get('res.partner')
301         if not partner_id:
302             return {'value': {
303                 'fiscal_position': False,
304                 'payment_term_id': False,
305                 }}
306         supplier_address = partner.address_get(cr, uid, [partner_id], ['default'])
307         supplier = partner.browse(cr, uid, partner_id)
308         return {'value': {
309             'pricelist_id': supplier.property_product_pricelist_purchase.id,
310             'fiscal_position': supplier.property_account_position and supplier.property_account_position.id or False,
311             'payment_term_id': supplier.property_supplier_payment_term.id or False,
312             }}
313
314     def invoice_open(self, cr, uid, ids, context=None):
315         mod_obj = self.pool.get('ir.model.data')
316         act_obj = self.pool.get('ir.actions.act_window')
317
318         result = mod_obj.get_object_reference(cr, uid, 'account', 'action_invoice_tree2')
319         id = result and result[1] or False
320         result = act_obj.read(cr, uid, [id], context=context)[0]
321         inv_ids = []
322         for po in self.browse(cr, uid, ids, context=context):
323             inv_ids+= [invoice.id for invoice in po.invoice_ids]
324         if not inv_ids:
325             raise osv.except_osv(_('Error!'), _('Please create Invoices.'))
326          #choose the view_mode accordingly
327         if len(inv_ids)>1:
328             result['domain'] = "[('id','in',["+','.join(map(str, inv_ids))+"])]"
329         else:
330             res = mod_obj.get_object_reference(cr, uid, 'account', 'invoice_supplier_form')
331             result['views'] = [(res and res[1] or False, 'form')]
332             result['res_id'] = inv_ids and inv_ids[0] or False
333         return result
334
335     def view_invoice(self, cr, uid, ids, context=None):
336         '''
337         This function returns an action that display existing invoices of given sales order ids. It can either be a in a list or in a form view, if there is only one invoice to show.
338         '''
339         mod_obj = self.pool.get('ir.model.data')
340         wizard_obj = self.pool.get('purchase.order.line_invoice')
341         #compute the number of invoices to display
342         inv_ids = []
343         for po in self.browse(cr, uid, ids, context=context):
344             if po.invoice_method == 'manual':
345                 if not po.invoice_ids:
346                     context.update({'active_ids' :  [line.id for line in po.order_line]})
347                     wizard_obj.makeInvoices(cr, uid, [], context=context)
348
349         for po in self.browse(cr, uid, ids, context=context):
350             inv_ids+= [invoice.id for invoice in po.invoice_ids]
351         res = mod_obj.get_object_reference(cr, uid, 'account', 'invoice_supplier_form')
352         res_id = res and res[1] or False
353
354         return {
355             'name': _('Supplier Invoices'),
356             'view_type': 'form',
357             'view_mode': 'form',
358             'view_id': [res_id],
359             'res_model': 'account.invoice',
360             'context': "{'type':'in_invoice', 'journal_type': 'purchase'}",
361             'type': 'ir.actions.act_window',
362             'nodestroy': True,
363             'target': 'current',
364             'res_id': inv_ids and inv_ids[0] or False,
365         }
366
367     def view_picking(self, cr, uid, ids, context=None):
368         '''
369         This function returns an action that display existing pîcking orders of given purchase order ids.
370         '''
371         mod_obj = self.pool.get('ir.model.data')
372         pick_ids = []
373         for po in self.browse(cr, uid, ids, context=context):
374             pick_ids += [picking.id for picking in po.picking_ids]
375
376         action_model, action_id = tuple(mod_obj.get_object_reference(cr, uid, 'stock', 'action_picking_tree4'))
377         action = self.pool[action_model].read(cr, uid, action_id, context=context)
378         ctx = eval(action['context'])
379         ctx.update({
380             'search_default_purchase_id': ids[0]
381         })
382         if pick_ids and len(pick_ids) == 1:
383             form_view_ids = [view_id for view_id, view in action['views'] if view == 'form']
384             view_id = form_view_ids and form_view_ids[0] or False
385             action.update({
386                 'views': [],
387                 'view_mode': 'form',
388                 'view_id': view_id,
389                 'res_id': pick_ids[0]
390             })
391
392         action.update({
393             'context': ctx,
394         })
395         return action
396
397     def wkf_approve_order(self, cr, uid, ids, context=None):
398         self.write(cr, uid, ids, {'state': 'approved', 'date_approve': fields.date.context_today(self,cr,uid,context=context)})
399         return True
400
401     def print_confirm(self,cr,uid,ids,context=None):
402         print "Confirmed"
403
404     def print_double(self,cr,uid,ids,context=None):
405         print "double Approval"
406
407     def print_router(self,cr,uid,ids,context=None):
408         print "Routed"
409
410     def wkf_send_rfq(self, cr, uid, ids, context=None):
411         '''
412         This function opens a window to compose an email, with the edi purchase template message loaded by default
413         '''
414         ir_model_data = self.pool.get('ir.model.data')
415         try:
416             template_id = ir_model_data.get_object_reference(cr, uid, 'purchase', 'email_template_edi_purchase')[1]
417         except ValueError:
418             template_id = False
419         try:
420             compose_form_id = ir_model_data.get_object_reference(cr, uid, 'mail', 'email_compose_message_wizard_form')[1]
421         except ValueError:
422             compose_form_id = False 
423         ctx = dict(context)
424         ctx.update({
425             'default_model': 'purchase.order',
426             'default_res_id': ids[0],
427             'default_use_template': bool(template_id),
428             'default_template_id': template_id,
429             'default_composition_mode': 'comment',
430         })
431         return {
432             'name': _('Compose Email'),
433             'type': 'ir.actions.act_window',
434             'view_type': 'form',
435             'view_mode': 'form',
436             'res_model': 'mail.compose.message',
437             'views': [(compose_form_id, 'form')],
438             'view_id': compose_form_id,
439             'target': 'new',
440             'context': ctx,
441         }
442
443     def print_quotation(self, cr, uid, ids, context=None):
444         '''
445         This function prints the request for quotation and mark it as sent, so that we can see more easily the next step of the workflow
446         '''
447         assert len(ids) == 1, 'This option should only be used for a single id at a time'
448         self.signal_send_rfq(cr, uid, ids)
449         datas = {
450                  'model': 'purchase.order',
451                  'ids': ids,
452                  'form': self.read(cr, uid, ids[0], context=context),
453         }
454         return {'type': 'ir.actions.report.xml', 'report_name': 'purchase.quotation', 'datas': datas, 'nodestroy': True}
455
456     #TODO: implement messages system
457     def wkf_confirm_order(self, cr, uid, ids, context=None):
458         todo = []
459         for po in self.browse(cr, uid, ids, context=context):
460             if not po.order_line:
461                 raise osv.except_osv(_('Error!'),_('You cannot confirm a purchase order without any purchase order line.'))
462             for line in po.order_line:
463                 if line.state=='draft':
464                     todo.append(line.id)
465
466         self.pool.get('purchase.order.line').action_confirm(cr, uid, todo, context)
467         for id in ids:
468             self.write(cr, uid, [id], {'state' : 'confirmed', 'validator' : uid})
469         return True
470
471     def _choose_account_from_po_line(self, cr, uid, po_line, context=None):
472         fiscal_obj = self.pool.get('account.fiscal.position')
473         property_obj = self.pool.get('ir.property')
474         if po_line.product_id:
475             acc_id = po_line.product_id.property_account_expense.id
476             if not acc_id:
477                 acc_id = po_line.product_id.categ_id.property_account_expense_categ.id
478             if not acc_id:
479                 raise osv.except_osv(_('Error!'), _('Define expense account for this company: "%s" (id:%d).') % (po_line.product_id.name, po_line.product_id.id,))
480         else:
481             acc_id = property_obj.get(cr, uid, 'property_account_expense_categ', 'product.category').id
482         fpos = po_line.order_id.fiscal_position or False
483         return fiscal_obj.map_account(cr, uid, fpos, acc_id)
484
485     def _prepare_inv_line(self, cr, uid, account_id, order_line, context=None):
486         """Collects require data from purchase order line that is used to create invoice line
487         for that purchase order line
488         :param account_id: Expense account of the product of PO line if any.
489         :param browse_record order_line: Purchase order line browse record
490         :return: Value for fields of invoice lines.
491         :rtype: dict
492         """
493         return {
494             'name': order_line.name,
495             'account_id': account_id,
496             'price_unit': order_line.price_unit or 0.0,
497             'quantity': order_line.product_qty,
498             'product_id': order_line.product_id.id or False,
499             'uos_id': order_line.product_uom.id or False,
500             'invoice_line_tax_id': [(6, 0, [x.id for x in order_line.taxes_id])],
501             'account_analytic_id': order_line.account_analytic_id.id or False,
502             'purchase_line_id': order_line.id,
503         }
504
505     def action_cancel_draft(self, cr, uid, ids, context=None):
506         if not len(ids):
507             return False
508         self.write(cr, uid, ids, {'state':'draft','shipped':0})
509         for p_id in ids:
510             # Deleting the existing instance of workflow for PO
511             self.delete_workflow(cr, uid, [p_id]) # TODO is it necessary to interleave the calls?
512             self.create_workflow(cr, uid, [p_id])
513         return True
514
515     def action_invoice_create(self, cr, uid, ids, context=None):
516         """Generates invoice for given ids of purchase orders and links that invoice ID to purchase order.
517         :param ids: list of ids of purchase orders.
518         :return: ID of created invoice.
519         :rtype: int
520         """
521         res = False
522
523         journal_obj = self.pool.get('account.journal')
524         inv_obj = self.pool.get('account.invoice')
525         inv_line_obj = self.pool.get('account.invoice.line')
526
527         for order in self.browse(cr, uid, ids, context=context):
528             pay_acc_id = order.partner_id.property_account_payable.id
529             journal_ids = journal_obj.search(cr, uid, [('type', '=','purchase'),('company_id', '=', order.company_id.id)], limit=1)
530             if not journal_ids:
531                 raise osv.except_osv(_('Error!'),
532                     _('Define purchase journal for this company: "%s" (id:%d).') % (order.company_id.name, order.company_id.id))
533
534             # generate invoice line correspond to PO line and link that to created invoice (inv_id) and PO line
535             inv_lines = []
536             for po_line in order.order_line:
537                 acc_id = self._choose_account_from_po_line(cr, uid, po_line, context=context)
538                 inv_line_data = self._prepare_inv_line(cr, uid, acc_id, po_line, context=context)
539                 inv_line_id = inv_line_obj.create(cr, uid, inv_line_data, context=context)
540                 inv_lines.append(inv_line_id)
541
542                 po_line.write({'invoiced':True, 'invoice_lines': [(4, inv_line_id)]}, context=context)
543
544             # get invoice data and create invoice
545             inv_data = {
546                 'name': order.partner_ref or order.name,
547                 'reference': order.partner_ref or order.name,
548                 'account_id': pay_acc_id,
549                 'type': 'in_invoice',
550                 'partner_id': order.partner_id.id,
551                 'currency_id': order.currency_id.id,
552                 'journal_id': len(journal_ids) and journal_ids[0] or False,
553                 'invoice_line': [(6, 0, inv_lines)],
554                 'origin': order.name,
555                 'fiscal_position': order.fiscal_position.id or False,
556                 'payment_term': order.payment_term_id.id or False,
557                 'company_id': order.company_id.id,
558             }
559             inv_id = inv_obj.create(cr, uid, inv_data, context=context)
560
561             # compute the invoice
562             inv_obj.button_compute(cr, uid, [inv_id], context=context, set_total=True)
563
564             # Link this new invoice to related purchase order
565             order.write({'invoice_ids': [(4, inv_id)]}, context=context)
566             res = inv_id
567         return res
568
569     def invoice_done(self, cr, uid, ids, context=None):
570         self.write(cr, uid, ids, {'state':'approved'}, context=context)
571         return True
572
573     def has_stockable_product(self, cr, uid, ids, *args):
574         for order in self.browse(cr, uid, ids):
575             for order_line in order.order_line:
576                 if order_line.product_id and order_line.product_id.type in ('product', 'consu'):
577                     return True
578         return False
579
580     def action_cancel(self, cr, uid, ids, context=None):
581         for purchase in self.browse(cr, uid, ids, context=context):
582             for pick in purchase.picking_ids:
583                 if pick.state not in ('draft','cancel'):
584                     raise osv.except_osv(
585                         _('Unable to cancel this purchase order.'),
586                         _('First cancel all receptions related to this purchase order.'))
587             self.pool.get('stock.picking') \
588                 .signal_button_cancel(cr, uid, map(attrgetter('id'), purchase.picking_ids))
589             for inv in purchase.invoice_ids:
590                 if inv and inv.state not in ('cancel','draft'):
591                     raise osv.except_osv(
592                         _('Unable to cancel this purchase order.'),
593                         _('You must first cancel all receptions related to this purchase order.'))
594             self.pool.get('account.invoice') \
595                 .signal_invoice_cancel(cr, uid, map(attrgetter('id'), purchase.invoice_ids))
596         self.write(cr,uid,ids,{'state':'cancel'})
597
598         self.signal_purchase_cancel(cr, uid, ids)
599         return True
600
601     def date_to_datetime(self, cr, uid, userdate, context=None):
602         """ Convert date values expressed in user's timezone to
603         server-side UTC timestamp, assuming a default arbitrary
604         time of 12:00 AM - because a time is needed.
605     
606         :param str userdate: date string in in user time zone
607         :return: UTC datetime string for server-side use
608         """
609         # TODO: move to fields.datetime in server after 7.0
610         user_date = datetime.strptime(userdate, DEFAULT_SERVER_DATE_FORMAT)
611         if context and context.get('tz'):
612             tz_name = context['tz']
613         else:
614             tz_name = self.pool.get('res.users').read(cr, SUPERUSER_ID, uid, ['tz'])['tz']
615         if tz_name:
616             utc = pytz.timezone('UTC')
617             context_tz = pytz.timezone(tz_name)
618             user_datetime = user_date + relativedelta(hours=12.0)
619             local_timestamp = context_tz.localize(user_datetime, is_dst=False)
620             user_datetime = local_timestamp.astimezone(utc)
621             return user_datetime.strftime(DEFAULT_SERVER_DATETIME_FORMAT)
622         return user_date.strftime(DEFAULT_SERVER_DATETIME_FORMAT)
623
624     def _prepare_order_picking(self, cr, uid, order, context=None):
625         return {
626             'name': self.pool.get('ir.sequence').get(cr, uid, 'stock.picking.in'),
627             'origin': order.name + ((order.origin and (':' + order.origin)) or ''),
628             'date': self.date_to_datetime(cr, uid, order.date_order, context),
629             'partner_id': order.dest_address_id.id or order.partner_id.id,
630             'invoice_state': '2binvoiced' if order.invoice_method == 'picking' else 'none',
631             'type': 'in',
632             'partner_id': order.dest_address_id.id or order.partner_id.id,
633             'purchase_id': order.id,
634             'company_id': order.company_id.id,
635             'move_lines' : [],
636         }
637
638     def _prepare_order_line_move(self, cr, uid, order, order_line, picking_id, context=None):
639         ''' prepare the stock move data from the PO line '''
640         return {
641             'name': order_line.name or '',
642             'product_id': order_line.product_id.id,
643             'product_qty': order_line.product_qty,
644             'product_uos_qty': order_line.product_qty,
645             'product_uom': order_line.product_uom.id,
646             'product_uos': order_line.product_uom.id,
647             'date': self.date_to_datetime(cr, uid, order.date_order, context),
648             'date_expected': self.date_to_datetime(cr, uid, order_line.date_planned, context),
649             'location_id': order.partner_id.property_stock_supplier.id,
650             'location_dest_id': order.location_id.id,
651             'picking_id': picking_id,
652             'partner_id': order.dest_address_id.id or order.partner_id.id,
653             'move_dest_id': order_line.move_dest_id.id,
654             'state': 'draft',
655             'type':'in',
656             'purchase_line_id': order_line.id,
657             'company_id': order.company_id.id,
658             'price_unit': order_line.price_unit
659         }
660
661     def _create_pickings(self, cr, uid, order, order_lines, picking_id=False, context=None):
662         """Creates pickings and appropriate stock moves for given order lines, then
663         confirms the moves, makes them available, and confirms the picking.
664
665         If ``picking_id`` is provided, the stock moves will be added to it, otherwise
666         a standard outgoing picking will be created to wrap the stock moves, as returned
667         by :meth:`~._prepare_order_picking`.
668
669         Modules that wish to customize the procurements or partition the stock moves over
670         multiple stock pickings may override this method and call ``super()`` with
671         different subsets of ``order_lines`` and/or preset ``picking_id`` values.
672
673         :param browse_record order: purchase order to which the order lines belong
674         :param list(browse_record) order_lines: purchase order line records for which picking
675                                                 and moves should be created.
676         :param int picking_id: optional ID of a stock picking to which the created stock moves
677                                will be added. A new picking will be created if omitted.
678         :return: list of IDs of pickings used/created for the given order lines (usually just one)
679         """
680         stock_picking = self.pool.get('stock.picking')
681         if not picking_id:
682             picking_id = stock_picking.create(cr, uid, self._prepare_order_picking(cr, uid, order, context=context))
683         todo_moves = []
684         stock_move = self.pool.get('stock.move')
685         for order_line in order_lines:
686             if not order_line.product_id:
687                 continue
688             if order_line.product_id.type in ('product', 'consu'):
689                 move = stock_move.create(cr, uid, self._prepare_order_line_move(cr, uid, order, order_line, picking_id, context=context))
690                 if order_line.move_dest_id:
691                     order_line.move_dest_id.write({'location_id': order.location_id.id})
692                 todo_moves.append(move)
693         stock_move.action_confirm(cr, uid, todo_moves)
694         stock_move.force_assign(cr, uid, todo_moves)
695         stock_picking.signal_button_confirm(cr, uid, [picking_id])
696         return [picking_id]
697
698     def action_picking_create(self, cr, uid, ids, context=None):
699         picking_ids = []
700         for order in self.browse(cr, uid, ids):
701             picking_ids.extend(self._create_pickings(cr, uid, order, order.order_line, None, context=context))
702
703         # Must return one unique picking ID: the one to connect in the subflow of the purchase order.
704         # In case of multiple (split) pickings, we should return the ID of the critical one, i.e. the
705         # one that should trigger the advancement of the purchase workflow.
706         # By default we will consider the first one as most important, but this behavior can be overridden.
707         return picking_ids[0] if picking_ids else False
708
709     def picking_done(self, cr, uid, ids, context=None):
710         self.write(cr, uid, ids, {'shipped':1,'state':'approved'}, context=context)
711         self.message_post(cr, uid, ids, body=_("Products received"), context=context)
712         return True
713
714     def copy(self, cr, uid, id, default=None, context=None):
715         if not default:
716             default = {}
717         default.update({
718             'state':'draft',
719             'shipped':False,
720             'invoiced':False,
721             'invoice_ids': [],
722             'picking_ids': [],
723             'name': self.pool.get('ir.sequence').get(cr, uid, 'purchase.order'),
724         })
725         return super(purchase_order, self).copy(cr, uid, id, default, context)
726
727     def do_merge(self, cr, uid, ids, context=None):
728         """
729         To merge similar type of purchase orders.
730         Orders will only be merged if:
731         * Purchase Orders are in draft
732         * Purchase Orders belong to the same partner
733         * Purchase Orders are have same stock location, same pricelist
734         Lines will only be merged if:
735         * Order lines are exactly the same except for the quantity and unit
736
737          @param self: The object pointer.
738          @param cr: A database cursor
739          @param uid: ID of the user currently logged in
740          @param ids: the ID or list of IDs
741          @param context: A standard dictionary
742
743          @return: new purchase order id
744
745         """
746         #TOFIX: merged order line should be unlink
747         def make_key(br, fields):
748             list_key = []
749             for field in fields:
750                 field_val = getattr(br, field)
751                 if field in ('product_id', 'move_dest_id', 'account_analytic_id'):
752                     if not field_val:
753                         field_val = False
754                 if isinstance(field_val, browse_record):
755                     field_val = field_val.id
756                 elif isinstance(field_val, browse_null):
757                     field_val = False
758                 elif isinstance(field_val, list):
759                     field_val = ((6, 0, tuple([v.id for v in field_val])),)
760                 list_key.append((field, field_val))
761             list_key.sort()
762             return tuple(list_key)
763
764         if context is None:
765             context = {}
766
767         # Compute what the new orders should contain
768
769         new_orders = {}
770
771         for porder in [order for order in self.browse(cr, uid, ids, context=context) if order.state == 'draft']:
772             order_key = make_key(porder, ('partner_id', 'location_id', 'pricelist_id'))
773             new_order = new_orders.setdefault(order_key, ({}, []))
774             new_order[1].append(porder.id)
775             order_infos = new_order[0]
776             if not order_infos:
777                 order_infos.update({
778                     'origin': porder.origin,
779                     'date_order': porder.date_order,
780                     'partner_id': porder.partner_id.id,
781                     'dest_address_id': porder.dest_address_id.id,
782                     'warehouse_id': porder.warehouse_id.id,
783                     'location_id': porder.location_id.id,
784                     'pricelist_id': porder.pricelist_id.id,
785                     'state': 'draft',
786                     'order_line': {},
787                     'notes': '%s' % (porder.notes or '',),
788                     'fiscal_position': porder.fiscal_position and porder.fiscal_position.id or False,
789                 })
790             else:
791                 if porder.date_order < order_infos['date_order']:
792                     order_infos['date_order'] = porder.date_order
793                 if porder.notes:
794                     order_infos['notes'] = (order_infos['notes'] or '') + ('\n%s' % (porder.notes,))
795                 if porder.origin:
796                     order_infos['origin'] = (order_infos['origin'] or '') + ' ' + porder.origin
797
798             for order_line in porder.order_line:
799                 line_key = make_key(order_line, ('name', 'date_planned', 'taxes_id', 'price_unit', 'product_id', 'move_dest_id', 'account_analytic_id'))
800                 o_line = order_infos['order_line'].setdefault(line_key, {})
801                 if o_line:
802                     # merge the line with an existing line
803                     o_line['product_qty'] += order_line.product_qty * order_line.product_uom.factor / o_line['uom_factor']
804                 else:
805                     # append a new "standalone" line
806                     for field in ('product_qty', 'product_uom'):
807                         field_val = getattr(order_line, field)
808                         if isinstance(field_val, browse_record):
809                             field_val = field_val.id
810                         o_line[field] = field_val
811                     o_line['uom_factor'] = order_line.product_uom and order_line.product_uom.factor or 1.0
812
813
814
815         allorders = []
816         orders_info = {}
817         for order_key, (order_data, old_ids) in new_orders.iteritems():
818             # skip merges with only one order
819             if len(old_ids) < 2:
820                 allorders += (old_ids or [])
821                 continue
822
823             # cleanup order line data
824             for key, value in order_data['order_line'].iteritems():
825                 del value['uom_factor']
826                 value.update(dict(key))
827             order_data['order_line'] = [(0, 0, value) for value in order_data['order_line'].itervalues()]
828
829             # create the new order
830             context.update({'mail_create_nolog': True})
831             neworder_id = self.create(cr, uid, order_data)
832             self.message_post(cr, uid, [neworder_id], body=_("RFQ created"), context=context)
833             orders_info.update({neworder_id: old_ids})
834             allorders.append(neworder_id)
835
836             # make triggers pointing to the old orders point to the new order
837             for old_id in old_ids:
838                 self.redirect_workflow(cr, uid, [(old_id, neworder_id)])
839                 self.signal_purchase_cancel(cr, uid, [old_id]) # TODO Is it necessary to interleave the calls?
840         return orders_info
841
842
843 class purchase_order_line(osv.osv):
844     def _amount_line(self, cr, uid, ids, prop, arg, context=None):
845         res = {}
846         cur_obj=self.pool.get('res.currency')
847         tax_obj = self.pool.get('account.tax')
848         for line in self.browse(cr, uid, ids, context=context):
849             taxes = tax_obj.compute_all(cr, uid, line.taxes_id, line.price_unit, line.product_qty, line.product_id, line.order_id.partner_id)
850             cur = line.order_id.pricelist_id.currency_id
851             res[line.id] = cur_obj.round(cr, uid, cur, taxes['total'])
852         return res
853
854     _columns = {
855         'name': fields.text('Description', required=True),
856         'product_qty': fields.float('Quantity', digits_compute=dp.get_precision('Product Unit of Measure'), required=True),
857         'date_planned': fields.date('Scheduled Date', required=True, select=True),
858         'taxes_id': fields.many2many('account.tax', 'purchase_order_taxe', 'ord_id', 'tax_id', 'Taxes'),
859         'product_uom': fields.many2one('product.uom', 'Product Unit of Measure', required=True),
860         'product_id': fields.many2one('product.product', 'Product', domain=[('purchase_ok','=',True)], change_default=True),
861         'move_ids': fields.one2many('stock.move', 'purchase_line_id', 'Reservation', readonly=True, ondelete='set null'),
862         'move_dest_id': fields.many2one('stock.move', 'Reservation Destination', ondelete='set null'),
863         'price_unit': fields.float('Unit Price', required=True, digits_compute= dp.get_precision('Product Price')),
864         'price_subtotal': fields.function(_amount_line, string='Subtotal', digits_compute= dp.get_precision('Account')),
865         'order_id': fields.many2one('purchase.order', 'Order Reference', select=True, required=True, ondelete='cascade'),
866         'account_analytic_id':fields.many2one('account.analytic.account', 'Analytic Account',),
867         'company_id': fields.related('order_id','company_id',type='many2one',relation='res.company',string='Company', store=True, readonly=True),
868         'state': fields.selection([('draft', 'Draft'), ('confirmed', 'Confirmed'), ('done', 'Done'), ('cancel', 'Cancelled')], 'Status', required=True, readonly=True,
869                                   help=' * The \'Draft\' status is set automatically when purchase order in draft status. \
870                                        \n* The \'Confirmed\' status is set automatically as confirm when purchase order in confirm status. \
871                                        \n* The \'Done\' status is set automatically when purchase order is set as done. \
872                                        \n* The \'Cancelled\' status is set automatically when user cancel purchase order.'),
873         'invoice_lines': fields.many2many('account.invoice.line', 'purchase_order_line_invoice_rel', 'order_line_id', 'invoice_id', 'Invoice Lines', readonly=True),
874         'invoiced': fields.boolean('Invoiced', readonly=True),
875         'partner_id': fields.related('order_id','partner_id',string='Partner',readonly=True,type="many2one", relation="res.partner", store=True),
876         'date_order': fields.related('order_id','date_order',string='Order Date',readonly=True,type="date"),
877
878     }
879     _defaults = {
880         'product_qty': lambda *a: 1.0,
881         'state': lambda *args: 'draft',
882         'invoiced': lambda *a: 0,
883     }
884     _table = 'purchase_order_line'
885     _name = 'purchase.order.line'
886     _description = 'Purchase Order Line'
887
888     def copy_data(self, cr, uid, id, default=None, context=None):
889         if not default:
890             default = {}
891         default.update({'state':'draft', 'move_ids':[],'invoiced':0,'invoice_lines':[]})
892         return super(purchase_order_line, self).copy_data(cr, uid, id, default, context)
893
894     def onchange_product_uom(self, cr, uid, ids, pricelist_id, product_id, qty, uom_id,
895             partner_id, date_order=False, fiscal_position_id=False, date_planned=False,
896             name=False, price_unit=False, context=None):
897         """
898         onchange handler of product_uom.
899         """
900         if not uom_id:
901             return {'value': {'price_unit': price_unit or 0.0, 'name': name or '', 'product_uom' : uom_id or False}}
902         return self.onchange_product_id(cr, uid, ids, pricelist_id, product_id, qty, uom_id,
903             partner_id, date_order=date_order, fiscal_position_id=fiscal_position_id, date_planned=date_planned,
904             name=name, price_unit=price_unit, context=context)
905
906     def _get_date_planned(self, cr, uid, supplier_info, date_order_str, context=None):
907         """Return the datetime value to use as Schedule Date (``date_planned``) for
908            PO Lines that correspond to the given product.supplierinfo,
909            when ordered at `date_order_str`.
910
911            :param browse_record | False supplier_info: product.supplierinfo, used to
912                determine delivery delay (if False, default delay = 0)
913            :param str date_order_str: date of order, as a string in
914                DEFAULT_SERVER_DATE_FORMAT
915            :rtype: datetime
916            :return: desired Schedule Date for the PO line
917         """
918         supplier_delay = int(supplier_info.delay) if supplier_info else 0
919         return datetime.strptime(date_order_str, DEFAULT_SERVER_DATE_FORMAT) + relativedelta(days=supplier_delay)
920
921     def _check_product_uom_group(self, cr, uid, context=None):
922         group_uom = self.pool.get('ir.model.data').get_object(cr, uid, 'product', 'group_uom')
923         res = [user for user in group_uom.users if user.id == uid]
924         return len(res) and True or False
925
926
927     def onchange_product_id(self, cr, uid, ids, pricelist_id, product_id, qty, uom_id,
928             partner_id, date_order=False, fiscal_position_id=False, date_planned=False,
929             name=False, price_unit=False, context=None):
930         """
931         onchange handler of product_id.
932         """
933         if context is None:
934             context = {}
935
936         res = {'value': {'price_unit': price_unit or 0.0, 'name': name or '', 'product_uom' : uom_id or False}}
937         if not product_id:
938             return res
939
940         product_product = self.pool.get('product.product')
941         product_uom = self.pool.get('product.uom')
942         res_partner = self.pool.get('res.partner')
943         product_supplierinfo = self.pool.get('product.supplierinfo')
944         product_pricelist = self.pool.get('product.pricelist')
945         account_fiscal_position = self.pool.get('account.fiscal.position')
946         account_tax = self.pool.get('account.tax')
947
948         # - check for the presence of partner_id and pricelist_id
949         #if not partner_id:
950         #    raise osv.except_osv(_('No Partner!'), _('Select a partner in purchase order to choose a product.'))
951         #if not pricelist_id:
952         #    raise osv.except_osv(_('No Pricelist !'), _('Select a price list in the purchase order form before choosing a product.'))
953
954         # - determine name and notes based on product in partner lang.
955         context_partner = context.copy()
956         if partner_id:
957             lang = res_partner.browse(cr, uid, partner_id).lang
958             context_partner.update( {'lang': lang, 'partner_id': partner_id} )
959         product = product_product.browse(cr, uid, product_id, context=context_partner)
960         #call name_get() with partner in the context to eventually match name and description in the seller_ids field
961         dummy, name = product_product.name_get(cr, uid, product_id, context=context_partner)[0]
962         if product.description_purchase:
963             name += '\n' + product.description_purchase
964         res['value'].update({'name': name})
965
966         # - set a domain on product_uom
967         res['domain'] = {'product_uom': [('category_id','=',product.uom_id.category_id.id)]}
968
969         # - check that uom and product uom belong to the same category
970         product_uom_po_id = product.uom_po_id.id
971         if not uom_id:
972             uom_id = product_uom_po_id
973
974         if product.uom_id.category_id.id != product_uom.browse(cr, uid, uom_id, context=context).category_id.id:
975             if self._check_product_uom_group(cr, uid, context=context):
976                 res['warning'] = {'title': _('Warning!'), 'message': _('Selected Unit of Measure does not belong to the same category as the product Unit of Measure.')}
977             uom_id = product_uom_po_id
978
979         res['value'].update({'product_uom': uom_id})
980
981         # - determine product_qty and date_planned based on seller info
982         if not date_order:
983             date_order = fields.date.context_today(self,cr,uid,context=context)
984
985
986         supplierinfo = False
987         for supplier in product.seller_ids:
988             if partner_id and (supplier.name.id == partner_id):
989                 supplierinfo = supplier
990                 if supplierinfo.product_uom.id != uom_id:
991                     res['warning'] = {'title': _('Warning!'), 'message': _('The selected supplier only sells this product by %s') % supplierinfo.product_uom.name }
992                 min_qty = product_uom._compute_qty(cr, uid, supplierinfo.product_uom.id, supplierinfo.min_qty, to_uom_id=uom_id)
993                 if (qty or 0.0) < min_qty: # If the supplier quantity is greater than entered from user, set minimal.
994                     if qty:
995                         res['warning'] = {'title': _('Warning!'), 'message': _('The selected supplier has a minimal quantity set to %s %s, you should not purchase less.') % (supplierinfo.min_qty, supplierinfo.product_uom.name)}
996                     qty = min_qty
997         dt = self._get_date_planned(cr, uid, supplierinfo, date_order, context=context).strftime(DEFAULT_SERVER_DATETIME_FORMAT)
998         qty = qty or 1.0
999         res['value'].update({'date_planned': date_planned or dt})
1000         if qty:
1001             res['value'].update({'product_qty': qty})
1002
1003         # - determine price_unit and taxes_id
1004         if pricelist_id:
1005             price = product_pricelist.price_get(cr, uid, [pricelist_id],
1006                     product.id, qty or 1.0, partner_id or False, {'uom': uom_id, 'date': date_order})[pricelist_id]
1007         else:
1008             price = product.standard_price
1009
1010         taxes = account_tax.browse(cr, uid, map(lambda x: x.id, product.supplier_taxes_id))
1011         fpos = fiscal_position_id and account_fiscal_position.browse(cr, uid, fiscal_position_id, context=context) or False
1012         taxes_ids = account_fiscal_position.map_tax(cr, uid, fpos, taxes)
1013         res['value'].update({'price_unit': price, 'taxes_id': taxes_ids})
1014
1015         return res
1016
1017     product_id_change = onchange_product_id
1018     product_uom_change = onchange_product_uom
1019
1020     def action_confirm(self, cr, uid, ids, context=None):
1021         self.write(cr, uid, ids, {'state': 'confirmed'}, context=context)
1022         return True
1023
1024
1025 class procurement_order(osv.osv):
1026     _inherit = 'procurement.order'
1027     _columns = {
1028         'purchase_id': fields.many2one('purchase.order', 'Purchase Order'),
1029     }
1030
1031     def check_buy(self, cr, uid, ids, context=None):
1032         ''' return True if the supply method of the mto product is 'buy'
1033         '''
1034         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1035         for procurement in self.browse(cr, uid, ids, context=context):
1036             if procurement.product_id.supply_method <> 'buy':
1037                 return False
1038         return True
1039
1040     def check_supplier_info(self, cr, uid, ids, context=None):
1041         partner_obj = self.pool.get('res.partner')
1042         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
1043         for procurement in self.browse(cr, uid, ids, context=context):
1044             message = ''
1045             partner = procurement.product_id.seller_id #Taken Main Supplier of Product of Procurement.
1046
1047             if not procurement.product_id.seller_ids:
1048                 message = _('No supplier defined for this product !')
1049             elif not partner:
1050                 message = _('No default supplier defined for this product')
1051             elif not partner_obj.address_get(cr, uid, [partner.id], ['delivery'])['delivery']:
1052                 message = _('No address defined for the supplier')
1053
1054             if message:
1055                 if procurement.message != message:
1056                     cr.execute('update procurement_order set message=%s where id=%s', (message, procurement.id))
1057                 return False
1058
1059             if user.company_id and user.company_id.partner_id:
1060                 if partner.id == user.company_id.partner_id.id:
1061                     raise osv.except_osv(_('Configuration Error!'), _('The product "%s" has been defined with your company as reseller which seems to be a configuration error!' % procurement.product_id.name))
1062
1063         return True
1064
1065
1066     def action_po_assign(self, cr, uid, ids, context=None):
1067         """ This is action which call from workflow to assign purchase order to procurements
1068         @return: True
1069         """
1070         res = self.make_po(cr, uid, ids, context=context)
1071         res = res.values()
1072         return len(res) and res[0] or 0 #TO CHECK: why workflow is generated error if return not integer value
1073
1074     def create_procurement_purchase_order(self, cr, uid, procurement, po_vals, line_vals, context=None):
1075         """Create the purchase order from the procurement, using
1076            the provided field values, after adding the given purchase
1077            order line in the purchase order.
1078
1079            :params procurement: the procurement object generating the purchase order
1080            :params dict po_vals: field values for the new purchase order (the
1081                                  ``order_line`` field will be overwritten with one
1082                                  single line, as passed in ``line_vals``).
1083            :params dict line_vals: field values of the single purchase order line that
1084                                    the purchase order will contain.
1085            :return: id of the newly created purchase order
1086            :rtype: int
1087         """
1088         po_vals.update({'order_line': [(0,0,line_vals)]})
1089         return self.pool.get('purchase.order').create(cr, uid, po_vals, context=context)
1090
1091     def _get_purchase_schedule_date(self, cr, uid, procurement, company, context=None):
1092         """Return the datetime value to use as Schedule Date (``date_planned``) for the
1093            Purchase Order Lines created to satisfy the given procurement.
1094
1095            :param browse_record procurement: the procurement for which a PO will be created.
1096            :param browse_report company: the company to which the new PO will belong to.
1097            :rtype: datetime
1098            :return: the desired Schedule Date for the PO lines
1099         """
1100         procurement_date_planned = datetime.strptime(procurement.date_planned, DEFAULT_SERVER_DATETIME_FORMAT)
1101         schedule_date = (procurement_date_planned - relativedelta(days=company.po_lead))
1102         return schedule_date
1103
1104     def _get_purchase_order_date(self, cr, uid, procurement, company, schedule_date, context=None):
1105         """Return the datetime value to use as Order Date (``date_order``) for the
1106            Purchase Order created to satisfy the given procurement.
1107
1108            :param browse_record procurement: the procurement for which a PO will be created.
1109            :param browse_report company: the company to which the new PO will belong to.
1110            :param datetime schedule_date: desired Scheduled Date for the Purchase Order lines.
1111            :rtype: datetime
1112            :return: the desired Order Date for the PO
1113         """
1114         seller_delay = int(procurement.product_id.seller_delay)
1115         return schedule_date - relativedelta(days=seller_delay)
1116
1117     def make_po(self, cr, uid, ids, context=None):
1118         """ Make purchase order from procurement
1119         @return: New created Purchase Orders procurement wise
1120         """
1121         res = {}
1122         if context is None:
1123             context = {}
1124         company = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id
1125         partner_obj = self.pool.get('res.partner')
1126         uom_obj = self.pool.get('product.uom')
1127         pricelist_obj = self.pool.get('product.pricelist')
1128         prod_obj = self.pool.get('product.product')
1129         acc_pos_obj = self.pool.get('account.fiscal.position')
1130         seq_obj = self.pool.get('ir.sequence')
1131         warehouse_obj = self.pool.get('stock.warehouse')
1132         for procurement in self.browse(cr, uid, ids, context=context):
1133             res_id = procurement.move_id.id
1134             partner = procurement.product_id.seller_id # Taken Main Supplier of Product of Procurement.
1135             seller_qty = procurement.product_id.seller_qty
1136             partner_id = partner.id
1137             address_id = partner_obj.address_get(cr, uid, [partner_id], ['delivery'])['delivery']
1138             pricelist_id = partner.property_product_pricelist_purchase.id
1139             warehouse_id = warehouse_obj.search(cr, uid, [('company_id', '=', procurement.company_id.id or company.id)], context=context)
1140             uom_id = procurement.product_id.uom_po_id.id
1141
1142             qty = uom_obj._compute_qty(cr, uid, procurement.product_uom.id, procurement.product_qty, uom_id)
1143             if seller_qty:
1144                 qty = max(qty,seller_qty)
1145
1146             price = pricelist_obj.price_get(cr, uid, [pricelist_id], procurement.product_id.id, qty, partner_id, {'uom': uom_id})[pricelist_id]
1147
1148             schedule_date = self._get_purchase_schedule_date(cr, uid, procurement, company, context=context)
1149             purchase_date = self._get_purchase_order_date(cr, uid, procurement, company, schedule_date, context=context)
1150
1151             #Passing partner_id to context for purchase order line integrity of Line name
1152             new_context = context.copy()
1153             new_context.update({'lang': partner.lang, 'partner_id': partner_id})
1154
1155             product = prod_obj.browse(cr, uid, procurement.product_id.id, context=new_context)
1156             taxes_ids = procurement.product_id.supplier_taxes_id
1157             taxes = acc_pos_obj.map_tax(cr, uid, partner.property_account_position, taxes_ids)
1158
1159             name = product.partner_ref
1160             if product.description_purchase:
1161                 name += '\n'+ product.description_purchase
1162             line_vals = {
1163                 'name': name,
1164                 'product_qty': qty,
1165                 'product_id': procurement.product_id.id,
1166                 'product_uom': uom_id,
1167                 'price_unit': price or 0.0,
1168                 'date_planned': schedule_date.strftime(DEFAULT_SERVER_DATETIME_FORMAT),
1169                 'move_dest_id': res_id,
1170                 'taxes_id': [(6,0,taxes)],
1171             }
1172             name = seq_obj.get(cr, uid, 'purchase.order') or _('PO: %s') % procurement.name
1173             po_vals = {
1174                 'name': name,
1175                 'origin': procurement.origin,
1176                 'partner_id': partner_id,
1177                 'location_id': procurement.location_id.id,
1178                 'warehouse_id': warehouse_id and warehouse_id[0] or False,
1179                 'pricelist_id': pricelist_id,
1180                 'date_order': purchase_date.strftime(DEFAULT_SERVER_DATETIME_FORMAT),
1181                 'company_id': procurement.company_id.id,
1182                 'fiscal_position': partner.property_account_position and partner.property_account_position.id or False,
1183                 'payment_term_id': partner.property_supplier_payment_term.id or False,
1184             }
1185             res[procurement.id] = self.create_procurement_purchase_order(cr, uid, procurement, po_vals, line_vals, context=new_context)
1186             self.write(cr, uid, [procurement.id], {'state': 'running', 'purchase_id': res[procurement.id]})
1187         self.message_post(cr, uid, ids, body=_("Draft Purchase Order created"), context=context)
1188         return res
1189
1190     def _product_virtual_get(self, cr, uid, order_point):
1191         procurement = order_point.procurement_id
1192         if procurement and procurement.state != 'exception' and procurement.purchase_id and procurement.purchase_id.state in ('draft', 'confirmed'):
1193             return None
1194         return super(procurement_order, self)._product_virtual_get(cr, uid, order_point)
1195
1196
1197 class mail_mail(osv.Model):
1198     _name = 'mail.mail'
1199     _inherit = 'mail.mail'
1200
1201     def _postprocess_sent_message(self, cr, uid, mail, context=None):
1202         if mail.model == 'purchase.order':
1203             self.pool.get('purchase.order').signal_send_rfq(cr, uid, [mail.res_id])
1204         return super(mail_mail, self)._postprocess_sent_message(cr, uid, mail=mail, context=context)
1205
1206
1207 class product_template(osv.Model):
1208     _name = 'product.template'
1209     _inherit = 'product.template'
1210     _columns = {
1211         'purchase_ok': fields.boolean('Can be Purchased', help="Specify if the product can be selected in a purchase order line."),
1212     }
1213     _defaults = {
1214         'purchase_ok': 1,
1215     }
1216
1217
1218 class mail_compose_message(osv.Model):
1219     _inherit = 'mail.compose.message'
1220
1221     def send_mail(self, cr, uid, ids, context=None):
1222         context = context or {}
1223         if context.get('default_model') == 'purchase.order' and context.get('default_res_id'):
1224             context = dict(context, mail_post_autofollow=True)
1225             self.pool.get('purchase.order').signal_send_rfq(cr, uid, [context['default_res_id']])
1226         return super(mail_compose_message, self).send_mail(cr, uid, ids, context=context)
1227
1228
1229 class account_invoice(osv.Model):
1230     """ Override account_invoice to add Chatter messages on the related purchase
1231         orders, logging the invoice reception or payment. """
1232     _inherit = 'account.invoice'
1233
1234     def invoice_validate(self, cr, uid, ids, context=None):
1235         res = super(account_invoice, self).invoice_validate(cr, uid, ids, context=context)
1236         purchase_order_obj = self.pool.get('purchase.order')
1237         po_ids = purchase_order_obj.search(cr, uid, [('invoice_ids', 'in', ids)], context=context)
1238         for po_id in po_ids:
1239             purchase_order_obj.message_post(cr, uid, po_id, body=_("Invoice received"), context=context)
1240         return res
1241
1242     def confirm_paid(self, cr, uid, ids, context=None):
1243         res = super(account_invoice, self).confirm_paid(cr, uid, ids, context=context)
1244         purchase_order_obj = self.pool.get('purchase.order')
1245         po_ids = purchase_order_obj.search(cr, uid, [('invoice_ids', 'in', ids)], context=context)
1246         if po_ids:
1247             purchase_order_obj.message_post(cr, uid, po_ids, body=_("Invoice paid"), context=context)
1248         return res
1249
1250 class account_invoice_line(osv.Model):
1251     """ Override account_invoice_line to add the link to the purchase order line it is related to"""
1252     _inherit = 'account.invoice.line'
1253     _columns = {
1254         'purchase_line_id': fields.many2one('purchase.order.line',
1255             'Purchase Order Line', ondelete='set null', select=True,
1256             readonly=True),
1257     }
1258
1259 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: