221 lines
10 KiB
Python
221 lines
10 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
import time
|
|
|
|
from openerp.osv import fields
|
|
from openerp.osv import osv
|
|
from openerp.tools.translate import _
|
|
import openerp
|
|
|
|
class hr_employee(osv.osv):
|
|
_name = "hr.employee"
|
|
_inherit = "hr.employee"
|
|
_columns = {
|
|
'product_id': fields.many2one('product.product', 'Product', help="If you want to reinvoice working time of employees, link this employee to a service to determinate the cost price of the job."),
|
|
'journal_id': fields.many2one('account.analytic.journal', 'Analytic Journal'),
|
|
'uom_id': fields.related('product_id', 'uom_id', type='many2one', relation='product.uom', string='Unit of Measure', store=True, readonly=True)
|
|
}
|
|
|
|
def _getAnalyticJournal(self, cr, uid, context=None):
|
|
md = self.pool.get('ir.model.data')
|
|
try:
|
|
dummy, res_id = md.get_object_reference(cr, uid, 'hr_timesheet', 'analytic_journal')
|
|
#search on id found in result to check if current user has read access right
|
|
check_right = self.pool.get('account.analytic.journal').search(cr, uid, [('id', '=', res_id)], context=context)
|
|
if check_right:
|
|
return res_id
|
|
except ValueError:
|
|
pass
|
|
return False
|
|
|
|
def _getEmployeeProduct(self, cr, uid, context=None):
|
|
md = self.pool.get('ir.model.data')
|
|
try:
|
|
dummy, res_id = md.get_object_reference(cr, uid, 'product', 'product_product_consultant')
|
|
#search on id found in result to check if current user has read access right
|
|
check_right = self.pool.get('product.template').search(cr, uid, [('id', '=', res_id)], context=context)
|
|
if check_right:
|
|
return res_id
|
|
except ValueError:
|
|
pass
|
|
return False
|
|
|
|
_defaults = {
|
|
'journal_id': _getAnalyticJournal,
|
|
'product_id': _getEmployeeProduct
|
|
}
|
|
|
|
|
|
class hr_analytic_timesheet(osv.osv):
|
|
_name = "hr.analytic.timesheet"
|
|
_table = 'hr_analytic_timesheet'
|
|
_description = "Timesheet Line"
|
|
_inherits = {'account.analytic.line': 'line_id'}
|
|
_order = "id desc"
|
|
_columns = {
|
|
'line_id': fields.many2one('account.analytic.line', 'Analytic Line', ondelete='cascade', required=True),
|
|
'partner_id': fields.related('account_id', 'partner_id', type='many2one', string='Partner', relation='res.partner', store=True),
|
|
}
|
|
|
|
def unlink(self, cr, uid, ids, context=None):
|
|
toremove = {}
|
|
for obj in self.browse(cr, uid, ids, context=context):
|
|
toremove[obj.line_id.id] = True
|
|
super(hr_analytic_timesheet, self).unlink(cr, uid, ids, context=context)
|
|
self.pool.get('account.analytic.line').unlink(cr, uid, toremove.keys(), context=context)
|
|
return True
|
|
|
|
|
|
def on_change_unit_amount(self, cr, uid, id, prod_id, unit_amount, company_id, unit=False, journal_id=False, context=None):
|
|
res = {'value':{}}
|
|
if prod_id and unit_amount:
|
|
# find company
|
|
company_id = self.pool.get('res.company')._company_default_get(cr, uid, 'account.analytic.line', context=context)
|
|
r = self.pool.get('account.analytic.line').on_change_unit_amount(cr, uid, id, prod_id, unit_amount, company_id, unit, journal_id, context=context)
|
|
if r:
|
|
res.update(r)
|
|
# update unit of measurement
|
|
if prod_id:
|
|
uom = self.pool.get('product.product').browse(cr, uid, prod_id, context=context)
|
|
if uom.uom_id:
|
|
res['value'].update({'product_uom_id': uom.uom_id.id})
|
|
else:
|
|
res['value'].update({'product_uom_id': False})
|
|
return res
|
|
|
|
def _getEmployeeProduct(self, cr, uid, context=None):
|
|
if context is None:
|
|
context = {}
|
|
emp_obj = self.pool.get('hr.employee')
|
|
emp_id = emp_obj.search(cr, uid, [('user_id', '=', context.get('user_id') or uid)], context=context)
|
|
if emp_id:
|
|
emp = emp_obj.browse(cr, uid, emp_id[0], context=context)
|
|
if emp.product_id:
|
|
return emp.product_id.id
|
|
return False
|
|
|
|
def _getEmployeeUnit(self, cr, uid, context=None):
|
|
emp_obj = self.pool.get('hr.employee')
|
|
if context is None:
|
|
context = {}
|
|
emp_id = emp_obj.search(cr, uid, [('user_id', '=', context.get('user_id') or uid)], context=context)
|
|
if emp_id:
|
|
emp = emp_obj.browse(cr, uid, emp_id[0], context=context)
|
|
if emp.product_id:
|
|
return emp.product_id.uom_id.id
|
|
return False
|
|
|
|
def _getGeneralAccount(self, cr, uid, context=None):
|
|
emp_obj = self.pool.get('hr.employee')
|
|
if context is None:
|
|
context = {}
|
|
emp_id = emp_obj.search(cr, uid, [('user_id', '=', context.get('user_id') or uid)], context=context)
|
|
if emp_id:
|
|
emp = emp_obj.browse(cr, uid, emp_id[0], context=context)
|
|
if bool(emp.product_id):
|
|
a = emp.product_id.property_account_expense.id
|
|
if not a:
|
|
a = emp.product_id.categ_id.property_account_expense_categ.id
|
|
if a:
|
|
return a
|
|
return False
|
|
|
|
def _getAnalyticJournal(self, cr, uid, context=None):
|
|
emp_obj = self.pool.get('hr.employee')
|
|
if context is None:
|
|
context = {}
|
|
if context.get('employee_id'):
|
|
emp_id = [context.get('employee_id')]
|
|
else:
|
|
emp_id = emp_obj.search(cr, uid, [('user_id','=',context.get('user_id') or uid)], limit=1, context=context)
|
|
if not emp_id:
|
|
model, action_id = self.pool['ir.model.data'].get_object_reference(cr, uid, 'hr', 'open_view_employee_list_my')
|
|
msg = _("Employee is not created for this user. Please create one from configuration panel.")
|
|
raise openerp.exceptions.RedirectWarning(msg, action_id, _('Go to the configuration panel'))
|
|
emp = emp_obj.browse(cr, uid, emp_id[0], context=context)
|
|
if emp.journal_id:
|
|
return emp.journal_id.id
|
|
else :
|
|
raise osv.except_osv(_('Warning!'), _('No analytic journal defined for \'%s\'.\nYou should assign an analytic journal on the employee form.')%(emp.name))
|
|
|
|
|
|
_defaults = {
|
|
'product_uom_id': _getEmployeeUnit,
|
|
'product_id': _getEmployeeProduct,
|
|
'general_account_id': _getGeneralAccount,
|
|
'journal_id': _getAnalyticJournal,
|
|
'date': lambda self, cr, uid, ctx: ctx.get('date', fields.date.context_today(self,cr,uid,context=ctx)),
|
|
'user_id': lambda obj, cr, uid, ctx: ctx.get('user_id') or uid,
|
|
}
|
|
def on_change_account_id(self, cr, uid, ids, account_id, context=None):
|
|
return {'value':{}}
|
|
|
|
def on_change_date(self, cr, uid, ids, date):
|
|
if ids:
|
|
new_date = self.read(cr, uid, ids[0], ['date'])['date']
|
|
if date != new_date:
|
|
warning = {'title':_('User Alert!'),'message':_('Changing the date will let this entry appear in the timesheet of the new date.')}
|
|
return {'value':{},'warning':warning}
|
|
return {'value':{}}
|
|
|
|
def create(self, cr, uid, vals, context=None):
|
|
if context is None:
|
|
context = {}
|
|
emp_obj = self.pool.get('hr.employee')
|
|
emp_id = emp_obj.search(cr, uid, [('user_id', '=', context.get('user_id') or uid)], context=context)
|
|
ename = ''
|
|
if emp_id:
|
|
ename = emp_obj.browse(cr, uid, emp_id[0], context=context).name
|
|
if not vals.get('journal_id',False):
|
|
raise osv.except_osv(_('Warning!'), _('No \'Analytic Journal\' is defined for employee %s \nDefine an employee for the selected user and assign an \'Analytic Journal\'!')%(ename,))
|
|
if not vals.get('account_id',False):
|
|
raise osv.except_osv(_('Warning!'), _('No analytic account is defined on the project.\nPlease set one or we cannot automatically fill the timesheet.'))
|
|
return super(hr_analytic_timesheet, self).create(cr, uid, vals, context=context)
|
|
|
|
def on_change_user_id(self, cr, uid, ids, user_id):
|
|
if not user_id:
|
|
return {}
|
|
context = {'user_id': user_id}
|
|
return {'value': {
|
|
'product_id': self. _getEmployeeProduct(cr, uid, context),
|
|
'product_uom_id': self._getEmployeeUnit(cr, uid, context),
|
|
'general_account_id': self._getGeneralAccount(cr, uid, context),
|
|
'journal_id': self._getAnalyticJournal(cr, uid, context),
|
|
}}
|
|
|
|
class account_analytic_account(osv.osv):
|
|
|
|
_inherit = 'account.analytic.account'
|
|
_description = 'Analytic Account'
|
|
_columns = {
|
|
'use_timesheets': fields.boolean('Timesheets', help="Check this field if this project manages timesheets"),
|
|
}
|
|
|
|
def on_change_template(self, cr, uid, ids, template_id, date_start=False, context=None):
|
|
res = super(account_analytic_account, self).on_change_template(cr, uid, ids, template_id, date_start=date_start, context=context)
|
|
if template_id and 'value' in res:
|
|
template = self.browse(cr, uid, template_id, context=context)
|
|
res['value']['use_timesheets'] = template.use_timesheets
|
|
return res
|
|
|
|
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|