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
  34 class hr_timesheet_invoice_factor(osv.osv):
  35         _name = "hr_timesheet_invoice.factor"
  36         _description = "Invoice rate"
  37         _columns = {
  38                 'name': fields.char('Internal name', size=128, required=True),
  39                 'customer_name': fields.char('Visible name', size=128),
  40                 'factor': fields.float('Discount (%)', required=True),
  41         }
  42         _defaults = {
  43                 'factor': lambda *a: 0.0,
  44         }
  45
  46 hr_timesheet_invoice_factor()
  47
  48
  49 class account_analytic_account(osv.osv):
  50         def _invoiced_calc(self, cr, uid, ids, name, arg, context={}):
  51                 res = {}
  52                 for account in self.browse(cr, uid, ids):
  53                         invoiced = {}
  54                         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,))
  55                         invoice_ids = filter(None, map(lambda x: x[0], cr.fetchall()))
  56                         for invoice in self.pool.get('account.invoice').browse(cr, uid, invoice_ids, context):
  57                                 res.setdefault(account.id, 0.0)
  58                                 res[account.id] += invoice.amount_untaxed
  59                 for id in ids:
  60                         res[id] = round(res.get(id, 0.0),2)
  61                 return res
  62
  63         _inherit = "account.analytic.account"
  64         _columns = {
  65                 'pricelist_id' : fields.many2one('product.pricelist', 'Sale Pricelist'),
  66                 'amount_max': fields.float('Max. Invoice Price'),
  67                 'amount_invoiced': fields.function(_invoiced_calc, method=True, string='Invoiced Amount'),
  68                 'to_invoice': fields.many2one('hr_timesheet_invoice.factor','Invoicing'),
  69         }
  70 account_analytic_account()
  71
  72
  73 class account_analytic_line(osv.osv):
  74         _inherit = 'account.analytic.line'
  75         _columns = {
  76                 'invoice_id': fields.many2one('account.invoice', 'Invoice'),
  77                 'to_invoice': fields.many2one('hr_timesheet_invoice.factor', 'Invoicing'),
  78         }
  79
  80         def unlink(self, cursor, user, ids, context=None):
  81                 self._check(cursor, user, ids)
  82                 return super(account_analytic_line,self).unlink(cursor, uid, ids,
  83                                 context=context)
  84
  85         def write(self, cr, uid, ids, vals, context=None):
  86                 self._check(cr, uid, ids)
  87                 return super(account_analytic_line,self).write(cr, uid, ids, vals,
  88                                 context=context)
  89
  90         def _check(self, cr, uid, ids):
  91                 select = ids
  92                 if isinstance(select, (int, long)):
  93                         select = [ids]
  94                 for line in self.browse(cr, uid, select):
  95                         if line.invoice_id:
  96                                 raise osv.except_osv('Error !',
  97                                                 'You can not modify an invoiced analytic line!')
  98                 return True
  99
 100 account_analytic_line()
 101
 102
 103 class hr_analytic_timesheet(osv.osv):
 104         _inherit = "hr.analytic.timesheet"
 105         def on_change_account_id(self, cr, uid, ids, account_id):
 106                 res = {}
 107                 if not account_id:
 108                         return res
 109                 res.setdefault('value',{})
 110                 st = self.pool.get('account.analytic.account').browse(cr, uid, account_id).to_invoice.id
 111                 res['value']['to_invoice'] = st or False
 112                 return res
 113 hr_analytic_timesheet()
 114
 115 class account_invoice(osv.osv):
 116         _inherit = "account.invoice"
 117
 118         def _get_analityc_lines(self, cr, uid, id):
 119                 iml = super(account_invoice, self)._get_analityc_lines(cr, uid, id)
 120
 121                 inv = self.browse(cr, uid, [id])[0]
 122                 if inv.type == 'in_invoice':
 123                         for il in iml:
 124                                 if il['account_analytic_id']:
 125                                         to_invoice = self.pool.get('account.analytic.account').read(cr, uid, [il['account_analytic_id']], ['to_invoice'])[0]['to_invoice']
 126                                         if to_invoice:
 127                                                 il['analytic_lines'][0][2]['to_invoice'] = to_invoice[0]
 128                 return iml
 129
 130 account_invoice()