addons/hr_expense/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 import time
  23
  24 from osv import fields, osv
  25 from tools.translate import _
  26
  27 def _employee_get(obj, cr, uid, context=None):
  28     ids = obj.pool.get('hr.employee').search(cr, uid, [('user_id','=', uid)])
  29     if ids:
  30         return ids[0]
  31     return False
  32
  33 class hr_expense_expense(osv.osv):
  34
  35     def copy(self, cr, uid, id, default=None, context=None):
  36         if not default: default = {}
  37         default.update( {'invoice_id':False,'date_confirm':False,'date_valid':False,'user_valid':False})
  38         return super(hr_expense_expense, self).copy(cr, uid, id, default, context)
  39
  40     def _amount(self, cr, uid, ids, field_name, arg, context):
  41         cr.execute("SELECT s.id,COALESCE(SUM(l.unit_amount*l.unit_quantity),0) AS amount FROM hr_expense_expense s LEFT OUTER JOIN hr_expense_line l ON (s.id=l.expense_id) WHERE s.id =ANY(%s) GROUP BY s.id ",(ids,))
  42         res = dict(cr.fetchall())
  43         return res
  44
  45     def _get_currency(self, cr, uid, context):
  46         user = self.pool.get('res.users').browse(cr, uid, [uid])[0]
  47         if user.company_id:
  48             return user.company_id.currency_id.id
  49         else:
  50             return self.pool.get('res.currency').search(cr, uid, [('rate','=',1.0)])[0]
  51
  52     _name = "hr.expense.expense"
  53     _description = "HR Expense"
  54     _columns = {
  55         'name': fields.char('Expense Sheet', size=128, required=True),
  56         'id': fields.integer('Sheet ID', readonly=True),
  57         'ref': fields.char('Reference', size=32),
  58         'date': fields.date('Date'),
  59         'journal_id': fields.many2one('account.journal', 'Force Journal', help = "The journal used when the expense is invoiced"),
  60         'employee_id': fields.many2one('hr.employee', "Employee's Name", required=True),
  61         'user_id': fields.many2one('res.users', 'User', required=True),
  62         'date_confirm': fields.date('Confirmation Date', help = "Date of the confirmation of the sheet expense. It's filled when the button Confirm is pressed."),
  63         'date_valid': fields.date('Validation Date', help = "Date of the acceptation of the sheet expense. It's filled when the button Accept is pressed."),
  64         'user_valid': fields.many2one('res.users', 'Validation User'),
  65         'account_move_id': fields.many2one('account.move', 'Ledger Posting'),
  66         'line_ids': fields.one2many('hr.expense.line', 'expense_id', 'Expense Lines', readonly=True, states={'draft':[('readonly',False)]} ),
  67         'note': fields.text('Note'),
  68         'amount': fields.function(_amount, method=True, string='Total Amount'),
  69         'invoice_id': fields.many2one('account.invoice', "Employee's Invoice"),
  70         'currency_id': fields.many2one('res.currency', 'Currency', required=True),
  71         'department_id':fields.many2one('hr.department','Department'),
  72         'company_id': fields.many2one('res.company', 'Company', required=True),
  73         'state': fields.selection([
  74             ('draft', 'Draft'),
  75             ('confirm', 'Waiting confirmation'),
  76             ('accepted', 'Accepted'),
  77             ('invoiced', 'Invoiced'),
  78             ('paid', 'Reimbursed'),
  79             ('cancelled', 'Cancelled')],
  80             'State', readonly=True, help='When the expense request is created the state is \'Draft\'.\n It is confirmed by the user and request is sent to admin, the state is \'Waiting Confirmation\'.\
  81             \nIf the admin accepts it, the state is \'Accepted\'.\n If an invoice is made for the expense request, the state is \'Invoiced\'.\n If the expense is paid to user, the state is \'Reimbursed\'.'),
  82     }
  83     _defaults = {
  84         'date' : lambda *a: time.strftime('%Y-%m-%d'),
  85         'state': lambda *a: 'draft',
  86         'employee_id' : _employee_get,
  87         'user_id' : lambda cr, uid, id, c={}: id,
  88         'currency_id': _get_currency,
  89         'company_id': lambda self, cr, uid, c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.id,
  90     }
  91     def expense_confirm(self, cr, uid, ids, *args):
  92         #for exp in self.browse(cr, uid, ids):
  93         self.write(cr, uid, ids, {
  94             'state':'confirm',
  95             'date_confirm': time.strftime('%Y-%m-%d')
  96         })
  97         return True
  98
  99     def expense_accept(self, cr, uid, ids, *args):
 100         self.write(cr, uid, ids, {
 101             'state':'accepted',
 102             'date_valid':time.strftime('%Y-%m-%d'),
 103             'user_valid': uid,
 104             })
 105         return True
 106
 107     def expense_canceled(self, cr, uid, ids, *args):
 108         self.write(cr, uid, ids, {'state':'cancelled'})
 109         return True
 110
 111     def expense_paid(self, cr, uid, ids, *args):
 112         self.write(cr, uid, ids, {'state':'paid'})
 113         return True
 114
 115     def action_invoice_create(self, cr, uid, ids):
 116         res = False
 117         invoice_obj = self.pool.get('account.invoice')
 118         for exp in self.browse(cr, uid, ids):
 119             lines = []
 120             for l in exp.line_ids:
 121                 tax_id = []
 122                 if l.product_id:
 123                     acc = l.product_id.product_tmpl_id.property_account_expense.id
 124                     if not acc:
 125                         acc = l.product_id.categ_id.property_account_expense_categ.id
 126                     tax_id = [x.id for x in l.product_id.supplier_taxes_id]
 127                 else:
 128                     acc = self.pool.get('ir.property').get(cr, uid, 'property_account_expense_categ', 'product.category')
 129                     if not acc:
 130                         raise osv.except_osv(_('Error !'), _('Please configure Default Expanse account for Product purchase, `property_account_expense_categ`'))
 131
 132                 lines.append((0, False, {
 133                     'name': l.name,
 134                     'account_id': acc.id,
 135                     'price_unit': l.unit_amount,
 136                     'quantity': l.unit_quantity,
 137                     'uos_id': l.uom_id.id,
 138                     'product_id': l.product_id and l.product_id.id or False,
 139                     'invoice_line_tax_id': tax_id and [(6, 0, tax_id)] or False,
 140                     'account_analytic_id': l.analytic_account.id,
 141                 }))
 142             if not exp.employee_id.address_id:
 143                 raise osv.except_osv(_('Error !'), _('The employee must have a working address'))
 144             acc = exp.employee_id.address_id.partner_id.property_account_payable.id
 145             payment_term_id = exp.employee_id.address_id.partner_id.property_payment_term.id
 146             inv = {
 147                 'name': exp.name,
 148                 'reference': self.pool.get('ir.sequence').get(cr, uid, 'hr.expense.invoice'),
 149                 'account_id': acc,
 150                 'type': 'in_invoice',
 151                 'partner_id': exp.employee_id.address_id.partner_id.id,
 152                 'address_invoice_id': exp.employee_id.address_id.id,
 153                 'address_contact_id': exp.employee_id.address_id.id,
 154                 'origin': exp.name,
 155                 'invoice_line': lines,
 156                 'price_type': 'tax_included',
 157                 'currency_id': exp.currency_id.id,
 158                 'payment_term': payment_term_id,
 159                 'fiscal_position': exp.employee_id.address_id.partner_id.property_account_position.id
 160             }
 161             if payment_term_id:
 162                 to_update = invoice_obj.onchange_payment_term_date_invoice(cr, uid, [],
 163                         payment_term_id, None)
 164                 if to_update:
 165                     inv.update(to_update['value'])
 166             if exp.journal_id:
 167                 inv['journal_id']=exp.journal_id.id
 168             inv_id = invoice_obj.create(cr, uid, inv, {'type':'in_invoice'})
 169             invoice_obj.button_compute(cr, uid, [inv_id], {'type':'in_invoice'},
 170                     set_total=True)
 171             self.write(cr, uid, [exp.id], {'invoice_id': inv_id, 'state': 'invoiced'})
 172             res = inv_id
 173         return res
 174 hr_expense_expense()
 175
 176 class product_product(osv.osv):
 177     _inherit = "product.product"
 178     _columns = {
 179         'hr_expense_ok': fields.boolean('Can Constitute an Expense', help="Determines if the product can be visible in the list of product within a selection from an HR expense sheet line."),
 180     }
 181
 182 product_product()
 183
 184 class hr_expense_line(osv.osv):
 185     _name = "hr.expense.line"
 186     _description = "Expense Line"
 187
 188     def _amount(self, cr, uid, ids, field_name, arg, context):
 189         if not len(ids):
 190             return {}
 191         cr.execute("SELECT l.id,COALESCE(SUM(l.unit_amount*l.unit_quantity),0) AS amount FROM hr_expense_line l WHERE id =ANY(%s) GROUP BY l.id ",(ids,))
 192         res = dict(cr.fetchall())
 193         return res
 194
 195     _columns = {
 196         'name': fields.char('Short Description', size=128, required=True),
 197         'date_value': fields.date('Date', required=True),
 198         'expense_id': fields.many2one('hr.expense.expense', 'Expense', ondelete='cascade', select=True),
 199         'total_amount': fields.function(_amount, method=True, string='Total'),
 200         'unit_amount': fields.float('Unit Price'),
 201         'unit_quantity': fields.float('Quantities' ),
 202         'product_id': fields.many2one('product.product', 'Product', domain=[('hr_expense_ok','=',True)]),
 203         'uom_id': fields.many2one('product.uom', 'UoM' ),
 204         'description': fields.text('Description'),
 205         'analytic_account': fields.many2one('account.analytic.account','Analytic account'),
 206         'ref': fields.char('Reference', size=32),
 207         'sequence' : fields.integer('Sequence', help="Gives the sequence order when displaying a list of expense lines."),
 208     }
 209     _defaults = {
 210         'unit_quantity': lambda *a: 1,
 211         'date_value' : lambda *a: time.strftime('%Y-%m-%d'),
 212     }
 213     _order = "sequence"
 214
 215     def onchange_product_id(self, cr, uid, ids, product_id, uom_id, employee_id, context=None):
 216         if context is None:
 217             context = {}
 218         v = {}
 219         if product_id:
 220             product=self.pool.get('product.product').browse(cr,uid,product_id, context=context)
 221             v['name']=product.name
 222
 223             # Compute based on pricetype of employee company
 224             pricetype_id = self.pool.get('hr.employee').browse(cr,uid,employee_id).user_id.company_id.property_valuation_price_type.id
 225             context['currency_id']=self.pool.get('hr.employee').browse(cr,uid,employee_id).user_id.company_id.currency_id.id
 226             pricetype=self.pool.get('product.price.type').browse(cr,uid,pricetype_id)
 227             amount_unit=product.price_get(pricetype.field, context)[product.id]
 228
 229             v['unit_amount']=amount_unit
 230             if not uom_id:
 231                 v['uom_id']=product.uom_id.id
 232         return {'value': v}
 233
 234 hr_expense_line()
 235
 236 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: