Changeset 4268 for waeup/branches/ulif-rewrite/src

Timestamp:

12 Jun 2009, 09:57:07 (15 years ago)

Author:

uli

Message:

Add course importer.

File:

• waeup/branches/ulif-rewrite/src/waeup/university/facultycontainer.py (modified) (1 diff)

waeup/branches/ulif-rewrite/src/waeup/university/facultycontainer.py

@@ -116 +116 @@
                 setattr(new_item, key, val)
             faculty.addDepartment(new_item)
+        return
+
+class ICourseCSVFile(ICSVFile):
+    """A CSV file that contains course data.
+    """
+
+class CourseCSVFile(CSVFile):
+    """An abstraction of a CSV file containing courses.
+    """
+    grok.implements(ICourseCSVFile)
+    grok.provides(ICourseCSVFile)
+    required_fields = ['code', 'title', 'level', 'passmark', 'credits',
+                       'semester', 'faculty', 'department']
+
+class CourseCSVImporter(CSVImporter):
+    """Shuffle data from course CSV files into faculty containers.
+    """
+    # Tell, what kinds of objects we connect...
+    grok.adapts(ICourseCSVFile, IFacultyContainer)
+    # Tell the world, that we are an importer...
+    grok.implements(IWAeUPCSVImporter)
+    grok.provides(IWAeUPCSVImporter)
+
+    datatype = u'Course Importer'
+
+    def doImport(self, clear_old_data=True, overwrite=True):
+        # CSVImporter instances have a `csvfile` and a `receiver`
+        # object defined which refer to the CSV file and the container.
+        for row in self.csvfile.getData():
+            new_item = createObject(u'waeup.Course')
+            faculty_code = row['faculty']
+            faculty = self.receiver[faculty_code]
+            del row[faculty_code]
+            dept_code = row['department']
+            dept = faculty[dept_code]
+            del row[dept_code]
+            for key, val in row.items():
+                setattr(new_item, key, val)
+            dept.addCourse(new_item)
         return