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