Prevent shift registration if member is exempted at shift date

lacagette_addons/lacagette_membership/models/shift_template.py

@@ -13,6 +13,21 @@ class ShiftTemplate(models.Model):
     _inherit = 'shift.template'
 
     
+    def _will_member_be_still_exempted_at_date(self, partner, rec_date):
+        answer = False
+        if partner.cooperative_state == "exempted":
+            #_logger.debug("%s is exempted now, but what about at %s", str(partner.id), str(rec_date))
+            rec_date = datetime.strptime(rec_date, '%Y-%m-%d')
+            for l in partner.leave_ids:
+                #_logger.debug("leave start end : %s %s", str(l.start_date), str(l.stop_date))
+                if (rec_date >= datetime.strptime(l.start_date, '%Y-%m-%d') and
+                   rec_date <= datetime.strptime(l.stop_date, '%Y-%m-%d')):
+                   answer = True
+                   #_logger.debug("Partner will be still exempted")
+                
+        return answer
+
+
     @api.model
     def ng_create_shifts_from_template(self, instance=None, after=False, before=False):
         conf = self.env['ir.config_parameter']
@@ -87,20 +102,22 @@ class ShiftTemplate(models.Model):
                     for attendee in ticket.registration_ids:
                         state, strl_id = attendee._get_state(rec_date)
                         if state:
-                            vals = {
-                                'partner_id': attendee.partner_id.id,
-                                'user_ids': [(6, 0, template.user_ids.ids)],
-                                'state': state,
-                                'email': attendee.email,
-                                'phone': attendee.phone,
-                                'name': attendee.name,
-                                'shift_id': shift_id.id,
-                                'shift_ticket_id': ticket_id.id,
-                                'tmpl_reg_line_id': strl_id,
-                                'template_created': True,
-                            }
-                            self.env['shift.registration'].with_context(
-                                from_shift_template=True).create(vals)
+                            # only register if attendee is not exempted when the event will occur
+                            if self._will_member_be_still_exempted_at_date(attendee.partner_id, rec_date) is False:
+                                vals = {
+                                    'partner_id': attendee.partner_id.id,
+                                    'user_ids': [(6, 0, template.user_ids.ids)],
+                                    'state': state,
+                                    'email': attendee.email,
+                                    'phone': attendee.phone,
+                                    'name': attendee.name,
+                                    'shift_id': shift_id.id,
+                                    'shift_ticket_id': ticket_id.id,
+                                    'tmpl_reg_line_id': strl_id,
+                                    'template_created': True,
+                                }
+                                self.env['shift.registration'].with_context(
+                                    from_shift_template=True).create(vals)
 
 
     @api.model
@@ -115,9 +132,9 @@ class ShiftTemplate(models.Model):
                    ('week_list', '=', today.strftime('%a').upper()[0:2]),
                    ('last_shift_date', '<=', datetime.strftime(futur, "%Y-%m-%d"))]
         templates = self.env['shift.template'].search(filters)
-        # templates = self.env['shift.template'].search([('id', '=', 60)])
+        #templates = self.env['shift.template'].search([('id', '=', 123)])
         for template in templates:
-            # _logger.info("Shift template = %s", str(template))
+            #_logger.debug("Shift template = %s", str(template))
             self.ng_create_shifts_from_template( instance=template,
                 before=fields.Datetime.to_string(
                     datetime.today() + timedelta(days=shift_creation_days)))