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

Author: cgarcia-solvos

maintenance_location/models/maintenance_location.py

@@ -35,6 +35,28 @@ class MaintenanceLocation(models.Model):
     sequence = fields.Integer(default=10)
     active = fields.Boolean(default=True)
 
+    equipment_ids = fields.One2many(
+        "maintenance.equipment", "location_id", string="Equipments"
+    )
+
+    equipment_count = fields.Integer(
+        string="Equipment Count",
+        compute="_compute_equipment_count",
+    )
+
+    @api.depends("equipment_ids")
+    def _compute_equipment_count(self):
+        res = self.env["maintenance.equipment"].read_group(
+            domain=[("location_id", "in", self.ids)],
+            fields=["location_id"],
+            groupby=["location_id"],
+        )
+
+        res_dict = {x["location_id"][0]: x["location_id_count"] for x in res}
+
+        for location in self:
+            location.equipment_count = res_dict.get(location.id, 0)
+
     @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

@@ -22,6 +22,7 @@
                             Latitude: <field name="latitude" nolabel="1" /><br />
                             Longitude: <field name="longitude" nolabel="1" />
                         </span>
+                        <field name="equipment_ids" widget="many2many_tags" />
                     </group>
                 </sheet>
             </form>
@@ -42,6 +43,7 @@
         <field name="arch" type="xml">
             <tree>
                 <field name="complete_name" />
+                <field name="equipment_count" />
             </tree>
         </field>
     </record>