[CHG] beesdoo_shift: compute next shifts

Author: MatteoLdx

beesdoo_shift/__manifest__.py

@@ -9,7 +9,7 @@
     "author": "Thibault Francois, Elouan Le Bars, Coop IT Easy SCRLfs",
     "website": "https://github.com/beescoop/Obeesdoo",
     "category": "Cooperative management",
-    "version": "12.0.1.1.4",
+    "version": "12.0.2.0.0",
     "depends": ["mail"],
     "data": [
         "data/system_parameter.xml",

beesdoo_shift/data/system_parameter.xml

@@ -31,4 +31,16 @@
         <field name="key">regular_counter_to_unsubscribe</field>
         <field name="value">-4</field>
     </record>
+    <record id="min_percentage_presence" model="ir.config_parameter">
+        <field name="key">beesdoo_shift.min_percentage_presence</field>
+        <field name="value">20</field>
+    </record>
+    <record id="regular_next_shift_limit" model="ir.config_parameter">
+        <field name="key">beesdoo_shift.regular_next_shift_limit</field>
+        <field name="value">7</field>
+    </record>
+    <record id="min_hours_to_unsubscribe" model="ir.config_parameter">
+        <field name="key">min_hours_to_unsubscribe</field>
+        <field name="value">2</field>
+    </record>
 </odoo>

beesdoo_shift/models/planning.py

@@ -106,12 +106,60 @@ def _generate_next_planning(self):
             _logger.error("Could not generate next planning: no task template defined.")
             return
 
-        planning.task_template_ids._generate_task_day()
+        planning.task_template_ids.generate_task_day()
 
         next_date = planning._get_next_planning_date(date)
         config.set_param("last_planning_seq", planning.sequence)
         config.set_param("next_planning_date", next_date)
 
+    @api.model
+    def get_future_shifts(self, end_date):
+        """
+        Calculates shifts between now and end_date without
+        storing them in the database
+        Uses a list of shifts instead of a recordset because
+        of issues occuring when copying records
+        :param end_date: Datetime
+        :return: beesdoo.shift.shift list
+        """
+        start_date = datetime.now()
+
+        shift_list = list(
+            self.env["beesdoo.shift.shift"]
+            .sudo()
+            .search(
+                [("start_time", ">", start_date.strftime("%Y-%m-%d %H:%M:%S"))],
+                order="start_time, task_template_id, task_type_id",
+            )
+        )
+
+        last_sequence = int(
+            self.env["ir.config_parameter"].sudo().get_param("last_planning_seq")
+        )
+
+        next_planning = self._get_next_planning(last_sequence)
+
+        next_planning_date = fields.Datetime.from_string(
+            self.env["ir.config_parameter"].sudo().get_param("next_planning_date", 0)
+        )
+
+        next_planning = next_planning.with_context(visualize_date=next_planning_date)
+
+        while next_planning_date < end_date:
+            for shift in next_planning.task_template_ids.get_task_day():
+                if shift.start_time > start_date:
+                    shift_list.append(shift)
+            next_planning_date = next_planning._get_next_planning_date(
+                next_planning_date
+            )
+            last_sequence = next_planning.sequence
+            next_planning = self._get_next_planning(last_sequence)
+            next_planning = next_planning.with_context(
+                visualize_date=next_planning_date
+            )
+
+        return shift_list
+
 
 class TaskTemplate(models.Model):
     _name = "beesdoo.shift.template"
@@ -144,7 +192,7 @@ class TaskTemplate(models.Model):
         domain=[("is_worker", "=", True)],
     )
     remaining_worker = fields.Integer(
-        compute="_compute_remaining", store=True, string="Remaining Spot"
+        compute="_compute_remaining", store=True, string="Remaining Place"
     )
     active = fields.Boolean(default=True)
     # For Kanban View Only
@@ -212,8 +260,12 @@ def _set_duration(self):
         if self.start_time:
             self.end_time = self.start_time + self.duration
 
-    def _generate_task_day(self):
-        tasks = self.env["beesdoo.shift.shift"]
+    def _prepare_task_day(self):
+        """
+        Generates a list of dict objects containing the informations
+        for the shifts to generate based on the template data
+        """
+        tasks = []
         for rec in self:
             for i in range(0, rec.worker_nb):
                 worker_id = rec.worker_ids[i] if len(rec.worker_ids) > i else False
