New project_sla module

2025-02-13 17:57:24 +02:00 · 2013-12-19 11:17:18 +00:00
parent 3b31d34476
commit f52e44fdb4
18 changed files with 1446 additions and 0 deletions
--- a/project_sla/analytic_account.py
+++ b/project_sla/analytic_account.py
@@ -0,0 +1,71 @@
+# -*- coding: utf-8 -*-
+##############################################################################
+#
+#    Copyright (C) 2013 Daniel Reis
+#
+#    This program is free software: you can redistribute it and/or modify
+#    it under the terms of the GNU Affero General Public License as
+#    published by the Free Software Foundation, either version 3 of the
+#    License, or (at your option) any later version.
+#
+#    This program is distributed in the hope that it will be useful,
+#    but WITHOUT ANY WARRANTY; without even the implied warranty of
+#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+#    GNU Affero General Public License for more details.
+#
+#    You should have received a copy of the GNU Affero General Public License
+#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+##############################################################################
+
+from openerp.osv import fields, orm
+import logging
+_logger = logging.getLogger(__name__)
+
+
+class AnalyticAccount(orm.Model):
+    """ Add SLA to Analytic Accounts """
+    _inherit = 'account.analytic.account'
+    _columns = {
+        'sla_ids': fields.many2many(
+            'project.sla', string='Service Level Agreement'),
+        }
+
+    def _reapply_sla(self, cr, uid, ids, recalc_closed=False, context=None):
+        """
+        Force SLA recalculation on open documents that already are subject to
+        this SLA Definition.
+        To use after changing a Contract SLA or it's Definitions.
+        The ``recalc_closed`` flag allows to also recompute closed documents.
+        """
+        ctrl_obj = self.pool.get('project.sla.control')
+        proj_obj = self.pool.get('project.project')
+        exclude_states = ['cancelled'] + (not recalc_closed and ['done'] or [])
+        for contract in self.browse(cr, uid, ids, context=context):
+            # for each contract, and for each model under SLA control ...
+            for m_name in set([sla.control_model for sla in contract.sla_ids]):
+                model = self.pool.get(m_name)
+                doc_ids = []
+                if 'analytic_account_id' in model._columns:
+                    doc_ids += model.search(
+                        cr, uid,
+                        [('analytic_account_id', '=', contract.id),
+                         ('state', 'not in', exclude_states)],
+                        context=context)
+                if 'project_id' in model._columns:
+                    proj_ids = proj_obj.search(
+                        cr, uid, [('analytic_account_id', '=', contract.id)],
+                        context=context)
+                    doc_ids += model.search(
+                        cr, uid,
+                        [('project_id', 'in', proj_ids),
+                         ('state', 'not in', exclude_states)],
+                        context=context)
+                if doc_ids:
+                    docs = model.browse(cr, uid, doc_ids, context=context)
+                    ctrl_obj.store_sla_control(cr, uid, docs, context=context)
+        return True
+
+    def reapply_sla(self, cr, uid, ids, context=None):
+        """ Reapply SLAs button action """
+        return self._reapply_sla(cr, uid, ids, context=context)