diff --git a/cor_custom/models/analytic.py b/cor_custom/models/analytic.py index e222f85..6de42d1 100755 --- a/cor_custom/models/analytic.py +++ b/cor_custom/models/analytic.py @@ -133,13 +133,22 @@ class AccountAnalyticLine(models.Model): print('VVVVVVVVVVVVVVVVVVVVVVVVVVVVVVVV', self, vals.get('employee_id'), vals.get('project_id')) project = self.env['project.project'].search([('id', '=', vals.get('project_id'))]) if project: - project.consultant_timesheet_hrs.create({'project_id': vals.get('project_id'), - 'employee_id': vals.get('employee_id'), - 'timesheet_hour': vals.get('unit_amount')}) + if project.consultant_timesheet_hrs: + for record in project.consultant_timesheet_hrs: + if record.employee_id.id == vals.get('employee_id'): + record.timesheet_hour = record.timesheet_hour + vals.get('unit_amount') + else: + project.consultant_timesheet_hrs.create({'project_id': vals.get('project_id'), + 'employee_id': vals.get('employee_id'), + 'timesheet_hour': vals.get('unit_amount')}) + else: + project.consultant_timesheet_hrs.create({'project_id': vals.get('project_id'), + 'employee_id': vals.get('employee_id'), + 'timesheet_hour': vals.get('unit_amount')}) value = super(AccountAnalyticLine, self).create(vals) - if self.project_id: - self.project_id._compute_consultant_timesheet_hour() - print('WWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWW') + # if self.project_id: + # self.project_id._compute_consultant_timesheet_hour() + # print('WWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWW') return value def write(self, vals):