Changeset 4158

Timestamp:

24 May 2009, 10:10:03 (16 years ago)

Author:

uli

Message:

Add a base for CSV imports.

File:

• waeup/branches/ulif-rewrite/src/waeup/utils/importexport.py (modified) (2 diffs)

waeup/branches/ulif-rewrite/src/waeup/utils/importexport.py

@@ -8 +8 @@
 from zope.interface import Interface
 from waeup.interfaces import (IWAeUPObject, IWAeUPExporter, IWAeUPXMLExporter,
-                              IWAeUPXMLImporter)
+                              IWAeUPXMLImporter, IWAeUPCSVImporter)
 
 def readFile(f):
@@ -86 +86 @@
         return obj
 
+class CSVImporter(grok.Adapter):
+    grok.context(IWAeUPObject)
+    grok.provides(IWAeUPCSVImporter)
+    grok.baseclass() # We don't want this base to be registered really.
+    column_terms = [] # The column terms supported by this importer
+    
+    def __init__(self, context):
+        self.context = context
+    
+    def doImport(self, filepath):
+        pass
+    
+    def getData(self, filepath):
+        result = readFile(open(filepath, 'rb'))
+        return result
+
+    def validate(self, filepath):
+        self.checkHeader(filepath)
+        self.getData(filepath)
+        return True
+
+    def checkHeader(self, filepath):
+        missing = []
+        headerline = ''
+
+        # Extract the header fields...
+        for line in open(filepath, 'rb'):
+            headerline = line
+            if len(headerline.strip()) == 0:
+                continue # Ignore leading empty lines...
+            break # We only want the first line...
+        tmpfile = StringIO(headerline)
+        headerfields = readFile(tmpfile)[0]
+
+        # Find missing header fields...
+        for key in self.column_terms:
+            if key not in headerfields:
+                missing.append(key)
+        if len(missing):
+            raise KeyError(
+                'Validation Error: the follwing '
+                'columns could not be found: %s' % (', '.join(missing))
+                )
+        return