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

Author: cgarcia-solvos

maintenance_location/models/maintenance_location.py

@@ -35,6 +35,41 @@ 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="All Equipments (incl. sublocations)",
+    )
+
+    equipment_count = fields.Integer(compute="_compute_equipment_count")
+
+    def _compute_child_equipment_ids(self):
+        for location in self:
+            location.child_equipment_ids = self.env["maintenance.equipment"].search([
+                ("location_id", "child_of", location.id),
+                ("location_id", "!=", location.id)
+            ])
+
+    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 l: l.parent_path and l.parent_path.startswith(location.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

@@ -26,6 +26,10 @@ def setUpClass(cls):
             {"name": "Laptop", "location_id": self.location_1.id}
         )
 
+        self.equipment = self.env["maintenance.equipment"].create(
+            {"name": "Printer", "location_id": self.location_1.id}
+        )
+
         self.plan = self.env["maintenance.plan"].create(
             {
                 "equipment_id": self.equipment.id,
@@ -51,3 +55,7 @@ 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, 3)
+        self.assertEqual(self.location_2.equipment_count, 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>