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 openerp import netsvc
  25 from openerp.osv import fields, osv
  26 from openerp.tools.translate import _
  27
  28 import openerp.addons.decimal_precision as dp
  29
  30 def _employee_get(obj, cr, uid, context=None):
  31     if context is None:
  32         context = {}
  33     ids = obj.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
  34     if ids:
  35         return ids[0]
  36     return False
  37
  38 class hr_expense_expense(osv.osv):
  39
  40     def copy(self, cr, uid, id, default=None, context=None):
  41         if context is None:
  42             context = {}
  43         if not default: default = {}
  44         default.update({'voucher_id': False, 'date_confirm': False, 'date_valid': False, 'user_valid': False})
  45         return super(hr_expense_expense, self).copy(cr, uid, id, default, context=context)
  46
  47     def _amount(self, cr, uid, ids, field_name, arg, context=None):
  48         res= {}
  49         for expense in self.browse(cr, uid, ids, context=context):
  50             total = 0.0
  51             for line in expense.line_ids:
  52                 total += line.unit_amount * line.unit_quantity
  53             res[expense.id] = total
  54         return res
  55
  56     def _get_currency(self, cr, uid, context=None):
  57         user = self.pool.get('res.users').browse(cr, uid, [uid], context=context)[0]
  58         if user.company_id:
  59             return user.company_id.currency_id.id
  60         else:
  61             return self.pool.get('res.currency').search(cr, uid, [('rate','=',1.0)], context=context)[0]
  62
  63     _name = "hr.expense.expense"
  64     _inherit = ['mail.thread']
  65     _description = "Expense"
  66     _order = "id desc"
  67     _columns = {
  68         'name': fields.char('Description', size=128, required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  69         'id': fields.integer('Sheet ID', readonly=True),
  70         'date': fields.date('Date', select=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  71         'journal_id': fields.many2one('account.journal', 'Force Journal', help = "The journal used when the expense is done."),
  72         'employee_id': fields.many2one('hr.employee', "Employee", required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  73         'user_id': fields.many2one('res.users', 'User', required=True),
  74         '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."),
  75         '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."),
  76         'user_valid': fields.many2one('res.users', 'Validation By', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  77         'account_move_id': fields.many2one('account.move', 'Ledger Posting'),
  78         'line_ids': fields.one2many('hr.expense.line', 'expense_id', 'Expense Lines', readonly=True, states={'draft':[('readonly',False)]} ),
  79         'note': fields.text('Note'),
  80         'amount': fields.function(_amount, string='Total Amount', digits_compute= dp.get_precision('Account')),
  81         'voucher_id': fields.many2one('account.voucher', "Employee's Receipt"),
  82         'currency_id': fields.many2one('res.currency', 'Currency', required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  83         'department_id':fields.many2one('hr.department','Department', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
  84         'company_id': fields.many2one('res.company', 'Company', required=True),
  85         'state': fields.selection([
  86             ('draft', 'New'),
  87             ('cancelled', 'Refused'),
  88             ('confirm', 'Waiting Approval'),
  89             ('accepted', 'Approved'),
  90             ('done', 'Done'),
  91             ],
  92             'Status', readonly=True, help='When the expense request is created the status is \'Draft\'.\n It is confirmed by the user and request is sent to admin, the status is \'Waiting Confirmation\'.\
  93             \nIf the admin accepts it, the status is \'Accepted\'.\n If a receipt is made for the expense request, the status is \'Done\'.'),
  94     }
  95     _defaults = {
  96         'company_id': lambda s, cr, uid, c: s.pool.get('res.company')._company_default_get(cr, uid, 'hr.employee', context=c),
  97         'date': fields.date.context_today,
  98         'state': 'draft',
  99         'employee_id': _employee_get,
 100         'user_id': lambda cr, uid, id, c={}: id,
 101         'currency_id': _get_currency,
 102     }
 103
 104     def unlink(self, cr, uid, ids, context=None):
 105         for rec in self.browse(cr, uid, ids, context=context):
 106             if rec.state != 'draft':
 107                 raise osv.except_osv(_('Warning!'),_('You can only delete draft expenses!'))
 108         return super(hr_expense_expense, self).unlink(cr, uid, ids, context)
 109
 110     def onchange_currency_id(self, cr, uid, ids, currency_id=False, company_id=False, context=None):
 111         res =  {'value': {'journal_id': False}}
 112         journal_ids = self.pool.get('account.journal').search(cr, uid, [('type','=','purchase'), ('currency','=',currency_id), ('company_id', '=', company_id)], context=context)
 113         if journal_ids:
 114             res['value']['journal_id'] = journal_ids[0]
 115         return res
 116
 117     def onchange_employee_id(self, cr, uid, ids, employee_id, context=None):
 118         emp_obj = self.pool.get('hr.employee')
 119         department_id = False
 120         company_id = False
 121         if employee_id:
 122             employee = emp_obj.browse(cr, uid, employee_id, context=context)
 123             department_id = employee.department_id.id
 124             company_id = employee.company_id.id
 125         return {'value': {'department_id': department_id, 'company_id': company_id}}
 126
 127     def expense_confirm(self, cr, uid, ids, *args):
 128         for expense in self.browse(cr, uid, ids):
 129             if expense.employee_id and expense.employee_id.parent_id.user_id:
 130                 self.message_subscribe_users(cr, uid, [expense.id], user_ids=[expense.employee_id.parent_id.user_id.id])
 131         self.write(cr, uid, ids, {
 132             'state':'confirm',
 133             'date_confirm': time.strftime('%Y-%m-%d')
 134         })
 135         return True
 136
 137     def expense_accept(self, cr, uid, ids, *args):
 138         self.write(cr, uid, ids, {
 139             'state':'accepted',
 140             'date_valid':time.strftime('%Y-%m-%d'),
 141             'user_valid': uid,
 142             })
 143         return True
 144
 145     def expense_canceled(self, cr, uid, ids, *args):
 146         self.write(cr, uid, ids, {'state':'cancelled'})
 147         return True
 148
 149     def action_receipt_create(self, cr, uid, ids, context=None):
 150         property_obj = self.pool.get('ir.property')
 151         sequence_obj = self.pool.get('ir.sequence')
 152         analytic_journal_obj = self.pool.get('account.analytic.journal')
 153         account_journal = self.pool.get('account.journal')
 154         voucher_obj = self.pool.get('account.voucher')
 155         currency_obj = self.pool.get('res.currency')
 156         wkf_service = netsvc.LocalService("workflow")
 157         if context is None:
 158             context = {}
 159         for exp in self.browse(cr, uid, ids, context=context):
 160             company_id = exp.company_id.id
 161             lines = []
 162             total = 0.0
 163             ctx = context.copy()
 164             ctx.update({'date': exp.date})
 165             journal = False
 166             if exp.journal_id:
 167                 journal = exp.journal_id
 168             else:
 169                 journal_id = voucher_obj._get_journal(cr, uid, context={'type': 'purchase', 'company_id': company_id})
 170                 if journal_id:
 171                     journal = account_journal.browse(cr, uid, journal_id, context=context)
 172             for line in exp.line_ids:
 173                 if line.product_id:
 174                     acc = line.product_id.product_tmpl_id.property_account_expense
 175                     if not acc:
 176                         acc = line.product_id.categ_id.property_account_expense_categ
 177                 else:
 178                     acc = property_obj.get(cr, uid, 'property_account_expense_categ', 'product.category', context={'force_company': company_id})
 179                     if not acc:
 180                         raise osv.except_osv(_('Error!'), _('Please configure Default Expense account for Product purchase: `property_account_expense_categ`.'))
 181                 total_amount = line.total_amount
 182                 if journal.currency:
 183                     if exp.currency_id != journal.currency:
 184                         total_amount = currency_obj.compute(cr, uid, exp.currency_id.id, journal.currency.id, total_amount, context=ctx)
 185                 elif exp.currency_id != exp.company_id.currency_id:
 186                     total_amount = currency_obj.compute(cr, uid, exp.currency_id.id, exp.company_id.currency_id.id, total_amount, context=ctx)
 187                 lines.append((0, False, {
 188                     'name': line.name,
 189                     'account_id': acc.id,
 190                     'account_analytic_id': line.analytic_account.id,
 191                     'amount': total_amount,
 192                     'type': 'dr'
 193                 }))
 194                 total += total_amount
 195             if not exp.employee_id.address_home_id:
 196                 raise osv.except_osv(_('Error!'), _('The employee must have a home address.'))
 197             acc = exp.employee_id.address_home_id.property_account_payable.id
 198             voucher = {
 199                 'name': exp.name or '/',
 200                 'reference': sequence_obj.get(cr, uid, 'hr.expense.invoice'),
 201                 'account_id': acc,
 202                 'type': 'purchase',
 203                 'partner_id': exp.employee_id.address_home_id.id,
 204                 'company_id': company_id,
 205                 'line_ids': lines,
 206                 'amount': total,
 207                 'journal_id': journal.id,
 208             }
 209             if journal and not journal.analytic_journal_id:
 210                 analytic_journal_ids = analytic_journal_obj.search(cr, uid, [('type','=','purchase')], context=context)
 211                 if analytic_journal_ids:
 212                     account_journal.write(cr, uid, [journal.id], {'analytic_journal_id': analytic_journal_ids[0]}, context=context)
 213             voucher_id = voucher_obj.create(cr, uid, voucher, context=context)
 214             wkf_service.trg_validate(uid, 'account.voucher', voucher_id, 'proforma_voucher', cr)
 215             self.write(cr, uid, [exp.id], {'voucher_id': voucher_id, 'state': 'done'}, context=context)
 216         return True
 217
 218     def action_view_receipt(self, cr, uid, ids, context=None):
 219         '''
 220         This function returns an action that display existing receipt of given expense ids.
 221         '''
 222         assert len(ids) == 1, 'This option should only be used for a single id at a time'
 223         voucher_id = self.browse(cr, uid, ids[0], context=context).voucher_id.id
 224         res = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account_voucher', 'view_purchase_receipt_form')
 225         result = {
 226             'name': _('Expense Receipt'),
 227             'view_type': 'form',
 228             'view_mode': 'form',
 229             'view_id': res and res[1] or False,
 230             'res_model': 'account.voucher',
 231             'type': 'ir.actions.act_window',
 232             'nodestroy': True,
 233             'target': 'current',
 234             'res_id': voucher_id,
 235         }
 236         return result
 237
 238 hr_expense_expense()
 239
 240 class product_product(osv.osv):
 241     _inherit = "product.product"
 242     _columns = {
 243         'hr_expense_ok': fields.boolean('Can be Expensed', help="Specify if the product can be selected in an HR expense line."),
 244     }
 245
 246 product_product()
 247
 248 class hr_expense_line(osv.osv):
 249     _name = "hr.expense.line"
 250     _description = "Expense Line"
 251
 252     def _amount(self, cr, uid, ids, field_name, arg, context=None):
 253         if not ids:
 254             return {}
 255         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),))
 256         res = dict(cr.fetchall())
 257         return res
 258
 259     def _get_uom_id(self, cr, uid, context=None):
 260         result = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'product', 'product_uom_unit')
 261         return result and result[1] or False
 262
 263     _columns = {
 264         'name': fields.char('Expense Note', size=128, required=True),
 265         'date_value': fields.date('Date', required=True),
 266         'expense_id': fields.many2one('hr.expense.expense', 'Expense', ondelete='cascade', select=True),
 267         'total_amount': fields.function(_amount, string='Total', digits_compute=dp.get_precision('Account')),
 268         'unit_amount': fields.float('Unit Price', digits_compute=dp.get_precision('Product Price')),
 269         'unit_quantity': fields.float('Quantities', digits_compute= dp.get_precision('Product Unit of Measure')),
 270         'product_id': fields.many2one('product.product', 'Product', domain=[('hr_expense_ok','=',True)]),
 271         'uom_id': fields.many2one('product.uom', 'Unit of Measure', required=True),
 272         'description': fields.text('Description'),
 273         'analytic_account': fields.many2one('account.analytic.account','Analytic account'),
 274         'ref': fields.char('Reference', size=32),
 275         'sequence': fields.integer('Sequence', select=True, help="Gives the sequence order when displaying a list of expense lines."),
 276         }
 277     _defaults = {
 278         'unit_quantity': 1,
 279         'date_value': lambda *a: time.strftime('%Y-%m-%d'),
 280         'uom_id': _get_uom_id,
 281     }
 282     _order = "sequence, date_value desc"
 283
 284     def onchange_product_id(self, cr, uid, ids, product_id, context=None):
 285         res = {}
 286         if product_id:
 287             product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
 288             res['name'] = product.name
 289             amount_unit = product.price_get('standard_price')[product.id]
 290             res['unit_amount'] = amount_unit
 291             res['uom_id'] = product.uom_id.id
 292         return {'value': res}
 293
 294     def onchange_uom(self, cr, uid, ids, product_id, uom_id, context=None):
 295         res = {'value':{}}
 296         if not uom_id or not product_id:
 297             return res
 298         product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
 299         uom = self.pool.get('product.uom').browse(cr, uid, uom_id, context=context)
 300         if uom.category_id.id != product.uom_id.category_id.id:
 301             res['warning'] = {'title': _('Warning'), 'message': _('Selected Unit of Measure does not belong to the same category as the product Unit of Measure')}
 302             res['value'].update({'uom_id': product.uom_id.id})
 303         return res
 304
 305 hr_expense_line()
 306
 307 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: