addons/analytic_contract_hr_expense/analytic_contract_hr_expense.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 osv, fields
  23 from osv.orm import intersect, except_orm
  24 import tools.sql
  25 from tools.translate import _
  26 from decimal_precision import decimal_precision as dp
  27
  28
  29 class account_analytic_account(osv.osv):
  30     _name = "account.analytic.account"
  31     _inherit = "account.analytic.account"
  32
  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
  37         return tot_est
  38
  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
  43         return total_invoiced
  44
  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
  50
  51     def _get_total_toinvoice(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
  56
  57     def _remaining_expnse_calc(self, cr, uid, ids, name, arg, context=None):
  58         res = {}
  59         for account in self.browse(cr, uid, ids, context=context):
  60             if account.est_expenses != 0:
  61                 res[account.id] = max(account.est_expenses - account.expense_invoiced, account.expense_to_invoice)
  62             else:
  63                 res[account.id]=0.0
  64         return res
  65
  66     def _expense_to_invoice_calc(self, cr, uid, ids, name, arg, context=None):
  67         res = {}
  68         res_final = {}
  69         child_ids = tuple(ids) #We don't want consolidation for each of these fields because those complex computation is resource-greedy.
  70         for i in child_ids:
  71             res[i] =  0.0
  72         if not child_ids:
  73             return res
  74
  75         if child_ids:
  76             cr.execute("SELECT hel.analytic_account, SUM(hel.unit_amount*hel.unit_quantity) \
  77                     FROM hr_expense_line AS hel\
  78                     LEFT JOIN hr_expense_expense AS he \
  79                         ON he.id = hel.expense_id\
  80                     WHERE he.state = 'invoiced' \
  81                         AND hel.analytic_account IN %s \
  82                     GROUP BY hel.analytic_account",(child_ids,))
  83             for account_id, sum in cr.fetchall():
  84                 res[account_id] = sum
  85         res_final = res
  86         return res_final
  87
  88     def _expense_invoiced_calc(self, cr, uid, ids, name, arg, context=None):
  89         res = {}
  90         res_final = {}
  91         child_ids = tuple(ids) #We don't want consolidation for each of these fields because those complex computation is resource-greedy.
  92         for i in child_ids:
  93             res[i] =  0.0
  94         if not child_ids:
  95             return res
  96
  97         if child_ids:
  98             cr.execute("SELECT hel.analytic_account,SUM(hel.unit_amount*hel.unit_quantity)\
  99                     FROM hr_expense_line AS hel\
 100                     LEFT JOIN hr_expense_expense AS he \
 101                         ON he.id = hel.expense_id\
 102                     WHERE he.state = 'paid' \
 103                          AND hel.analytic_account IN %s \
 104                     GROUP BY hel.analytic_account",(child_ids,))
 105             for account_id, sum in cr.fetchall():
 106                 res[account_id] = sum
 107         res_final = res
 108         return res_final
 109
 110     _columns = {
 111         'charge_expenses' : fields.boolean('Charge Expenses'),
 112         'expense_invoiced' : fields.function(_expense_invoiced_calc, type="float"),
 113         'expense_to_invoice' : fields.function(_expense_to_invoice_calc, type='float'),
 114         'remaining_expense' : fields.function(_remaining_expnse_calc, type="float"),
 115         'est_expenses': fields.float('Estimation of Expenses to Invoice'),
 116     }
 117
 118     def on_change_template(self, cr, uid, id, template_id, context=None):
 119         res = super(account_analytic_account, self).on_change_template(cr, uid, id, template_id, context=context)
 120         if template_id and 'value' in res:
 121             template = self.browse(cr, uid, template_id, context=context)
 122             res['value']['charge_expenses'] = template.charge_expenses
 123             res['value']['expense_max'] = template.expense_max
 124         return res
 125
 126     def open_hr_expense(self, cr, uid, ids, context=None):
 127         account = self.browse(cr, uid, ids[0], context)
 128         data_obj = self.pool.get('ir.model.data')
 129         try:
 130             journal_id = data_obj.get_object(cr, uid, 'hr_timesheet', 'analytic_journal').id
 131         except ValueError:
 132             journal_id = False
 133         line_ids = self.pool.get('hr.expense.line').search(cr,uid,[('analytic_account','=',account.id)])
 134         id2 = data_obj._get_id(cr, uid, 'hr_expense', 'view_expenses_form')
 135         id3 = data_obj._get_id(cr, uid, 'hr_expense', 'view_expenses_tree')
 136         if id2:
 137             id2 = data_obj.browse(cr, uid, id2, context=context).res_id
 138         if id3:
 139             id3 = data_obj.browse(cr, uid, id3, context=context).res_id
 140         domain = [('line_ids','in',line_ids)]
 141         return {
 142             'type': 'ir.actions.act_window',
 143             'name': _('Expenses'),
 144             'view_type': 'form',
 145             'view_mode': 'tree,form',
 146             'views': [(id3,'tree'),(id2,'form')],
 147             'domain' : domain,
 148             'res_model': 'hr.expense.expense',
 149             'nodestroy': True,
 150         }
 151
 152     def hr_to_invoiced_expense(self, cr, uid, ids, context=None):
 153          res = self.open_hr_expense(cr,uid,ids,context)
 154          account = self.browse(cr, uid, ids[0], context)
 155          line_ids = self.pool.get('hr.expense.line').search(cr,uid,[('analytic_account','=',account.id)])
 156          res['domain'] = [('line_ids','in',line_ids),('state','=','invoiced')]
 157          return res
 158
 159 account_analytic_account()
 160
 161 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: