addons/project_timesheet/project_timesheet.py

   1 # -*- encoding: utf-8 -*-
   2 from osv import fields, osv
   3 import time
   4 import datetime
   5 import pooler
   6
   7
   8 class project_work(osv.osv):
   9     _inherit = "project.task.work"
  10     _description = "Task Work"
  11     def create(self, cr, uid, vals, *args, **kwargs):
  12
  13         obj = self.pool.get('hr.analytic.timesheet')
  14         vals_line={}
  15         obj_task = self.pool.get('project.task').browse(cr, uid, vals['task_id'])
  16
  17         vals_line['name']=vals['name']
  18         vals_line['user_id']=vals['user_id']
  19         c_date=datetime.date(int(vals['date'][:4]),int(vals['date'][5:7]),int(vals['date'][8:10]))
  20         vals_line['date']=c_date
  21         vals_line['unit_amount']=vals['hours']
  22         vals_line['account_id']=obj_task.project_id.category_id.id
  23         vals_line['amount']=00.0
  24         timeline_id=obj.create(cr, uid,vals_line,{})
  25
  26         vals_line['amount']=(-1) * vals['hours']*obj.browse(cr,uid,timeline_id).product_id.standard_price
  27         obj.write(cr, uid,[timeline_id],vals_line,{})
  28         vals['hr_analytic_timesheet_id']=timeline_id
  29
  30         return super(project_work,self).create(cr, uid, vals, *args, **kwargs)
  31
  32     def write(self, cr, uid, ids,vals,context={}):
  33         vals_line={}
  34
  35         line_id=self.pool.get('project.task.work').browse(cr,uid,ids)[0].hr_analytic_timesheet_id
  36         # in case,if a record is deleted from timesheet,but we change it from tasks!
  37         list_avail_ids=self.pool.get('hr.analytic.timesheet').search(cr,uid,[])
  38         if line_id in list_avail_ids:
  39             obj = self.pool.get('hr.analytic.timesheet')
  40             vals_line['name']=vals['name']
  41             vals_line['user_id']=vals['user_id']
  42             c_date=datetime.date(int(vals['date'][:4]),int(vals['date'][5:7]),int(vals['date'][8:10]))
  43             vals_line['date']=c_date
  44             vals_line['unit_amount']=vals['hours']
  45             vals_line['amount']=(-1) * vals['hours'] * obj.browse(cr,uid,line_id).product_id.standard_price
  46             obj.write(cr, uid,[line_id],vals_line,{})
  47
  48         return super(project_work,self).write(cr, uid, ids,vals,context)
  49
  50     def unlink(self, cr, uid, ids, *args, **kwargs):
  51         timesheet_id=self.pool.get('project.task.work').browse(cr,uid,ids)[0].hr_analytic_timesheet_id
  52         # delete entry from timesheet too while deleting entry to task.
  53         list_avail_ids=self.pool.get('hr.analytic.timesheet').search(cr,uid,[])
  54         if timesheet_id in list_avail_ids:
  55            obj = self.pool.get('hr.analytic.timesheet').unlink(cr,uid,[timesheet_id],*args)
  56
  57         return super(project_work,self).unlink(cr, uid, ids,*args, **kwargs)
  58
  59     _columns={
  60         'hr_analytic_timesheet_id':fields.integer('Related Timeline Id')
  61     }
  62
  63
  64 project_work()
  65 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
  66