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

Author: cgarcia-solvos

maintenance_location/models/maintenance_location.py

@@ -35,6 +35,25 @@ 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(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

@@ -15,13 +15,23 @@
                         </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"
+                            />
+                        </group>
                     </group>
                 </sheet>
             </form>
@@ -42,6 +52,7 @@
         <field name="arch" type="xml">
             <tree>
                 <field name="complete_name" />
+                <field name="equipment_count" />
             </tree>
         </field>
     </record>