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 osv, fields
23 from osv.orm import intersect, except_orm
25 from tools.translate import _
26 from decimal_precision import decimal_precision as dp
29 class account_analytic_account(osv.osv):
30 _name = "account.analytic.account"
31 _inherit = "account.analytic.account"
33 def _get_total_estimation(self, account):
34 tot_est = super(account_analytic_account, self)._get_total_estimation(account)
35 if account.charge_expenses:
36 tot_est += account.est_expenses
39 def _get_total_invoiced(self, account):
40 total_invoiced = super(account_analytic_account, self)._get_total_invoiced(account)
41 if account.charge_expenses:
42 total_invoiced += account.expense_invoiced
45 def _get_total_remaining(self, account):
46 total_remaining = super(account_analytic_account, self)._get_total_remaining(account)
47 if account.charge_expenses:
48 total_remaining += account.remaining_expense
49 return total_remaining
51 def _get_total_remaining(self, account):
52 total_toinvoice = super(account_analytic_account, self)._get_total_toinvoice(account)
53 if account.charge_expenses:
54 total_toinvoice += account.expense_to_invoice
55 return total_toinvoice
57 def _remaining_expnse_calc(self, cr, uid, ids, name, arg, context=None):
59 for account in self.browse(cr, uid, ids, context=context):
60 if account.est_expenses != 0:
61 res[account.id] = account.est_expenses - account.expense_invoiced
65 res[id] = round(res.get(id, 0.0),2)
68 def _expense_to_invoice_calc(self, cr, uid, ids, name, arg, context=None):
71 child_ids = tuple(ids) #We don't want consolidation for each of these fields because those complex computation is resource-greedy.
78 cr.execute("SELECT hel.analytic_account, SUM(hel.unit_amount*hel.unit_quantity) \
79 FROM hr_expense_line AS hel\
80 LEFT JOIN hr_expense_expense AS he \
81 ON he.id = hel.expense_id\
82 WHERE he.state = 'invoiced' \
83 AND hel.analytic_account IN %s \
84 GROUP BY hel.analytic_account",(child_ids,))
85 for account_id, sum in cr.fetchall():
90 def _expense_invoiced_calc(self, cr, uid, ids, name, arg, context=None):
93 child_ids = tuple(ids) #We don't want consolidation for each of these fields because those complex computation is resource-greedy.
100 cr.execute("SELECT hel.analytic_account,SUM(hel.unit_amount*hel.unit_quantity)\
101 FROM hr_expense_line AS hel\
102 LEFT JOIN hr_expense_expense AS he \
103 ON he.id = hel.expense_id\
104 WHERE he.state = 'paid' \
105 AND hel.analytic_account IN %s \
106 GROUP BY hel.analytic_account",(child_ids,))
107 for account_id, sum in cr.fetchall():
108 res[account_id] = sum
113 'charge_expenses' : fields.boolean('Charge Expenses'),
114 'expense_invoiced' : fields.function(_expense_invoiced_calc, type="float"),
115 'expense_to_invoice' : fields.function(_expense_to_invoice_calc, type='float'),
116 'remaining_expense' : fields.function(_remaining_expnse_calc, type="float"),
117 'est_expenses': fields.float('Estimation of Expenses to Invoice'),
120 def on_change_template(self, cr, uid, id, template_id, context=None):
121 res = super(account_analytic_account, self).on_change_template(cr, uid, id, template_id, context=context)
122 if template_id and 'value' in res:
123 template = self.browse(cr, uid, template_id, context=context)
124 res['value']['charge_expenses'] = template.charge_expenses
125 res['value']['expense_max'] = template.expense_max
128 def open_hr_expense(self, cr, uid, ids, context=None):
129 account = self.browse(cr, uid, ids[0], context)
130 data_obj = self.pool.get('ir.model.data')
132 journal_id = data_obj.get_object(cr, uid, 'hr_timesheet', 'analytic_journal').id
135 line_ids = self.pool.get('hr.expense.line').search(cr,uid,[('analytic_account','=',account.id)])
136 id2 = data_obj._get_id(cr, uid, 'hr_expense', 'view_expenses_form')
137 id3 = data_obj._get_id(cr, uid, 'hr_expense', 'view_expenses_tree')
139 id2 = data_obj.browse(cr, uid, id2, context=context).res_id
141 id3 = data_obj.browse(cr, uid, id3, context=context).res_id
142 domain = [('line_ids','in',line_ids)]
144 'type': 'ir.actions.act_window',
145 'name': _('Expenses'),
147 'view_mode': 'tree,form',
148 'views': [(id3,'tree'),(id2,'form')],
150 'res_model': 'hr.expense.expense',
154 def hr_to_invoiced_expense(self, cr, uid, ids, context=None):
155 res = self.open_hr_expense(cr,uid,ids,context)
156 account = self.browse(cr, uid, ids[0], context)
157 line_ids = self.pool.get('hr.expense.line').search(cr,uid,[('analytic_account','=',account.id)])
158 res['domain'] = [('line_ids','in',line_ids),('state','=','invoiced')]
161 account_analytic_account()
163 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: