Changeset 6825 for main/waeup.sirp/trunk/src/waeup/sirp

Timestamp:

28 Sep 2011, 17:08:45 (13 years ago)

Author:

Henrik Bettermann

Message:

Add StudentStudyCourseProcessor?.

Location:

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

Files:

• batching.py (modified) (2 diffs)
• interfaces.py (modified) (1 diff)

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

@@ -9 +9 @@
 import grok
 from zope.interface import Interface
-from waeup.sirp.interfaces import IBatchProcessor
-from waeup.sirp.students.interfaces import IStudent
+from zope.schema import getFields
+from zope.component import queryUtility
+from zope.catalog.interfaces import ICatalog
+from waeup.sirp.interfaces import IBatchProcessor, FatalCSVError
+from waeup.sirp.students.interfaces import (
+    IStudent, IStudentStudyCourse, IStudentStudyCourseImport)
 from waeup.sirp.utils.batching import BatchProcessor
 
@@ -66 +70 @@
         del parent[row['student_id']]
         pass
+
+class StudentStudyCourseProcessor(BatchProcessor):
+    """A batch processor for IStudentStudyCourse objects.
+    """
+    grok.implements(IBatchProcessor)
+    grok.provides(IBatchProcessor)
+    grok.context(Interface)
+    util_name = 'studycourseimporter'
+    grok.name(util_name)
+
+    name = u'StudentStudyCourse Importer'
+    iface = IStudentStudyCourseImport
+    factory_name = 'waeup.StudentStudyCourse'
+
+    @property
+    def available_fields(self):
+        result = []
+        return sorted(list(set(
+            ['student_id','reg_number'] + getFields(self.iface).keys())))
+
+    def checkHeaders(self, headerfields, mode='update'):
+        if not 'reg_number' in headerfields or 'student_id' in headerfields:
+            raise FatalCSVError(
+                "Need at least columns student_id or reg_number for import!")
+        # Check for double fields...
+        not_ignored_fields = [x for x in headerfields
+                              if not x.startswith('--')]
+        if len(set(not_ignored_fields)) < len(not_ignored_fields):
+            raise FatalCSVError(
+                "Double headers: each column name may only appear once.")
+        return True
+
+    def parentsExist(self, row, site):
+        if not 'students' in site.keys():
+            return False
+        if 'student_id' in row.keys():
+            if row['student_id'] in site['students']:
+                student = site['students'][row['student_id']]
+                return student
+        else:
+            # Here we know that the reg_number is in row
+            reg_number = row['reg_number']
+            cat = queryUtility(ICatalog, name='students_catalog')
+            results = list(
+                cat.searchResults(reg_number=(reg_number, reg_number)))
+            if results:
+                return results[0]
+        return False
+
+    def entryExists(self, row, site):
+        student = self.parentsExist(row, site)
+        if not student:
+            return False
+        if 'studycourse' in student:
+            return student
+        return False
+
+    def getEntry(self, row, site):
+        student = self.entryExists(row, site)
+        if not student:
+            return None
+        return student.get('studycourse')

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

@@ -206 +206 @@
         )
 
+class IStudentStudyCourseImport(IStudentStudyCourse):
+    """A container for student study levels.
+
+    """
+
+    current_level = schema.Int(
+        title = u'Current Level',
+        default = None,
+        )
+
 class IStudentStudyLevel(IWAeUPObject):
     """A container for course tickets.