addons/hr_timesheet_invoice/hr_timesheet_invoice.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 # Copyright (c) 2004-2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
   5 #
   6 # $Id: account.py 1005 2005-07-25 08:41:42Z nicoe $
   7 #
   8 # WARNING: This program as such is intended to be used by professional
   9 # programmers who take the whole responsability of assessing all potential
  10 # consequences resulting from its eventual inadequacies and bugs
  11 # End users who are looking for a ready-to-use solution with commercial
  12 # garantees and support are strongly adviced to contract a Free Software
  13 # Service Company
  14 #
  15 # This program is Free Software; you can redistribute it and/or
  16 # modify it under the terms of the GNU General Public License
  17 # as published by the Free Software Foundation; either version 2
  18 # of the License, or (at your option) any later version.
  19 #
  20 # This program is distributed in the hope that it will be useful,
  21 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  22 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  23 # GNU General Public License for more details.
  24 #
  25 # You should have received a copy of the GNU General Public License
  26 # along with this program; if not, write to the Free Software
  27 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  28 #
  29 ##############################################################################
  30
  31 from osv import fields, osv
  32
  33 class hr_timesheet_invoice_factor(osv.osv):
  34         _name = "hr_timesheet_invoice.factor"
  35         _description = "Invoice rate"
  36         _columns = {
  37                 'name': fields.char('Internal name', size=128, required=True),
  38                 'customer_name': fields.char('Visible name', size=128),
  39                 'factor': fields.float('Discount (%)', required=True),
  40         }
  41         _defaults = {
  42                 'factor': lambda *a: 0.0,
  43         }
  44 hr_timesheet_invoice_factor()
  45
  46 class account_analytic_account(osv.osv):
  47         def _invoiced_calc(self, cr, uid, ids, name, arg, context={}):
  48                 res = {}
  49                 for account in self.browse(cr, uid, ids):
  50                         invoiced = {}
  51                         cr.execute('select distinct l.invoice_id from hr_analytic_timesheet h left join account_analytic_line l on (h.line_id=l.id) where account_id=1', (account.id,))
  52                         invoice_ids = filter(None, map(lambda x: x[0], cr.fetchall()))
  53                         for invoice in self.pool.get('account.invoice').browse(cr, uid, invoice_ids, context):
  54                                 res.setdefault(account.id, 0.0)
  55                                 res[account.id] += invoice.amount_untaxed
  56                 for id in ids:
  57                         res[id] = round(res.get(id, 0.0),2)
  58                 return res
  59
  60         _inherit = "account.analytic.account"
  61         _columns = {
  62                 'pricelist_id' : fields.many2one('product.pricelist', 'Sale Pricelist'),
  63                 'amount_max': fields.float('Max. Invoice Price'),
  64                 'amount_invoiced': fields.function(_invoiced_calc, method=True, string='Invoiced Amount'),
  65                 'to_invoice': fields.many2one('hr_timesheet_invoice.factor','Invoicing'),
  66         }
  67 account_analytic_account()
  68
  69 class account_analytic_line(osv.osv):
  70         _inherit = 'account.analytic.line'
  71         _columns = {
  72                 'invoice_id': fields.many2one('account.invoice', 'Invoice'),
  73                 'to_invoice': fields.many2one('hr_timesheet_invoice.factor', 'Invoicing'),
  74         }
  75 account_analytic_line()
  76
  77 class hr_analytic_timesheet(osv.osv):
  78         _inherit = "hr.analytic.timesheet"
  79         def on_change_account_id(self, cr, uid, ids, account_id):
  80                 res = super(hr_analytic_timesheet,self).on_change_account_id(cr, uid, ids, account_id)
  81                 if not account_id:
  82                         return res
  83                 res.setdefault('value',{})
  84                 st = self.pool.get('account.analytic.account').browse(cr, uid, account_id).to_invoice.id
  85                 res['value']['to_invoice'] = st or False
  86                 return res
  87 hr_analytic_timesheet()
  88
  89 class account_invoice(osv.osv):
  90         _inherit = "account.invoice"
  91
  92         def _get_analityc_lines(self, cr, uid, id):
  93                 iml = super(account_invoice, self)._get_analityc_lines(cr, uid, id)
  94
  95                 inv = self.browse(cr, uid, [id])[0]
  96                 if inv.type == 'in_invoice':
  97                         for il in iml:
  98                                 if il['account_analytic_id']:
  99                                         to_invoice = self.pool.get('account.analytic.account').read(cr, uid, [il['account_analytic_id']], ['to_invoice'])[0]['to_invoice']
 100                                         if to_invoice:
 101                                                 il['analytic_lines'][0][2]['to_invoice'] = to_invoice[0]
 102                 return iml
 103 account_invoice()