1 files changed, 51 insertions, 0 deletions
diff --git a/ishtar_common/models.py b/ishtar_common/models.py
index 7362abc5c..f585623c2 100644
--- a/ishtar_common/models.py
+++ b/ishtar_common/models.py
@@ -2137,3 +2137,54 @@ class Town(models.Model):
         if settings.COUNTRY == "fr":
             return u"%s (%s)" % (self.name, self.numero_insee[:2])
         return self.name
+
+
+class OperationType(GeneralType):
+    order = models.IntegerField(_(u"Order"), default=1)
+    preventive = models.BooleanField(_(u"Is preventive"), default=True)
+
+    class Meta:
+        verbose_name = _(u"Operation type")
+        verbose_name_plural = _(u"Operation types")
+        ordering = ['-preventive', 'order', 'label']
+
+    @classmethod
+    def get_types(cls, dct={}, instances=False, exclude=[], empty_first=True,
+                  default=None):
+        tuples = []
+        dct['available'] = True
+        if not instances and empty_first and not default:
+            tuples.append(('', '--'))
+        if default:
+            try:
+                default = cls.objects.get(txt_idx=default)
+                tuples.append((default.pk, _(unicode(default))))
+            except cls.DoesNotExist:
+                pass
+        items = cls.objects.filter(**dct)
+        if default:
+            exclude.append(default.txt_idx)
+        if exclude:
+            items = items.exclude(txt_idx__in=exclude)
+        current_preventive, current_lst = None, None
+        for item in items.order_by(*cls._meta.ordering).all():
+            if not current_lst or item.preventive != current_preventive:
+                if current_lst:
+                    tuples.append(current_lst)
+                current_lst = [_(u"Preventive") if item.preventive else
+                               _(u"Research"), []]
+                current_preventive = item.preventive
+            current_lst[1].append((item.pk, _(unicode(item))))
+        if current_lst:
+            tuples.append(current_lst)
+        return tuples
+
+    @classmethod
+    def is_preventive(cls, ope_type_id, key=''):
+        try:
+            op_type = cls.objects.get(pk=ope_type_id)
+        except cls.DoesNotExist:
+            return False
+        if not key:
+            return op_type.preventive
+        return key == op_type.txt_idx