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.

499 lines
22 KiB

7 years ago
7 years ago
7 years ago
7 years ago
  1. # -*- coding: utf-8 -*-
  2. from openerp import models, fields, api, _
  3. from openerp.exceptions import ValidationError, UserError
  4. from datetime import timedelta, datetime
  5. import logging
  6. from openerp.osv.fields import related
  7. _logger = logging.getLogger(__name__)
  8. PERIOD = 28 # TODO: use system parameter
  9. def add_days_delta(date_from, days_delta):
  10. if not date_from:
  11. return date_from
  12. next_date = fields.Date.from_string(date_from) + timedelta(days=days_delta)
  13. return fields.Date.to_string(next_date)
  14. class ExemptReason(models.Model):
  15. _name = 'cooperative.exempt.reason'
  16. name = fields.Char(required=True)
  17. class HistoryStatus(models.Model):
  18. _name = 'cooperative.status.history'
  19. _order= 'create_date desc'
  20. status_id = fields.Many2one('cooperative.status')
  21. cooperator_id = fields.Many2one('res.partner')
  22. change = fields.Char()
  23. type = fields.Selection([('status', 'Status Change'), ('counter', 'Counter Change')])
  24. user_id = fields.Many2one('res.users', string="User")
  25. class CooperativeStatus(models.Model):
  26. _name = 'cooperative.status'
  27. _rec_name = 'cooperator_id'
  28. _order = 'cooperator_id'
  29. today = fields.Date(help="Field that allow to compute field and store them even if they are based on the current date", default=fields.Date.today)
  30. cooperator_id = fields.Many2one('res.partner')
  31. active = fields.Boolean(related="cooperator_id.active", store=True, index=True)
  32. info_session = fields.Boolean('Information Session ?')
  33. info_session_date = fields.Datetime('Information Session Date')
  34. super = fields.Boolean("Super Cooperative")
  35. sr = fields.Integer("Compteur shift regulier", default=0)
  36. sc = fields.Integer("Compteur shift de compensation", default=0)
  37. time_extension = fields.Integer("Extension Days NB", default=0, help="Addtional days to the automatic extension, 5 mean that you have a total of 15 extension days of default one is set to 10")
  38. holiday_start_time = fields.Date("Holidays Start Day")
  39. holiday_end_time = fields.Date("Holidays End Day")
  40. alert_start_time = fields.Date("Alert Start Day")
  41. extension_start_time = fields.Date("Extension Start Day")
  42. #Champ compute
  43. working_mode = fields.Selection(
  44. [
  45. ('regular', 'Regular worker'),
  46. ('irregular', 'Irregular worker'),
  47. ('exempt', 'Exempted'),
  48. ],
  49. string="Working mode"
  50. )
  51. exempt_reason_id = fields.Many2one('cooperative.exempt.reason', 'Exempt Reason')
  52. status = fields.Selection([('ok', 'Up to Date'),
  53. ('holiday', 'Holidays'),
  54. ('alert', 'Alerte'),
  55. ('extension', 'Extension'),
  56. ('suspended', 'Suspended'),
  57. ('exempted', 'Exempted'),
  58. ('unsubscribed', 'Unsubscribed'),
  59. ('resigning', 'Resigning')],
  60. compute="_compute_status", string="Cooperative Status", store=True)
  61. can_shop = fields.Boolean(compute='_compute_status', store=True)
  62. history_ids = fields.One2many('cooperative.status.history', 'status_id', readonly=True)
  63. unsubscribed = fields.Boolean(default=False, help="Manually unsubscribed")
  64. resigning = fields.Boolean(default=False, help="Want to leave the beescoop")
  65. #Specific to irregular
  66. irregular_start_date = fields.Date() #TODO migration script
  67. irregular_absence_date = fields.Date()
  68. irregular_absence_counter = fields.Integer() #TODO unsubscribe when reach -2
  69. future_alert_date = fields.Date(compute='_compute_future_alert_date')
  70. next_countdown_date = fields.Date(compute='_compute_next_countdown_date')
  71. temporary_exempt_reason_id = fields.Many2one('cooperative.exempt.reason', 'Exempt Reason')
  72. temporary_exempt_start_date = fields.Date()
  73. temporary_exempt_end_date = fields.Date()
  74. @api.depends('today', 'sr', 'sc', 'holiday_end_time',
  75. 'holiday_start_time', 'time_extension',
  76. 'alert_start_time', 'extension_start_time',
  77. 'unsubscribed', 'irregular_absence_date',
  78. 'irregular_absence_counter', 'temporary_exempt_start_date',
  79. 'temporary_exempt_end_date', 'resigning', 'cooperator_id.subscribed_shift_ids')
  80. def _compute_status(self):
  81. alert_delay = int(self.env['ir.config_parameter'].get_param('alert_delay', 28))
  82. grace_delay = int(self.env['ir.config_parameter'].get_param('default_grace_delay', 10))
  83. update = int(self.env['ir.config_parameter'].get_param('always_update', False))
  84. for rec in self:
  85. if update or not rec.today:
  86. rec.status = 'ok'
  87. rec.can_shop = True
  88. continue
  89. if rec.resigning:
  90. rec.status = 'resigning'
  91. rec.can_shop = False
  92. continue
  93. if rec.working_mode == 'regular':
  94. rec._set_regular_status(grace_delay, alert_delay)
  95. elif rec.working_mode == 'irregular':
  96. rec._set_irregular_status(grace_delay, alert_delay)
  97. elif rec.working_mode == 'exempt':
  98. rec.status = 'ok'
  99. rec.can_shop = True
  100. @api.depends('today', 'irregular_start_date', 'sr', 'holiday_start_time',
  101. 'holiday_end_time', 'temporary_exempt_start_date',
  102. 'temporary_exempt_end_date')
  103. def _compute_future_alert_date(self):
  104. """Compute date before which the worker is up to date"""
  105. for rec in self:
  106. # Only for irregular worker
  107. if rec.working_mode != 'irregular' and not rec.irregular_start_date:
  108. rec.future_alert_date = False
  109. # Alert start time already set
  110. elif rec.alert_start_time:
  111. rec.future_alert_date = False
  112. # Holidays are not set properly
  113. elif bool(rec.holiday_start_time) != bool(rec.holiday_end_time):
  114. rec.future_alert_date = False
  115. # Exemption have not a start and end time
  116. elif (bool(rec.temporary_exempt_start_date)
  117. != bool(rec.temporary_exempt_end_date)):
  118. rec.future_alert_date = False
  119. else:
  120. date = rec.today
  121. counter = rec.sr
  122. # Simulate the countdown
  123. while counter > 0:
  124. date = add_days_delta(date, 1)
  125. date = self._next_countdown_date(rec.irregular_start_date,
  126. date)
  127. # Check holidays
  128. if (rec.holiday_start_time and rec.holiday_end_time
  129. and date >= rec.holiday_start_time
  130. and date <= rec.holiday_end_time):
  131. continue
  132. # Check temporary exemption
  133. elif (rec.temporary_exempt_start_date
  134. and rec.temporary_exempt_end_date
  135. and date >= rec.temporary_exempt_start_date
  136. and date <= rec.temporary_exempt_end_date):
  137. continue
  138. else:
  139. counter -= 1
  140. rec.future_alert_date = self._next_countdown_date(
  141. rec.irregular_start_date, date
  142. )
  143. @api.depends('today', 'irregular_start_date', 'holiday_start_time',
  144. 'holiday_end_time', 'temporary_exempt_start_date',
  145. 'temporary_exempt_end_date')
  146. def _compute_next_countdown_date(self):
  147. """
  148. Compute the following countdown date. This date is the date when
  149. the worker will see his counter changed du to the cron. This
  150. date is like the birthday date of the worker that occurred each
  151. PERIOD.
  152. """
  153. for rec in self:
  154. # Only for irregular worker
  155. if rec.working_mode != 'irregular' and not rec.irregular_start_date:
  156. rec.next_countdown_date = False
  157. # Holidays are not set properly
  158. elif bool(rec.holiday_start_time) != bool(rec.holiday_end_time):
  159. rec.next_countdown_date = False
  160. # Exemption have not a start and end time
  161. elif (bool(rec.temporary_exempt_start_date)
  162. != bool(rec.temporary_exempt_end_date)):
  163. rec.next_countdown_date = False
  164. else:
  165. date = rec.today
  166. next_countdown_date = False
  167. while not next_countdown_date:
  168. date = self._next_countdown_date(rec.irregular_start_date, date)
  169. # Check holidays
  170. if (rec.holiday_start_time and rec.holiday_end_time
  171. and date >= rec.holiday_start_time
  172. and date <= rec.holiday_end_time):
  173. date = add_days_delta(date, 1)
  174. continue
  175. # Check temporary exemption
  176. elif (rec.temporary_exempt_start_date
  177. and rec.temporary_exempt_end_date
  178. and date >= rec.temporary_exempt_start_date
  179. and date <= rec.temporary_exempt_end_date):
  180. date = add_days_delta(date, 1)
  181. continue
  182. else:
  183. next_countdown_date = date
  184. rec.next_countdown_date = next_countdown_date
  185. @api.constrains("working_mode", "irregular_start_date")
  186. def _constrains_irregular_start_date(self):
  187. if self.working_mode == "irregular" and not self.irregular_start_date:
  188. raise UserError(_("Irregular workers must have an irregular start date."))
  189. def _next_countdown_date(self, irregular_start_date, today=False):
  190. """
  191. Return the next countdown date given irregular_start_date and
  192. today dates.
  193. This does not take holiday and other status into account.
  194. """
  195. today = today or fields.Date.today()
  196. today_dt = fields.Date.from_string(today)
  197. irregular_start_dt = fields.Date.from_string(irregular_start_date)
  198. delta = (today_dt - irregular_start_dt).days
  199. if not delta % PERIOD:
  200. return today
  201. return add_days_delta(today, PERIOD - (delta % PERIOD))
  202. def _set_regular_status(self, grace_delay, alert_delay):
  203. self.ensure_one()
  204. counter_unsubscribe = int(self.env['ir.config_parameter'].get_param('regular_counter_to_unsubscribe', -4))
  205. ok = self.sr >= 0 and self.sc >= 0
  206. grace_delay = grace_delay + self.time_extension
  207. if (self.sr + self.sc) <= counter_unsubscribe or self.unsubscribed:
  208. self.status = 'unsubscribed'
  209. self.can_shop = False
  210. elif self.today >= self.temporary_exempt_start_date and self.today <= self.temporary_exempt_end_date:
  211. self.status = 'exempted'
  212. self.can_shop = True
  213. #Transition to alert sr < 0 or stay in alert sr < 0 or sc < 0 and thus alert time is defined
  214. elif not ok and self.alert_start_time and self.extension_start_time and self.today <= add_days_delta(self.extension_start_time, grace_delay):
  215. self.status = 'extension'
  216. self.can_shop = True
  217. elif not ok and self.alert_start_time and self.extension_start_time and self.today > add_days_delta(self.extension_start_time, grace_delay):
  218. self.status = 'suspended'
  219. self.can_shop = False
  220. elif not ok and self.alert_start_time and self.today > add_days_delta(self.alert_start_time, alert_delay):
  221. self.status = 'suspended'
  222. self.can_shop = False
  223. elif (self.sr < 0) or (not ok and self.alert_start_time):
  224. self.status = 'alert'
  225. self.can_shop = True
  226. #Check for holidays; Can be in holidays even in alert or other mode ?
  227. elif self.today >= self.holiday_start_time and self.today <= self.holiday_end_time:
  228. self.status = 'holiday'
  229. self.can_shop = False
  230. elif ok or (not self.alert_start_time and self.sr >= 0):
  231. self.status = 'ok'
  232. self.can_shop = True
  233. def _set_irregular_status(self, grace_delay, alert_delay):
  234. counter_unsubscribe = int(self.env['ir.config_parameter'].get_param('irregular_counter_to_unsubscribe', -3))
  235. self.ensure_one()
  236. ok = self.sr >= 0
  237. grace_delay = grace_delay + self.time_extension
  238. if self.sr <= counter_unsubscribe or self.unsubscribed:
  239. self.status = 'unsubscribed'
  240. self.can_shop = False
  241. elif self.today >= self.temporary_exempt_start_date and self.today <= self.temporary_exempt_end_date:
  242. self.status = 'exempted'
  243. self.can_shop = True
  244. #Transition to alert sr < 0 or stay in alert sr < 0 or sc < 0 and thus alert time is defined
  245. elif not ok and self.alert_start_time and self.extension_start_time and self.today <= add_days_delta(self.extension_start_time, grace_delay):
  246. self.status = 'extension'
  247. self.can_shop = True
  248. elif not ok and self.alert_start_time and self.extension_start_time and self.today > add_days_delta(self.extension_start_time, grace_delay):
  249. self.status = 'suspended'
  250. self.can_shop = False
  251. elif not ok and self.alert_start_time and self.today > add_days_delta(self.alert_start_time, alert_delay):
  252. self.status = 'suspended'
  253. self.can_shop = False
  254. elif (self.sr < 0) or (not ok and self.alert_start_time):
  255. self.status = 'alert'
  256. self.can_shop = True
  257. #Check for holidays; Can be in holidays even in alert or other mode ?
  258. elif self.today >= self.holiday_start_time and self.today <= self.holiday_end_time:
  259. self.status = 'holiday'
  260. self.can_shop = False
  261. elif ok or (not self.alert_start_time and self.sr >= 0):
  262. self.status = 'ok'
  263. self.can_shop = True
  264. @api.multi
  265. def write(self, vals):
  266. """
  267. Overwrite write to historize the change
  268. """
  269. for field in ['sr', 'sc', 'time_extension', 'extension_start_time', 'alert_start_time', 'unsubscribed']:
  270. if not field in vals:
  271. continue
  272. for rec in self:
  273. data = {
  274. 'status_id': rec.id,
  275. 'cooperator_id': rec.cooperator_id.id,
  276. 'type': 'counter',
  277. 'user_id': self.env.context.get('real_uid', self.env.uid),
  278. }
  279. if vals.get(field, rec[field]) != rec[field]:
  280. data['change'] = '%s: %s -> %s' % (field.upper(), rec[field], vals.get(field))
  281. self.env['cooperative.status.history'].sudo().create(data)
  282. return super(CooperativeStatus, self).write(vals)
  283. def _state_change(self, new_state):
  284. self.ensure_one()
  285. if new_state == 'alert':
  286. self.write({'alert_start_time': self.today, 'extension_start_time': False, 'time_extension': 0})
  287. if new_state == 'ok':
  288. data = {'extension_start_time': False, 'time_extension': 0}
  289. data['alert_start_time'] = False
  290. self.write(data)
  291. if new_state == 'unsubscribed' or new_state == 'resigning':
  292. # Remove worker from task_templates
  293. self.cooperator_id.sudo().write(
  294. {'subscribed_shift_ids': [(5, 0, 0)]})
  295. # Remove worker from supercoop in task_templates
  296. task_tpls = self.env['beesdoo.shift.template'].search(
  297. [('super_coop_id', 'in', self.cooperator_id.user_ids.ids)]
  298. )
  299. task_tpls.write({'super_coop_id': False})
  300. # Remove worker for future tasks (remove also supercoop)
  301. self.env['beesdoo.shift.shift'].sudo().unsubscribe_from_today(
  302. [self.cooperator_id.id], now=fields.Datetime.now()
  303. )
  304. def _change_counter(self, data):
  305. self.sc += data.get('sc', 0)
  306. self.sr += data.get('sr', 0)
  307. self.irregular_absence_counter += data.get('irregular_absence_counter', 0)
  308. self.irregular_absence_date = data.get('irregular_absence_date', False)
  309. @api.multi
  310. def _write(self, vals):
  311. """
  312. Overwrite write to historize the change of status
  313. and make action on status change
  314. """
  315. if 'status' in vals:
  316. self._cr.execute('select id, status, sr, sc from "%s" where id in %%s' % self._table, (self._ids,))
  317. result = self._cr.dictfetchall()
  318. old_status_per_id = {r['id'] : r for r in result}
  319. for rec in self:
  320. if old_status_per_id[rec.id]['status'] != vals['status']:
  321. data = {
  322. 'status_id': rec.id,
  323. 'cooperator_id': rec.cooperator_id.id,
  324. 'type': 'status',
  325. 'change': "STATUS: %s -> %s" % (old_status_per_id[rec.id]['status'], vals['status']),
  326. 'user_id': self.env.context.get('real_uid', self.env.uid),
  327. }
  328. self.env['cooperative.status.history'].sudo().create(data)
  329. rec._state_change(vals['status'])
  330. return super(CooperativeStatus, self)._write(vals)
  331. _sql_constraints = [
  332. ('cooperator_uniq', 'unique (cooperator_id)', _('You can only set one cooperator status per cooperator')),
  333. ]
  334. @api.model
  335. def _set_today(self):
  336. """
  337. Method call by the cron to update store value base on the date
  338. """
  339. self.search([]).write({'today': fields.Date.today()})
  340. @api.multi
  341. def clear_history(self):
  342. self.ensure_one()
  343. self.history_ids.unlink()
  344. @api.model
  345. def _cron_compute_counter_irregular(self, today=False):
  346. today = today or fields.Date.today()
  347. journal = self.env['beesdoo.shift.journal'].search([('date', '=', today)])
  348. if not journal:
  349. journal = self.env['beesdoo.shift.journal'].create({'date': today})
  350. domain = ['&',
  351. '&',
  352. '&', ('status', '!=', 'unsubscribed'),
  353. ('working_mode', '=', 'irregular'),
  354. ('irregular_start_date', '!=', False),
  355. '|',
  356. '|', ('holiday_start_time', '=', False), ('holiday_end_time', '=', False),
  357. '|', ('holiday_start_time', '>', today), ('holiday_end_time', '<', today),
  358. ]
  359. irregular = self.search(domain)
  360. today_date = fields.Date.from_string(today)
  361. for status in irregular:
  362. if status.status == 'exempted':
  363. continue
  364. delta = (today_date - fields.Date.from_string(status.irregular_start_date)).days
  365. if delta and delta % PERIOD == 0 and status not in journal.line_ids:
  366. if status.sr > 0:
  367. status.sr -= 1
  368. elif status.alert_start_time:
  369. status.sr -= 1
  370. else:
  371. status.sr -= 2
  372. journal.line_ids |= status
  373. class ShiftCronJournal(models.Model):
  374. _name = 'beesdoo.shift.journal'
  375. _order = 'date desc'
  376. _rec_name = 'date'
  377. date = fields.Date()
  378. line_ids = fields.Many2many('cooperative.status')
  379. _sql_constraints = [
  380. ('one_entry_per_day', 'unique (date)', _('You can only create one journal per day')),
  381. ]
  382. @api.multi
  383. def run(self):
  384. self.ensure_one()
  385. if not self.user_has_groups('beesdoo_shift.group_cooperative_admin'):
  386. raise ValidationError(_("You don't have the access to perform this action"))
  387. self.sudo().env['cooperative.status']._cron_compute_counter_irregular(today=self.date)
  388. class ResPartner(models.Model):
  389. _inherit = 'res.partner'
  390. cooperative_status_ids = fields.One2many('cooperative.status', 'cooperator_id', readonly=True)
  391. super = fields.Boolean(related='cooperative_status_ids.super', string="Super Cooperative", readonly=True, store=True)
  392. info_session = fields.Boolean(related='cooperative_status_ids.info_session', string='Information Session ?', readonly=True, store=True)
  393. info_session_date = fields.Datetime(related='cooperative_status_ids.info_session_date', string='Information Session Date', readonly=True, store=True)
  394. working_mode = fields.Selection(related='cooperative_status_ids.working_mode', readonly=True, store=True)
  395. exempt_reason_id = fields.Many2one(related='cooperative_status_ids.exempt_reason_id', readonly=True, store=True)
  396. state = fields.Selection(related='cooperative_status_ids.status', readonly=True, store=True)
  397. extension_start_time = fields.Date(related='cooperative_status_ids.extension_start_time', string="Extension Start Day", readonly=True, store=True)
  398. subscribed_shift_ids = fields.Many2many('beesdoo.shift.template')
  399. @api.multi
  400. def coop_subscribe(self):
  401. return {
  402. 'name': _('Subscribe Cooperator'),
  403. 'type': 'ir.actions.act_window',
  404. 'view_type': 'form',
  405. 'view_mode': 'form',
  406. 'res_model': 'beesdoo.shift.subscribe',
  407. 'target': 'new',
  408. }
  409. @api.multi
  410. def coop_unsubscribe(self):
  411. res = self.coop_subscribe()
  412. res['context'] = {'default_unsubscribed': True}
  413. return res
  414. @api.multi
  415. def manual_extension(self):
  416. return {
  417. 'name': _('Manual Extension'),
  418. 'type': 'ir.actions.act_window',
  419. 'view_type': 'form',
  420. 'view_mode': 'form',
  421. 'res_model': 'beesdoo.shift.extension',
  422. 'target': 'new',
  423. }
  424. @api.multi
  425. def auto_extension(self):
  426. res = self.manual_extension()
  427. res['context'] = {'default_auto': True}
  428. res['name'] = _('Trigger Grace Delay')
  429. return res
  430. @api.multi
  431. def register_holiday(self):
  432. return {
  433. 'name': _('Register Holiday'),
  434. 'type': 'ir.actions.act_window',
  435. 'view_type': 'form',
  436. 'view_mode': 'form',
  437. 'res_model': 'beesdoo.shift.holiday',
  438. 'target': 'new',
  439. }
  440. @api.multi
  441. def temporary_exempt(self):
  442. return {
  443. 'name': _('Temporary Exemption'),
  444. 'type': 'ir.actions.act_window',
  445. 'view_type': 'form',
  446. 'view_mode': 'form',
  447. 'res_model': 'beesdoo.shift.temporary_exemption',
  448. 'target': 'new',
  449. }
  450. #TODO access right + vue on res.partner
  451. #TODO can_shop : Status can_shop ou extempted ou part C