addons/hr_timesheet/hr_timesheet.py

   1 ##############################################################################
   2 #
   3 # Copyright (c) 2006 TINY SPRL. (http://tiny.be) All Rights Reserved.
   4 #
   5 # $Id$
   6 #
   7 # WARNING: This program as such is intended to be used by professional
   8 # programmers who take the whole responsability of assessing all potential
   9 # consequences resulting from its eventual inadequacies and bugs
  10 # End users who are looking for a ready-to-use solution with commercial
  11 # garantees and support are strongly adviced to contract a Free Software
  12 # Service Company
  13 #
  14 # This program is Free Software; you can redistribute it and/or
  15 # modify it under the terms of the GNU General Public License
  16 # as published by the Free Software Foundation; either version 2
  17 # of the License, or (at your option) any later version.
  18 #
  19 # This program is distributed in the hope that it will be useful,
  20 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  21 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  22 # GNU General Public License for more details.
  23 #
  24 # You should have received a copy of the GNU General Public License
  25 # along with this program; if not, write to the Free Software
  26 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  27 #
  28 ##############################################################################
  29
  30 import time
  31 from osv import fields
  32 from osv import osv
  33
  34
  35 class hr_employee(osv.osv):
  36         _name = "hr.employee"
  37         _inherit = "hr.employee"
  38         _columns = {
  39                 'product_id': fields.many2one('product.product', 'Product'),
  40                 'journal_id': fields.many2one('account.analytic.journal', 'Analytic Journal')
  41         }
  42 hr_employee()
  43
  44
  45 class hr_analytic_timesheet(osv.osv):
  46         _name = "hr.analytic.timesheet"
  47         _table = 'hr_analytic_timesheet'
  48         _inherits = {'account.analytic.line': 'line_id'}
  49         _columns = {
  50                 'line_id' : fields.many2one('account.analytic.line', 'Analytic line', ondelete='cascade'),
  51         }
  52
  53         def unlink(self, cr, uid, ids, context={}):
  54                 toremove = {}
  55                 for obj in self.browse(cr, uid, ids, context):
  56                         toremove[obj.line_id.id] = True
  57                 self.pool.get('account.analytic.line').unlink(cr, uid, toremove.keys(), context)
  58                 return super(hr_analytic_timesheet, self).unlink(cr, uid, ids, context)
  59
  60         def on_change_unit_amount(self, cr, uid, id, prod_id, unit_amount, unit, context={}):
  61                 return self.pool.get('account.analytic.line').on_change_unit_amount(cr, uid, id, prod_id, unit_amount,unit, context)
  62
  63         def _getEmployeeProduct(self, cr, uid, context):
  64                 emp_obj = self.pool.get('hr.employee')
  65                 emp_id = emp_obj.search(cr, uid, [('user_id', '=', uid)])
  66                 if emp_id:
  67                         return emp_obj.browse(cr, uid, emp_id[0], context).product_id.id
  68                 return False
  69
  70         def _getEmployeeUnit(self, cr, uid, context):
  71                 emp_obj = self.pool.get('hr.employee')
  72                 emp_id = emp_obj.search(cr, uid, [('user_id', '=', uid)])
  73                 if emp_id:
  74                         emp=emp_obj.browse(cr, uid, emp_id[0], context)
  75                         if not emp.product_id:
  76                                 raise osv.except_osv('No product !', "The employee haven't an associated product")
  77                         return emp.product_id.uom_id.id
  78                 return False
  79
  80         def _getGeneralAccount(self, cr, uid, context):
  81                 emp_obj = self.pool.get('hr.employee')
  82                 emp_id = emp_obj.search(cr, uid, [('user_id', '=', uid)])
  83                 if emp_id:
  84                         emp = self.pool.get('hr.employee').browse(cr, uid, emp_id[0], context=context)
  85                         if bool(emp.product_id):
  86                                 a =  emp.product_id.product_tmpl_id.property_account_income
  87                                 if not a:
  88                                         a = emp.product_id.categ_id.property_account_income_categ
  89                                 return a[0]
  90                 return False
  91
  92         def _getAnalyticJournal(self, cr, uid, context):
  93                 emp_obj = self.pool.get('hr.employee')
  94                 emp_id = emp_obj.search(cr, uid, [('user_id', '=', uid)])
  95                 if emp_id:
  96                         emp = self.pool.get('hr.employee').browse(cr, uid, emp_id[0], context=context)
  97                         if emp.journal_id:
  98                                 return emp.journal_id.id
  99                 return False
 100
 101
 102         _defaults = {
 103                 'product_uom_id' : _getEmployeeUnit,
 104                 'product_id' : _getEmployeeProduct,
 105                 'general_account_id' : _getGeneralAccount,
 106                 'journal_id' : _getAnalyticJournal,
 107                 'user_id' : lambda obj, cr, uid, ctx : uid,
 108         }
 109
 110         def on_change_account_id(self, cr, uid, ids, account_id):
 111                 return {}
 112                 if not account_id:
 113                         return {}
 114                 account = self.pool.get('account.analytic.account').read(cr, uid, [account_id], ['name'])[0]['name']
 115                 user = self.pool.get('res.users').read(cr, uid, [uid], ['name'])[0]['name']
 116                 return {'value' : {'name' : '%s (%s)' % (account, user)}}
 117 hr_analytic_timesheet()