[IMP] project_mrp: Improved code. Fixed computation problem.
[odoo/odoo.git] / 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_id': fields.many2one('sale.order','Sale Order')
31     }
32
33     def _validate_subflows(self, cr, uid, ids):
34         for task in self.browse(cr, uid, ids):
35             if task.procurement_id:
36                 wf_service = netsvc.LocalService("workflow")
37                 wf_service.trg_validate(uid, 'procurement.order', task.procurement_id.id, 'subflow.done', 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     
49 project_task()
50
51 class product_product(osv.osv):
52     _inherit = "product.product"
53     _columns = {
54         'project_id': fields.many2one('project.project', 'Project', ondelete='set null',)
55     }
56 product_product()
57
58 class sale_order(osv.osv):
59     _inherit ='sale.order'
60
61     def _picked_rate(self, cr, uid, ids, name, arg, context=None):
62         if not ids:
63             return {}
64         res_sale = {}
65         res = super(sale_order, self)._picked_rate(cr, uid, ids, name, arg, context=context)
66         cr.execute('''select so.id as sale_id, t.state as task_state ,
67                     t.id as task_id, count(t.id) as total
68                     from project_task as t
69                     left join sale_order as so on so.id = t.sale_id
70                     where so.id in %s group by so.id,t.state,t.id ''',(tuple(ids),))
71         sale_task_data = cr.dictfetchall()
72
73         if not sale_task_data:
74             return res
75         
76         for id in ids:
77             res_sale[id] = {
78                 'number_of_done': 0,
79                 'percentage': 0.0,
80                 'number_of_stockable': 0.0,
81                 'total_no_task': 0,
82                 'total': 0
83             }
84
85         for item in sale_task_data:
86             res_sale[item['sale_id']]['total_no_task'] += item['total']
87             if item['task_state'] == 'done':
88                 res_sale[item['sale_id']]['number_of_done'] += 1
89             else: 
90                 pass
91         for sale in self.browse(cr, uid, ids, context=context):
92             # Percent of service + other' Type product
93             res_sale[sale.id]['percentage'] = res_sale[sale.id]['total_no_task'] and (float(res_sale[sale.id]['number_of_done']) / res_sale[sale.id]['total_no_task']) * 100
94             res_sale[sale.id]['number_of_stockable'] = len(sale.order_line) - res_sale[sale.id]['total_no_task']
95             if res_sale[sale.id]['percentage'] == 100 and res[sale.id] == 100 or res_sale[sale.id]['total_no_task'] == 0:
96                 continue
97             elif res_sale[sale.id]['number_of_stockable'] == 0:
98                 res[sale.id] = (res_sale[sale.id]['percentage'])
99             else:
100                 res[sale.id] = round((res[sale.id] + res_sale[sale.id]['percentage']) / (res_sale[sale.id]['total_no_task']), 2)
101                 if res[sale.id] > 100:
102                     res[sale.id] = 100
103         return res
104
105     _columns = {
106         'picked_rate': fields.function(_picked_rate, method=True, string='Picked', type='float'),
107     }
108
109 sale_order()
110 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: