Changeset 6787 for main/waeup.sirp/trunk/src/waeup/sirp/students

Timestamp:

18 Sep 2011, 00:07:07 (13 years ago)

Author:

uli

Message:

Add contextual sources for reg_numbers and mat_numbers.

File:

• main/waeup.sirp/trunk/src/waeup/sirp/students/vocabularies.py (modified) (4 diffs)

main/waeup.sirp/trunk/src/waeup/sirp/students/vocabularies.py

@@ -2 +2 @@
 """
 from datetime import datetime
-from zope.component import getUtility
+from zope.component import getUtility, queryUtility
 from zope.catalog.interfaces import ICatalog
+from zope.interface import implements, directlyProvides
+from zope.schema.interfaces import ISource, IContextSourceBinder
+from zope.schema.interfaces import ValidationError
 from zc.sourcefactory.basic import BasicSourceFactory
 from zc.sourcefactory.contextual import BasicContextualSourceFactory
@@ -9 +12 @@
 from waeup.sirp.students.lgas import LGAS
 from waeup.sirp.university.vocabularies import course_levels
+
+
 
 def year_range():
@@ -33 +38 @@
     except AttributeError:
         return []
-    
+
 class StudyLevelSource(BasicContextualSourceFactory):
     """The StudyLevelSource is based on and extends the
@@ -130 +135 @@
         if value == 'f':
             return 'Female'
+
+class RegNumNotInSource(ValidationError):
+    """Registration number exists already
+    """
+    # The docstring of ValidationErrors is used as error description
+    # by zope.formlib.
+    pass
+
+class MatNumNotInSource(ValidationError):
+    """Matriculation number exists already
+    """
+    # The docstring of ValidationErrors is used as error description
+    # by zope.formlib.
+    pass
+
+class RegNumberSource(object):
+    implements(ISource)
+    cat_name = 'students_catalog'
+    field_name = 'reg_number'
+    validation_error = RegNumNotInSource
+    def __init__(self, context):
+        self.context = context
+        return
+
+    def __contains__(self, value):
+        cat = queryUtility(ICatalog, self.cat_name)
+        if cat is None:
+            return True
+        kw = {self.field_name: (value, value)}
+        results = cat.searchResults(**kw)
+        for entry in results:
+            if entry.student_id != self.context.student_id:
+                # XXX: sources should simply return False.
+                #      But then we get some stupid error message in forms
+                #      when validation fails.
+                raise self.validation_error(value)
+                #return False
+        return True
+
+def contextual_reg_num_source(context):
+    source = RegNumberSource(context)
+    return source
+directlyProvides(contextual_reg_num_source, IContextSourceBinder)
+
+class MatNumberSource(RegNumberSource):
+    field_name = 'matric_number'
+    validation_error = MatNumNotInSource
+
+def contextual_mat_num_source(context):
+    source = MatNumberSource(context)
+    return source
+directlyProvides(contextual_mat_num_source, IContextSourceBinder)