[11.0][ADD] agreement_fsm module

agreement_fsm/models/agreement.py

@@ -0,0 +1,38 @@
+# Copyright (C) 2018 - TODAY, Open Source Integrators
+# License AGPL-3.0 or later (http://www.gnu.org/licenses/agpl).
+
+from odoo import fields, models, api
+
+
+class Agreement(models.Model):
+    _inherit = 'agreement'
+
+    service_order_count = fields.Integer(
+        compute='_compute_service_order_count',
+        string='# Service Orders'
+    )
+
+    @api.multi
+    def _compute_service_order_count(self):
+        for agreement in self:
+            res = self.env['fsm.order'].search_count(
+                [('agreement_id', '=', agreement.id)])
+            agreement.service_order_count = res or 0
+
+    @api.multi
+    def action_view_service_order(self):
+        for agreement in self:
+            fsm_order_ids = self.env['fsm.order'].search(
+                [('agreement_id', '=', agreement.id)])
+            action = self.env.ref(
+                'fieldservice.action_fsm_operation_order').read()[0]
+            if len(fsm_order_ids) > 1:
+                action['domain'] = [('id', 'in', fsm_order_ids.ids)]
+            elif len(fsm_order_ids) == 1:
+                action['views'] = [(
+                    self.env.ref('fieldservice.fsm_order_form').id,
+                    'form')]
+                action['res_id'] = fsm_order_ids.ids[0]
+            else:
+                action = {'type': 'ir.actions.act_window_close'}
+            return action