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