Tests to assure that operation code is unique (closes #307)

1 files changed, 9 insertions, 0 deletions

diff --git a/ishtar/ishtar_base/models.py b/ishtar/ishtar_base/models.py
index 0cf15f072..6de782fcf 100644
--- a/ishtar/ishtar_base/models.py
+++ b/ishtar/ishtar_base/models.py
@@ -678,6 +678,15 @@ class Operation(BaseHistorizedItem, OwnPerms):
                                unicode(self.operation_code))))
         return JOINT.join(items)
 
+    def clean(self):
+        objs = self.__class__.objects.filter(year=self.year,
+                                             operation_code=self.operation_code)
+        if self.pk:
+            objs = objs.exclude(pk=self.pk)
+        if objs.count():
+            raise ValidationError(_(u"This operation code already exists for "
+                                    u"this year"))
+
     def is_own(self, person):
         return False