addons/project_mrp/project_mrp.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 from osv import fields, osv
  23 import netsvc
  24
  25 class project_task(osv.osv):
  26     _name = "project.task"
  27     _inherit = "project.task"
  28     _columns = {
  29         'procurement_id': fields.many2one('procurement.order', 'Procurement', ondelete='set null'),
  30         'sale_line_id': fields.related('procurement_id', 'sale_line_id', type='many2one', relation='sale.order.line', store=True, string='Sale Order Line'),
  31     }
  32
  33     def _validate_subflows(self, cr, uid, ids):
  34         wf_service = netsvc.LocalService("workflow")
  35         for task in self.browse(cr, uid, ids):
  36             if task.procurement_id:
  37                 wf_service.trg_write(uid, 'procurement.order', task.procurement_id.id, cr)
  38
  39     def do_close(self, cr, uid, ids, *args, **kwargs):
  40         res = super(project_task, self).do_close(cr, uid, ids, *args, **kwargs)
  41         self._validate_subflows(cr, uid, ids)
  42         return res
  43
  44     def do_cancel(self, cr, uid, ids, *args, **kwargs):
  45         res = super(project_task, self).do_cancel(cr, uid, ids, *args, **kwargs)
  46         self._validate_subflows(cr, uid, ids)
  47         return res
  48 project_task()
  49
  50 class product_product(osv.osv):
  51     _inherit = "product.product"
  52     _columns = {
  53         'project_id': fields.many2one('project.project', 'Project', ondelete='set null',)
  54     }
  55 product_product()
  56
  57 class sale_order(osv.osv):
  58     _inherit ='sale.order'
  59
  60     def _prepare_order_line_procurement(self, cr, uid, order, line, move_id, date_planned, context=None):
  61         proc_data = super(sale_order, self)._prepare_order_line_procurement(cr,
  62                 uid, order, line, move_id, date_planned, context=context)
  63         proc_data['sale_line_id'] = line.id
  64         return proc_data
  65
  66     def _picked_rate(self, cr, uid, ids, name, arg, context=None):
  67         if not ids:
  68             return {}
  69         res_sale = {}
  70         res = super(sale_order, self)._picked_rate(cr, uid, ids, name, arg, context=context)
  71         cr.execute('''select sol.order_id as sale_id, t.state as task_state ,
  72                     t.id as task_id, sum(sol.product_uom_qty) as total
  73                     from project_task as t
  74                     left join sale_order_line as sol on sol.id = t.sale_line_id
  75                     where sol.order_id in %s group by sol.order_id,t.state,t.id ''',(tuple(ids),))
  76         sale_task_data = cr.dictfetchall()
  77
  78         if not sale_task_data:
  79             return res
  80
  81         for id in ids:
  82             res_sale[id] = {
  83                 'number_of_done': 0,
  84                 'total_no_task': 0,
  85             }
  86         #compute the sum of quantity for each SO
  87         cr.execute('''select sol.order_id as sale_id, sum(sol.product_uom_qty) as total
  88                     from sale_order_line sol where sol.order_id in %s group by sol.order_id''',(tuple(ids),))
  89         total_qtty_ref = cr.dictfetchall()
  90         for item in total_qtty_ref:
  91             res_sale[item['sale_id']]['number_of_stockable'] = item['total']
  92
  93         for item in sale_task_data:
  94             res_sale[item['sale_id']]['total_no_task'] += item['total']
  95             if item['task_state'] == 'done':
  96                 res_sale[item['sale_id']]['number_of_done'] += item['total']
  97
  98         for sale in self.browse(cr, uid, ids, context=context):
  99             if 'number_of_stockable' in res_sale[sale.id]:
 100                 res_sale[sale.id]['number_of_stockable'] -= res_sale[sale.id]['total_no_task']
 101                 #adjust previously percentage because now we must also count the product of type service
 102                 res[sale.id] = res[sale.id] * float(res_sale[sale.id]['number_of_stockable']) / (res_sale[sale.id]['number_of_stockable'] + res_sale[sale.id]['total_no_task'])
 103                 #add the task
 104                 res[sale.id] += res_sale[sale.id]['number_of_done'] * 100 /  (res_sale[sale.id]['number_of_stockable'] + res_sale[sale.id]['total_no_task'])
 105         return res
 106
 107     _columns = {
 108         'picked_rate': fields.function(_picked_rate, method=True, string='Picked', type='float'),
 109     }
 110
 111 sale_order()
 112 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: