X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sla%2Fwsag_helper.py;fp=sla%2Fwsag_helper.py;h=2256ea9a327442e224ab4cad88b1397507ff76d4;hb=c99f1bd98ff355bd3f63b4929739ea5bd92cd6a3;hp=0000000000000000000000000000000000000000;hpb=e49329b43a640671476f2a793bde1d40b57e4d2e;p=myslice.git

diff --git a/sla/wsag_helper.py b/sla/wsag_helper.py
new file mode 100755
index 00000000..2256ea9a
--- /dev/null
+++ b/sla/wsag_helper.py
@@ -0,0 +1,116 @@
+import re
+import datetime
+
+from slaclient import wsag_model
+from slaclient.wsag_model import AgreementStatus
+from slaclient.wsag_model import Violation
+
+
+VIOLATED = AgreementStatus.StatusEnum.VIOLATED
+NON_DETERMINED = AgreementStatus.StatusEnum.NON_DETERMINED
+FULFILLED = AgreementStatus.StatusEnum.FULFILLED
+
+
+def get_violations_bydate(violations):
+    """Returns a list of violations per date, from a list of violations
+
+    :param violations list[Violation]:
+    :rtype: list
+    """
+    d = dict()
+    for v in violations:
+        assert isinstance(v, Violation)
+        date = v.datetime.date()
+        if not date in d:
+            d[date] = []
+        d[date].append(v)
+
+    result = [(key, d[key]) for key in sorted(d.keys(), reverse=True)]
+    return result
+
+
+class AgreementAnnotator(object):
+    """Annotates an agreement with the following attributes:
+
+    agreement.guaranteestatus
+    agreement.statusclass
+    agreement.guaranteeterms[*].status
+    agreement.guaranteeterms[*].statusclass
+    agreement.guaranteeterms[*].nviolations
+    agreement.guaranteeterms[*].servicelevelobjetive.bounds
+
+    """
+    def __init__(self):
+        pass
+
+    @staticmethod
+    def _get_statusclass(status):
+        if status is None or status == "" or status == NON_DETERMINED:
+            return "non-determined"
+        return "success" if status == FULFILLED else "error"
+
+    @staticmethod
+    def _parse_bounds(servicelevel):
+#         pattern = re.compile(".*BETWEEN *[(]?(.*), *([^)]*)[)]?")
+        pattern = re.compile(".*GT *([+-]?\\d*\\.\\d+)(?![-+0-9\\.])")
+        constraint = eval(servicelevel.strip(' \t\n\r'))
+        m = pattern.match(constraint['constraint'])
+        return m.groups()
+
+    def _annotate_guaranteeterm(self, term, violations):
+        #
+        # Annotate a guarantee term: set bounds and violations
+        #
+        level = term.servicelevelobjective.customservicelevel
+        bounds = AgreementAnnotator._parse_bounds(level)
+        term.servicelevelobjective.bounds = bounds
+
+        #
+        # set status attribute if not set before
+        #
+        if not hasattr(term, 'status'):
+            term.status = wsag_model.AgreementStatus.StatusEnum.NON_DETERMINED
+        #
+        # TODO: efficiency
+        #
+        n = 0
+        for violation in violations:
+            if violation.metric_name == term.servicelevelobjective.kpiname:
+                n += 1
+        term.nviolations = n
+
+    def _annotate_guaranteeterm_by_status(
+            self, agreement, termstatus, violations):
+        #
+        # Annotate a guarantee term: it is different from the previous
+        # one in that this takes the status into account.
+        #
+        name = termstatus.name
+        status = termstatus.status
+
+        term = agreement.guaranteeterms[name]
+        term.status = status
+        term.statusclass = AgreementAnnotator._get_statusclass(status)
+        self._annotate_guaranteeterm(term, violations)
+
+    def annotate_agreement(self, agreement, status=None, violations=()):
+
+        """Annotate an agreement with certain values needed in the templates
+
+        :param wsag_model.Agreement agreement: agreement to annotate
+        :param wsag_model.AgreementStatus status: status of the agreement.
+        :param violations: list of agreement's violations
+            (wsag_model.Violation[])
+        """
+        a = agreement
+
+        if status is not None:
+            a.guaranteestatus = status.guaranteestatus
+            a.statusclass = self._get_statusclass(status.guaranteestatus)
+            for termstatus in status.guaranteeterms:
+                self._annotate_guaranteeterm_by_status(
+                    agreement, termstatus, violations)
+        else:
+            a.guaranteestatus = NON_DETERMINED
+            for termname, term in agreement.guaranteeterms.items():
+                self._annotate_guaranteeterm(term, violations)