Changeset 13926

Timestamp:

13 Jun 2016, 01:19:01 (9 years ago)

Author:

uli

Message:

Implement csv extractor.

This simple reader is not very robust.

File:

• main/waeup.kofa/branches/uli-scores-upload/src/waeup/kofa/students/browser.py (modified) (3 diffs)

main/waeup.kofa/branches/uli-scores-upload/src/waeup/kofa/students/browser.py

@@ -18 +18 @@
 """UI components for students and related components.
 """
+import csv
 import sys
 import grok
 import pytz
+from cStringIO import StringIO
 from urllib import urlencode
 from datetime import datetime
@@ -3104 +3106 @@
         return list(coursetickets)
 
+    def _extract_uploadfile(self, uploadfile):
+        """Get a mapping of student-ids to scores.
+
+        The mapping is constructed by reading contents from `uploadfile`.
+        """
+        data = StringIO(uploadfile.read())  # ensure we have something seekable
+        reader = csv.DictReader(data)
+        result = dict()
+        for row in reader:
+            if not 'student_id' in row or not 'score' in row:
+                continue
+            result[row['student_id']] = row['score']
+        return result
+
     def update(self,  *args, **kw):
         form = self.request.form
@@ -3129 +3145 @@
                 return
             formvals = dict(zip(form['sids'], form['scores']))
+            if form['uploadfile']:
+                formvals = self._extract_uploadfile(form['uploadfile'])
             for ticket in editable_tickets:
                 score = ticket.score
                 sid = ticket.student.student_id
+                if sid not in formvals:
+                    continue
                 if formvals[sid] == '':
                     score = None