Changeset 9209 for main/waeup.kofa/branches/uli-zc-async/src/waeup/kofa/university/batching.py

Timestamp:

20 Sep 2012, 08:49:37 (12 years ago)

Author:

uli

Message:

Merge changes from update branch (includes trunk changes until r9107).

Location:

main/waeup.kofa/branches/uli-zc-async

Files:

• . (modified) (1 prop)
• src/waeup/kofa/university/batching.py (modified) (13 diffs)

main/waeup.kofa/branches/uli-zc-async/src/waeup/kofa/university/batching.py

@@ -29 +29 @@
 from zope.schema import getFields
 from zope.catalog.interfaces import ICatalog
-from waeup.kofa.interfaces import IBatchProcessor, IGNORE_MARKER, FatalCSVError
+from zope.event import notify
+from zope.securitypolicy.interfaces import (
+    IPrincipalRoleManager, IPrincipalRoleMap)
+from waeup.kofa.authentication import LocalRoleSetEvent
+from waeup.kofa.interfaces import (
+    IBatchProcessor, IGNORE_MARKER, DELETION_MARKER, FatalCSVError)
 from waeup.kofa.university.interfaces import (
     IFacultiesContainer, IFaculty, ICourse, IDepartment, ICertificate,
     ICertificateCourse)
+from waeup.kofa.university import (
+    Faculty, Department, Course, Certificate)
 from waeup.kofa.utils.batching import BatchProcessor
 
@@ -46 +53 @@
     name = u'Faculty Processor'
     iface = IFaculty
+    allowed_roles = Faculty.local_roles
 
     location_fields = ['code',]
@@ -54 +62 @@
     def parentsExist(self, row, site):
         return 'faculties' in site.keys()
+
+    @property
+    def available_fields(self):
+        fields = getFields(self.iface)
+        return sorted(list(set(
+            self.location_fields + fields.keys() + ['local_roles']
+            )))
 
     def entryExists(self, row, site):
@@ -77 +92 @@
         pass
 
-class DepartmentProcessor(BatchProcessor):
+    def updateEntry(self, obj, row, site):
+        """Update obj to the values given in row.
+        """
+        items_changed = ''
+
+        if row.has_key('local_roles') and row['local_roles'] not in (
+            None, IGNORE_MARKER):
+            role_manager = IPrincipalRoleManager(obj)
+            role_map = IPrincipalRoleMap(obj)
+            # Remove all existing local roles.
+            for local_role, user_name, setting in role_map.getPrincipalsAndRoles():
+                role_manager.unsetRoleForPrincipal(local_role, user_name)
+                notify(LocalRoleSetEvent(
+                        obj, local_role, user_name, granted=False))
+            # Add new local roles.
+            if row['local_roles'] != DELETION_MARKER:
+                local_roles = eval(row['local_roles'])
+                for rolemap in local_roles:
+                    user = rolemap['user_name']
+                    local_role = rolemap['local_role']
+                    role_manager.assignRoleToPrincipal(local_role, user)
+                    notify(LocalRoleSetEvent(obj, local_role, user, granted=True))
+                    items_changed += (
+                        '%s=%s, ' % ('local_roles', '%s|%s' % (user,local_role)))
+            row.pop('local_roles')
+
+        # apply other values...
+        items_changed += super(FacultyProcessor, self).updateEntry(
+            obj, row, site)
+
+        # Log actions...
+        location_field = self.location_fields[0]
+        grok.getSite().logger.info('%s - %s - Record updated: %s'
+            % (self.name, row[location_field], items_changed))
+        return items_changed
+
+    def checkConversion(self, row, mode='create'):
+        """Validates all values in row.
+        """
+        errs, inv_errs, conv_dict =  super(
+            FacultyProcessor, self).checkConversion(row, mode=mode)
+        if row.has_key('local_roles'):
+            if row['local_roles'] in (None, DELETION_MARKER, IGNORE_MARKER):
+                return errs, inv_errs, conv_dict
+            try:
+                local_roles = eval(row['local_roles'])
+            except:
+                errs.append(('local_roles','Error'))
+                return errs, inv_errs, conv_dict
+            if not isinstance(local_roles, list):
+                errs.append(('local_roles','no list'))
+                return errs, inv_errs, conv_dict
+            for rolemap in local_roles:
+                if not isinstance(rolemap, dict):
+                    errs.append(('local_roles','no dicts'))
+                    return errs, inv_errs, conv_dict
+                if not 'user_name' in rolemap.keys() or not \
+                    'local_role' in rolemap.keys():
+                    errs.append(('local_roles','user_name or local_role missing'))
+                    return errs, inv_errs, conv_dict
+                local_role = rolemap['local_role']
+                if not local_role in self.allowed_roles:
+                    errs.append(('local_roles','%s not allowed' % local_role))
+                    return errs, inv_errs, conv_dict
+                user = rolemap['user_name']
+                users = grok.getSite()['users']
+                if not user in users.keys():
+                    errs.append(('local_roles','%s does not exist' % user))
+                    return errs, inv_errs, conv_dict
+        return errs, inv_errs, conv_dict
+
+class DepartmentProcessor(FacultyProcessor):
     """A batch processor for IDepartment objects.
     """
@@ -88 +174 @@
     name = u'Department Processor'
     iface = IDepartment
+    allowed_roles = Department.local_roles
 
     location_fields = ['code', 'faculty_code']
@@ -124 +211 @@
         return
 
-class CourseProcessor(BatchProcessor):
+class CourseProcessor(FacultyProcessor):
     """A batch processor for ICourse objects.
     """
@@ -135 +222 @@
     name = u'Course Processor'
     iface = ICourse
+    allowed_roles = Course.local_roles
 
     location_fields = ['code', 'faculty_code', 'department_code']
@@ -175 +263 @@
         return
 
-class CertificateProcessor(BatchProcessor):
+class CertificateProcessor(FacultyProcessor):
     """A batch processor for ICertificate objects.
     """
@@ -186 +274 @@
     name = u'Certificate Processor'
     iface = ICertificate
+    allowed_roles = Certificate.local_roles
 
     location_fields = ['code']
@@ -196 +285 @@
         fields = getFields(self.iface)
         return sorted(list(set(
-            ['faculty_code','department_code'] + fields.keys())))
+            ['faculty_code','department_code'] + fields.keys()
+            + ['local_roles'])))
 
     def checkHeaders(self, headerfields, mode='create'):
@@ -269 +359 @@
         return
 
-class CertificateCourseProcessor(BatchProcessor):
+class CertificateCourseProcessor(FacultyProcessor):
     """A batch processor for ICertificateCourse objects.
     """
@@ -317 +407 @@
     def addEntry(self, obj, row, site):
         parent = self.getParent(row, site)
-        parent.addCourseRef(row['course'],
+        parent.addCertCourse(row['course'],
                             row['level'], row['mandatory'])
         return
@@ -323 +413 @@
     def delEntry(self, row, site):
         parent = self.getParent(row, site)
-        parent.delCourseRef(row['course'].code, row['level'])
-        return
+        parent.delCertCourse(row['course'].code, row['level'])
+        return