1 # -*- coding: utf-8 -*-
2 ##############################################################################
4 # OpenERP, Open Source Management Solution
5 # Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
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.
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.
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/>.
20 ##############################################################################
24 from openerp import netsvc
25 from openerp.osv import fields, osv
26 from openerp.tools.translate import _
28 import openerp.addons.decimal_precision as dp
30 def _employee_get(obj, cr, uid, context=None):
33 ids = obj.pool.get('hr.employee').search(cr, uid, [('user_id', '=', uid)], context=context)
38 class hr_expense_expense(osv.osv):
40 def copy(self, cr, uid, id, default=None, context=None):
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)
47 def _amount(self, cr, uid, ids, field_name, arg, context=None):
49 for expense in self.browse(cr, uid, ids, context=context):
51 for line in expense.line_ids:
52 total += line.unit_amount * line.unit_quantity
53 res[expense.id] = total
56 def _get_currency(self, cr, uid, context=None):
57 user = self.pool.get('res.users').browse(cr, uid, [uid], context=context)[0]
59 return user.company_id.currency_id.id
61 return self.pool.get('res.currency').search(cr, uid, [('rate','=',1.0)], context=context)[0]
63 _name = "hr.expense.expense"
64 _inherit = ['mail.thread']
65 _description = "Expense"
69 'hr_expense.mt_expense_approved': lambda self, cr, uid, obj, ctx=None: obj['state'] == 'accepted',
70 'hr_expense.mt_expense_refused': lambda self, cr, uid, obj, ctx=None: obj['state'] == 'cancelled',
71 'hr_expense.mt_expense_confirmed': lambda self, cr, uid, obj, ctx=None: obj['state'] == 'confirm',
76 'name': fields.char('Description', size=128, required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
77 'id': fields.integer('Sheet ID', readonly=True),
78 'date': fields.date('Date', select=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
79 'journal_id': fields.many2one('account.journal', 'Force Journal', help = "The journal used when the expense is done."),
80 'employee_id': fields.many2one('hr.employee', "Employee", required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
81 'user_id': fields.many2one('res.users', 'User', required=True),
82 '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."),
83 '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."),
84 'user_valid': fields.many2one('res.users', 'Validation By', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
85 'account_move_id': fields.many2one('account.move', 'Ledger Posting'),
86 'line_ids': fields.one2many('hr.expense.line', 'expense_id', 'Expense Lines', readonly=True, states={'draft':[('readonly',False)]} ),
87 'note': fields.text('Note'),
88 'amount': fields.function(_amount, string='Total Amount', digits_compute=dp.get_precision('Account')),
89 'voucher_id': fields.many2one('account.voucher', "Employee's Receipt"),
90 'currency_id': fields.many2one('res.currency', 'Currency', required=True, readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
91 'department_id':fields.many2one('hr.department','Department', readonly=True, states={'draft':[('readonly',False)], 'confirm':[('readonly',False)]}),
92 'company_id': fields.many2one('res.company', 'Company', required=True),
93 'state': fields.selection([
95 ('cancelled', 'Refused'),
96 ('confirm', 'Waiting Approval'),
97 ('accepted', 'Approved'),
100 'Status', readonly=True, track_visibility=1,
101 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\'.\
102 \nIf the admin accepts it, the status is \'Accepted\'.\n If a receipt is made for the expense request, the status is \'Done\'.'),
105 'company_id': lambda s, cr, uid, c: s.pool.get('res.company')._company_default_get(cr, uid, 'hr.employee', context=c),
106 'date': fields.date.context_today,
108 'employee_id': _employee_get,
109 'user_id': lambda cr, uid, id, c={}: id,
110 'currency_id': _get_currency,
113 def unlink(self, cr, uid, ids, context=None):
114 for rec in self.browse(cr, uid, ids, context=context):
115 if rec.state != 'draft':
116 raise osv.except_osv(_('Warning!'),_('You can only delete draft expenses!'))
117 return super(hr_expense_expense, self).unlink(cr, uid, ids, context)
119 def onchange_currency_id(self, cr, uid, ids, currency_id=False, company_id=False, context=None):
120 res = {'value': {'journal_id': False}}
121 journal_ids = self.pool.get('account.journal').search(cr, uid, [('type','=','purchase'), ('currency','=',currency_id), ('company_id', '=', company_id)], context=context)
123 res['value']['journal_id'] = journal_ids[0]
126 def onchange_employee_id(self, cr, uid, ids, employee_id, context=None):
127 emp_obj = self.pool.get('hr.employee')
128 department_id = False
131 employee = emp_obj.browse(cr, uid, employee_id, context=context)
132 department_id = employee.department_id.id
133 company_id = employee.company_id.id
134 return {'value': {'department_id': department_id, 'company_id': company_id}}
136 def expense_confirm(self, cr, uid, ids, context=None):
137 for expense in self.browse(cr, uid, ids):
138 if expense.employee_id and expense.employee_id.parent_id.user_id:
139 self.message_subscribe_users(cr, uid, [expense.id], user_ids=[expense.employee_id.parent_id.user_id.id])
140 return self.write(cr, uid, ids, {'state': 'confirm', 'date_confirm': time.strftime('%Y-%m-%d')}, context=context)
142 def expense_accept(self, cr, uid, ids, context=None):
143 return self.write(cr, uid, ids, {'state': 'accepted', 'date_valid': time.strftime('%Y-%m-%d'), 'user_valid': uid}, context=context)
145 def expense_canceled(self, cr, uid, ids, context=None):
146 return self.write(cr, uid, ids, {'state': 'cancelled'}, context=context)
148 def action_receipt_create(self, cr, uid, ids, context=None):
149 property_obj = self.pool.get('ir.property')
150 sequence_obj = self.pool.get('ir.sequence')
151 analytic_journal_obj = self.pool.get('account.analytic.journal')
152 account_journal = self.pool.get('account.journal')
153 voucher_obj = self.pool.get('account.voucher')
154 currency_obj = self.pool.get('res.currency')
155 wkf_service = netsvc.LocalService("workflow")
158 for exp in self.browse(cr, uid, ids, context=context):
159 company_id = exp.company_id.id
163 ctx.update({'date': exp.date})
166 journal = exp.journal_id
168 journal_id = voucher_obj._get_journal(cr, uid, context={'type': 'purchase', 'company_id': company_id})
170 journal = account_journal.browse(cr, uid, journal_id, context=context)
171 for line in exp.line_ids:
173 acc = line.product_id.property_account_expense
175 acc = line.product_id.categ_id.property_account_expense_categ
177 acc = property_obj.get(cr, uid, 'property_account_expense_categ', 'product.category', context={'force_company': company_id})
179 raise osv.except_osv(_('Error!'), _('Please configure Default Expense account for Product purchase: `property_account_expense_categ`.'))
180 total_amount = line.total_amount
182 if exp.currency_id != journal.currency:
183 total_amount = currency_obj.compute(cr, uid, exp.currency_id.id, journal.currency.id, total_amount, context=ctx)
184 elif exp.currency_id != exp.company_id.currency_id:
185 total_amount = currency_obj.compute(cr, uid, exp.currency_id.id, exp.company_id.currency_id.id, total_amount, context=ctx)
186 lines.append((0, False, {
188 'account_id': acc.id,
189 'account_analytic_id': line.analytic_account.id,
190 'amount': total_amount,
193 total += total_amount
194 if not exp.employee_id.address_home_id:
195 raise osv.except_osv(_('Error!'), _('The employee must have a home address.'))
196 acc = exp.employee_id.address_home_id.property_account_payable.id
198 'name': exp.name or '/',
199 'reference': sequence_obj.get(cr, uid, 'hr.expense.invoice'),
202 'partner_id': exp.employee_id.address_home_id.id,
203 'company_id': company_id,
206 'journal_id': journal.id,
208 if journal and not journal.analytic_journal_id:
209 analytic_journal_ids = analytic_journal_obj.search(cr, uid, [('type','=','purchase')], context=context)
210 if analytic_journal_ids:
211 account_journal.write(cr, uid, [journal.id], {'analytic_journal_id': analytic_journal_ids[0]}, context=context)
212 voucher_id = voucher_obj.create(cr, uid, voucher, context=context)
213 wkf_service.trg_validate(uid, 'account.voucher', voucher_id, 'proforma_voucher', cr)
214 self.write(cr, uid, [exp.id], {'voucher_id': voucher_id, 'state': 'done'}, context=context)
217 def action_view_receipt(self, cr, uid, ids, context=None):
219 This function returns an action that display existing receipt of given expense ids.
221 assert len(ids) == 1, 'This option should only be used for a single id at a time'
222 voucher_id = self.browse(cr, uid, ids[0], context=context).voucher_id.id
223 res = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'account_voucher', 'view_purchase_receipt_form')
225 'name': _('Expense Receipt'),
228 'view_id': res and res[1] or False,
229 'res_model': 'account.voucher',
230 'type': 'ir.actions.act_window',
233 'res_id': voucher_id,
239 class product_product(osv.osv):
240 _inherit = "product.product"
242 'hr_expense_ok': fields.boolean('Can be Expensed', help="Specify if the product can be selected in an HR expense line."),
247 class hr_expense_line(osv.osv):
248 _name = "hr.expense.line"
249 _description = "Expense Line"
251 def _amount(self, cr, uid, ids, field_name, arg, context=None):
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())
258 def _get_uom_id(self, cr, uid, context=None):
259 result = self.pool.get('ir.model.data').get_object_reference(cr, uid, 'product', 'product_uom_unit')
260 return result and result[1] or False
263 'name': fields.char('Expense Note', size=128, required=True),
264 'date_value': fields.date('Date', required=True),
265 'expense_id': fields.many2one('hr.expense.expense', 'Expense', ondelete='cascade', select=True),
266 'total_amount': fields.function(_amount, string='Total', digits_compute=dp.get_precision('Account')),
267 'unit_amount': fields.float('Unit Price', digits_compute=dp.get_precision('Product Price')),
268 'unit_quantity': fields.float('Quantities', digits_compute= dp.get_precision('Product Unit of Measure')),
269 'product_id': fields.many2one('product.product', 'Product', domain=[('hr_expense_ok','=',True)]),
270 'uom_id': fields.many2one('product.uom', 'Unit of Measure', required=True),
271 'description': fields.text('Description'),
272 'analytic_account': fields.many2one('account.analytic.account','Analytic account'),
273 'ref': fields.char('Reference', size=32),
274 'sequence': fields.integer('Sequence', select=True, help="Gives the sequence order when displaying a list of expense lines."),
278 'date_value': lambda *a: time.strftime('%Y-%m-%d'),
279 'uom_id': _get_uom_id,
281 _order = "sequence, date_value desc"
283 def onchange_product_id(self, cr, uid, ids, product_id, context=None):
286 product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
287 res['name'] = product.name
288 amount_unit = product.price_get('standard_price')[product.id]
289 res['unit_amount'] = amount_unit
290 res['uom_id'] = product.uom_id.id
291 return {'value': res}
293 def onchange_uom(self, cr, uid, ids, product_id, uom_id, context=None):
295 if not uom_id or not product_id:
297 product = self.pool.get('product.product').browse(cr, uid, product_id, context=context)
298 uom = self.pool.get('product.uom').browse(cr, uid, uom_id, context=context)
299 if uom.category_id.id != product.uom_id.category_id.id:
300 res['warning'] = {'title': _('Warning'), 'message': _('Selected Unit of Measure does not belong to the same category as the product Unit of Measure')}
301 res['value'].update({'uom_id': product.uom_id.id})
306 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: