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

Timestamp:

24 May 2009, 10:12:58 (16 years ago)

Author:

uli

Message:

Add facultyimporter, use new faculty factory, fix container counter.

File:

• waeup/branches/ulif-rewrite/src/waeup/university/facultycontainer.py (modified) (5 diffs)

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

@@ -3 +3 @@
 from zope.component.factory import Factory
 from zope.component.interfaces import Invalid, IFactory
+from zope.component import createObject
 from zope.exceptions import DuplicationError
 from zope.interface import implementedBy
-from waeup.interfaces import IFacultyContainer, IFaculty
+from waeup.interfaces import IFacultyContainer, IFaculty, IWAeUPCSVImporter
+from waeup.utils.importexport import CSVImporter
 from waeup.viewlets import MainArea, LeftSidebar, Index, Add, FormWrapMixin
 
@@ -22 +24 @@
         id = str(self.next_id)
         self[id] = faculty
-        while self.next_id in self.keys():
+        while str(self.next_id) in self.keys():
             # Look for next unused int...
             self.next_id += 1
         return id
+
+    def clear(self):
+        keys = self.keys()
+        for key in keys:
+            del self[key]
+        next_id = 0L
 
 class FacultyContainerFactory(grok.GlobalUtility):
@@ -40 +48 @@
     def getInterfaces(self):
         return implementedBy(FacultyContainer)
+
+class FacultyCSVImporter(CSVImporter):
+    """Importer to import CSV data into faculty containers.
+    """
+    grok.context(FacultyContainer)
+    grok.implements(IWAeUPCSVImporter)
+    datatype = u'Faculty importer'
+    column_terms = ['code', 'title', 'title_prefix']
+
+    def doImport(self, filepath, clear_old_data=True, overwrite=True):
+        headers, data = self.getData(filepath)
+        if clear_old_data:
+            self.context.clear()
+        keys = self.context.keys()
+        for item in data:
+            data_dict = dict(map(None, headers, item))
+            new_item = createObject(u'waeup.Faculty')
+            for key, val in data_dict.items():
+               setattr(new_item, key, val)
+            self.context.addFaculty(new_item)
+        return
+
 
 #
@@ -67 +97 @@
     @grok.action('Add faculty')
     def addFaculty(self, **data):
-        faculty = getUtility(IFaculty)
+        faculty = createObject(u'waeup.Faculty')
         self.applyData(faculty, **data)
         try:
@@ -104 +134 @@
     def render(self):
         return u'<div class="portlet"><a href="add">Add faculty</a></div>'
+
+