[IMP] maintenance_location: Add inverse relation and equipment count to locations

Author: cgarcia-solvos

maintenance_location/models/maintenance_location.py

@@ -35,6 +35,55 @@ class MaintenanceLocation(models.Model):
     sequence = fields.Integer(default=10)
     active = fields.Boolean(default=True)
 
+    equipment_ids = fields.One2many(
+        "maintenance.equipment", "location_id", string="Equipments"
+    )
+
+    child_equipment_ids = fields.Many2many(
+        comodel_name="maintenance.equipment",
+        compute="_compute_child_equipment_ids",
+        string="Child Equipments",
+    )
+
+    equipment_count = fields.Integer(compute="_compute_equipment_count")
+
+    def _compute_child_equipment_ids(self):
+        all_locations = self.env["maintenance.location"].search(
+            [("id", "child_of", self.ids)]
+        )
+        all_equipments = self.env["maintenance.equipment"].search(
+            [("location_id", "in", all_locations.ids)]
+        )
+        for location in self:
+            descendant_locs = all_locations.filtered(
+                lambda sub_loc, loc=location: sub_loc.parent_path
+                and sub_loc.parent_path.startswith(loc.parent_path)
+                and sub_loc.id != loc.id
+            )
+            location.child_equipment_ids = all_equipments.filtered(
+                lambda eq, d_locs=descendant_locs: eq.location_id in d_locs
+            )
+
+    def _compute_equipment_count(self):
+        all_locations = self.env["maintenance.location"].search(
+            [("id", "child_of", self.ids)]
+        )
+        equip_data = self.env["maintenance.equipment"].read_group(
+            domain=[("location_id", "in", all_locations.ids)],
+            fields=["location_id"],
+            groupby=["location_id"],
+        )
+        count_dict = {x["location_id"][0]: x["location_id_count"] for x in equip_data}
+        for location in self:
+            descendant_ids = all_locations.filtered(
+                lambda sub_loc, loc=location: sub_loc.parent_path
+                and sub_loc.parent_path.startswith(loc.parent_path)
+            ).ids
+
+            location.equipment_count = sum(
+                count_dict.get(d_id, 0) for d_id in descendant_ids
+            )
+
     @api.depends("name", "parent_id.complete_name")
     def _compute_complete_name(self):
         for location in self:

maintenance_location/tests/test_maintenance_location.py

@@ -22,10 +22,18 @@ def setUpClass(cls):
             {"name": "Request", "maintenance_team_id": self.team.id}
         )
 
-        self.equipment = self.env["maintenance.equipment"].create(
+        self.env["maintenance.equipment"].create(
             {"name": "Laptop", "location_id": self.location_1.id}
         )
 
+        self.env["maintenance.equipment"].create(
+            {"name": "Printer", "location_id": self.location_1.id}
+        )
+
+        self.env["maintenance.equipment"].create(
+            {"name": "Scanner", "location_id": self.location_2.id}
+        )
+
         self.plan = self.env["maintenance.plan"].create(
             {
                 "equipment_id": self.equipment.id,
@@ -38,6 +46,8 @@ def setUpClass(cls):
             }
         )
 
+        self.env.invalidate_all()
+
     def test_maintenance_location(self):
         self.assertEqual(self.location_2.complete_name, "L1 / L2")
         with self.assertRaises(UserError):
@@ -51,3 +61,11 @@ def test_request_creation(self):
         self.assertTrue(request)
         for r in request:
             self.assertEqual(r.location_id.id, self.location_1.id)
+
+    def test_count_equipment(self):
+        self.assertEqual(self.location_1.equipment_count, 4)
+        self.assertEqual(self.location_2.equipment_count, 1)
+
+    def test_child_equipment_ids(self):
+        self.assertEqual(len(self.location_1.child_equipment_ids), 1)
+        self.assertEqual(len(self.location_2.child_equipment_ids), 0)

maintenance_location/views/maintenance_location.xml

@@ -15,13 +15,28 @@
                         </h1>
                     </div>
                     <group name="first" col="2">
-                        <field name="description" />
-                        <field name="parent_id" class="oe_inline" />
-                        <label for="latitude" string="Location" />
-                        <span class="oe_inline">
-                            Latitude: <field name="latitude" nolabel="1" /><br />
-                            Longitude: <field name="longitude" nolabel="1" />
-                        </span>
+                        <group>
+                            <field name="description" />
+                            <field name="parent_id" class="oe_inline" />
+                            <label for="latitude" string="Location" />
+                            <span class="oe_inline">
+                                Latitude: <field name="latitude" nolabel="1" /><br />
+                                Longitude: <field name="longitude" nolabel="1" />
+                            </span>
+                        </group>
+                        <group>
+                                <field
+                                name="equipment_ids"
+                                widget="many2many_tags"
+                                string="Equipments"
+                            />
+                                <field
+                                name="child_equipment_ids"
+                                widget="many2many_tags"
+                                string="Child Equipments"
+                            />
+                        </group>
+
                     </group>
                 </sheet>
             </form>
@@ -42,6 +57,7 @@
         <field name="arch" type="xml">
             <tree>
                 <field name="complete_name" />
+                <field name="equipment_count" />
             </tree>
         </field>
     </record>