@@ -234,7 +286,7 @@ def _generate_task_day(self):
                         and status.temporary_exempt_end_date >= rec.end_date.date()
                     ):
                         worker_id = False
-                tasks |= tasks.create(
+                tasks.append(
                     {
                         "name": "[%s] %s %s (%s - %s) [%s]"
                         % (
@@ -258,6 +310,34 @@ def _generate_task_day(self):
 
         return tasks
 
+    @api.multi
+    def get_task_day(self):
+        """
+        Creates the shifts according to the template without saving
+        them into the database.
+        To adapt the behaviour, function _prepare_task_day()
+        should be overwritten.
+        """
+        tasks = self.env["beesdoo.shift.shift"]
+        task_list = self._prepare_task_day()
+        for task in task_list:
+            tasks |= tasks.new(task)
+        return tasks
+
+    @api.multi
+    def generate_task_day(self):
+        """
+        Creates the shifts according to the template and saves
+        them into the database.
+        To adapt the behaviour, function _prepare_task_day()
+        should be overwritten.
+        """
+        tasks = self.env["beesdoo.shift.shift"]
+        task_list = self._prepare_task_day()
+        for task in task_list:
+            tasks |= tasks.create(task)
+        return tasks
+
     @api.onchange("worker_ids")
     def check_for_multiple_shifts(self):
         original_ids = {worker.id for worker in self._origin.worker_ids}

beesdoo_shift/models/res_partner.py

@@ -1,4 +1,4 @@
-from datetime import datetime
+from datetime import datetime, timedelta
 
 from odoo import _, api, fields, models
 
@@ -195,4 +195,26 @@ def _update_shifts_on_subscribed_task_tmpl(
                 now=datetime.now(),
             )
 
-    # TODO access right + vue on res.partner
+    def get_next_shifts(self):
+        """
+        Get the shifts of the user between now and an end_date,
+        with end_date = 4 weeks * regular_next_shift_limit
+        :return: beesdoo.shift.shift list
+        """
+        regular_next_shift_limit = int(
+            self.env["ir.config_parameter"]
+            .sudo()
+            .get_param("beesdoo_shift.regular_next_shift_limit")
+        )
+        nb_days = 28 * regular_next_shift_limit
+        start_date = datetime.now()
+        end_date = start_date + timedelta(days=nb_days)
+
+        shifts = self.env["beesdoo.shift.planning"].get_future_shifts(end_date)
+
+        next_shifts = []
+        for rec in shifts:
+            if rec.worker_id.id == self.id:
+                next_shifts.append(rec)
+
+        return next_shifts

beesdoo_shift/models/task.py

@@ -82,6 +82,7 @@ def _get_final_state(self):
     )
     revert_info = fields.Text(copy=False)
     working_mode = fields.Selection(related="worker_id.working_mode")
+    can_unsubscribe = fields.Boolean(compute="_compute_can_unsubscribe")
 
     def _expand_states(self, states, domain, order):
         return [key for key, val in self._fields["state"].selection(self)]
@@ -91,6 +92,25 @@ def _compute_color(self):
         for rec in self:
             rec.color = self._get_color_mapping(rec.state)
 
+    def _compute_can_unsubscribe(self):
+        now = datetime.now()
+        ICP = self.env["ir.config_parameter"].sudo()
+        min_hours = int(ICP.get_param("min_hours_to_unsubscribe", 2))
+        for rec in self:
+            if (
+                rec.start_time > now
+                and rec.state == "open"
+                and (
+                    (rec.worker_id and rec.worker_id.working_mode == "irregular")
+                    or rec.is_compensation
+                )
+            ):
+                delta = rec.start_time - now
+                delta = delta.seconds / 3600.0 + delta.days * 24
+                rec.can_unsubscribe = delta >= min_hours
+            else:
+                rec.can_unsubscribe = False
+
     @api.constrains("state")
     def _lock_future_task(self):
         if datetime.now() < self.start_time:

beesdoo_shift/wizard/instanciate_planning.py

@@ -17,7 +17,7 @@ def _get_planning(self):
     def generate_task(self):
         self.ensure_one()
         self = self.with_context(visualize_date=self.date_start, tracking_disable=True)
-        shifts = self.planning_id.task_template_ids._generate_task_day()
+        shifts = self.planning_id.task_template_ids.generate_task_day()
         return {
             "name": _("Generated Shift"),
             "type": "ir.actions.act_window",