You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

115 lines
4.7 KiB

7 years ago
7 years ago
  1. # -*- coding: utf-8 -*-
  2. from openerp import models, fields, api, _
  3. from openerp.exceptions import UserError
  4. class StatusActionMixin(models.AbstractModel):
  5. _name = "beesdoo.shift.action_mixin"
  6. cooperator_id = fields.Many2one('res.partner', default=lambda self: self.env['res.partner'].browse(self._context.get('active_id')), required=True)
  7. def _check(self, group='beesdoo_shift.group_shift_management'):
  8. self.ensure_one()
  9. if not self.env.user.has_group(group):
  10. raise UserError(_("You don't have the required access for this operation."))
  11. if self.cooperator_id == self.env.user.partner_id and not self.env.user.has_group('beesdoo_shift.group_cooperative_admin'):
  12. raise UserError(_("You cannot perform this operation on yourself"))
  13. return self.with_context(real_uid=self._uid)
  14. class Subscribe(models.TransientModel):
  15. _name = 'beesdoo.shift.subscribe'
  16. _inherit = 'beesdoo.shift.action_mixin'
  17. def _get_date(self):
  18. date = self.env['res.partner'].browse(self._context.get('active_id')).info_session_date
  19. if not date:
  20. return fields.Date.today()
  21. else:
  22. return date
  23. def _get_super(self):
  24. return self.env['res.partner'].browse(self._context.get('active_id')).super
  25. def _get_mode(self):
  26. return self.env['res.partner'].browse(self._context.get('active_id')).working_mode
  27. def _get_reset_counter_default(self):
  28. partner = self.env['res.partner'].browse(self._context.get('active_id'))
  29. return partner.state == 'unsubscribed' and partner.working_mode == 'regular'
  30. info_session = fields.Boolean(string="Followed an information session", default=False)
  31. info_session_date = fields.Date(string="Date of information session")
  32. super = fields.Boolean(string="Super Cooperator", default=_get_super)
  33. working_mode = fields.Selection(
  34. [
  35. ('regular', 'Regular worker'),
  36. ('irregular', 'Irregular worker'),
  37. ('exempt', 'Exempted'),
  38. ], default=_get_mode
  39. )
  40. exempt_reason_id = fields.Many2one('cooperative.exempt.reason', 'Exempt Reason')
  41. shift_id = fields.Many2one('beesdoo.shift.template')
  42. reset_counter = fields.Boolean(default=_get_reset_counter_default)
  43. reset_compensation_counter = fields.Boolean(default=False)
  44. unsubscribed = fields.Boolean(default=False, string="Are you sure to unsubscribe this cooperator")
  45. irregular_start_date = fields.Date(string="Start Date", default=fields.Date.today)
  46. resigning = fields.Boolean(default=False, help="Want to leave the beescoop")
  47. @api.multi
  48. def unsubscribe(self):
  49. self = self._check()
  50. if not self.unsubscribed:
  51. return
  52. status_id = self.env['cooperative.status'].search([('cooperator_id', '=', self.cooperator_id.id)])
  53. data = {
  54. 'unsubscribed': True,
  55. 'cooperator_id': self.cooperator_id.id,
  56. 'resigning' : self.resigning,
  57. }
  58. if status_id:
  59. status_id.sudo().write(data)
  60. else:
  61. self.env['cooperative.status'].sudo().create(data)
  62. @api.multi
  63. def subscribe(self):
  64. self = self._check()
  65. if self.shift_id and self.shift_id.remaining_worker <= 0:
  66. raise UserError(_('There is no remaining space for this shift'))
  67. if self.shift_id:
  68. #Remove existing shift then subscribe to the new shift
  69. self.cooperator_id.sudo().write({'subscribed_shift_ids' : [(6,0, [self.shift_id.id])]})
  70. if self.working_mode != 'regular':
  71. #Remove existing shift then subscribe to the new shift
  72. self.cooperator_id.sudo().write({'subscribed_shift_ids' : [(5,)]})
  73. data = {
  74. 'info_session' : self.info_session,
  75. 'info_session_date': self.info_session_date,
  76. 'working_mode' : self.working_mode,
  77. 'exempt_reason_id' : self.exempt_reason_id.id,
  78. 'super' : self.super,
  79. 'cooperator_id': self.cooperator_id.id,
  80. 'unsubscribed': False,
  81. 'irregular_start_date': self.irregular_start_date,
  82. 'irregular_absence_date': False,
  83. 'irregular_absence_counter': 0,
  84. }
  85. if self.reset_counter:
  86. data['sr'] = 0
  87. data['extension_start_time'] = False
  88. data['alert_start_time'] = False
  89. data['time_extension'] = 0
  90. if self.reset_compensation_counter:
  91. data['sc'] = 0
  92. status_id = self.env['cooperative.status'].search([('cooperator_id', '=', self.cooperator_id.id)])
  93. if status_id:
  94. status_id.sudo().write(data)
  95. else:
  96. self.env['cooperative.status'].sudo().create(data)