1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
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.
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.
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/>.
20 ##############################################################################
22 from osv import fields,osv
24 from tools.translate import _
26 class analytic_user_funct_grid(osv.osv):
28 _name="analytic_user_funct_grid"
29 _description= "Relation table between users and products on a analytic account"
31 'user_id': fields.many2one("res.users", "User", required=True,),
32 'product_id': fields.many2one("product.product", "Product", required=True,),
33 'account_id': fields.many2one("account.analytic.account", "Analytic Account", required=True,),
36 analytic_user_funct_grid()
39 class account_analytic_account(osv.osv):
41 _inherit = "account.analytic.account"
43 'user_product_ids': fields.one2many('analytic_user_funct_grid', 'account_id', 'Users/Products Rel.'),
46 account_analytic_account()
49 class hr_analytic_timesheet(osv.osv):
51 _inherit = "hr.analytic.timesheet"
54 # Look in account, if no value for the user => look in parent until there is no more parent to look
55 # Take the first found... if nothing found => return False
56 def _get_related_user_account_recursiv(self, cr, uid, user_id, account_id):
58 temp=self.pool.get('analytic_user_funct_grid').search(cr, uid, [('user_id', '=', user_id),('account_id', '=', account_id) ])
59 account=self.pool.get('account.analytic.account').browse(cr, uid, account_id)
64 return self._get_related_user_account_recursiv(cr, uid, user_id, account.parent_id.id)
69 def on_change_account_id(self, cr, uid, ids, account_id, user_id=False, unit_amount=0):
72 #avoid a useless call to super
76 return super(hr_analytic_timesheet, self).on_change_account_id(cr, uid, ids, account_id)
78 #get the browse record related to user_id and account_id
79 temp = self._get_related_user_account_recursiv(cr, uid, user_id, account_id)
81 #if there isn't any record for this user_id and account_id
82 return super(hr_analytic_timesheet, self).on_change_account_id(cr, uid, ids, account_id)
84 #get the old values from super and add the value from the new relation analytic_user_funct_grid
85 r = self.pool.get('analytic_user_funct_grid').browse(cr, uid, temp)[0]
86 res.setdefault('value',{})
87 res['value']= super(hr_analytic_timesheet, self).on_change_account_id(cr, uid, ids, account_id)['value']
88 res['value']['product_id'] = r.product_id.id
89 res['value']['product_uom_id'] = r.product_id.product_tmpl_id.uom_id.id
91 #the change of product has to impact the amount, uom and general_account_id
92 a = r.product_id.product_tmpl_id.property_account_expense.id
94 a = r.product_id.categ_id.property_account_expense_categ.id
96 raise osv.except_osv(_('Error !'),
97 _('There is no expense account define ' \
98 'for this product: "%s" (id:%d)') % \
99 (r.product_id.name, r.product_id.id,))
100 # Compute based on pricetype
101 amount_unit = self.on_change_unit_amount(cr, uid, ids,
102 r.product_id.id, unit_amount, False, r.product_id.uom_id.id)['value']['amount']
104 amount = unit_amount * amount_unit
105 res ['value']['amount']= - round(amount, 2)
106 res ['value']['general_account_id']= a
109 def on_change_user_id(self, cr, uid, ids, user_id, account_id, unit_amount=0):
112 #avoid a useless call to super
115 #get the old values from super
116 res = super(hr_analytic_timesheet, self).on_change_user_id(cr, uid, ids, user_id)
119 #get the browse record related to user_id and account_id
120 temp = self._get_related_user_account_recursiv(cr, uid, user_id, account_id)
122 #add the value from the new relation analytic_user_funct_grid
123 r = self.pool.get('analytic_user_funct_grid').browse(cr, uid, temp)[0]
124 res['value']['product_id'] = r.product_id.id
126 #the change of product has to impact the amount, uom and general_account_id
127 a = r.product_id.product_tmpl_id.property_account_expense.id
129 a = r.product_id.categ_id.property_account_expense_categ.id
131 raise osv.except_osv(_('Error !'),
132 _('There is no expense account define ' \
133 'for this product: "%s" (id:%d)') % \
134 (r.product_id.name, r.product_id.id,))
135 # Compute based on pricetype
136 amount_unit = self.on_change_unit_amount(cr, uid, ids,
137 r.product_id.id, unit_amount, False, r.product_id.uom_id.id)['value']['amount']
139 amount = unit_amount * amount_unit
140 res ['value']['amount']= - round(amount, 2)
141 res ['value']['general_account_id']= a
144 hr_analytic_timesheet()