roundup/rcsv.py

   1 """Supplies a Python-2.3 Object Craft csv module work-alike to the extent
   2 needed by Roundup using the Python 2.3 csv module.
   3 """
   4 __docformat__ = 'restructuredtext'
   5
   6 from roundup.i18n import _
   7 from cStringIO import StringIO
   8 error = """
   9 Sorry, you need a csv module. Either upgrade your Python to 2.3 or later,
  10 or get and install the csv module from:
  11 http://www.object-craft.com.au/projects/csv/
  12 """
  13 try:
  14     import csv
  15     try:
  16         _reader = csv.reader
  17         writer = csv.writer
  18         excel = csv.excel
  19         error = ''
  20     except AttributeError:
  21         # fake it all up using the Object-Craft CSV module
  22         class excel:
  23             pass
  24         if hasattr(csv, 'parser'):
  25             error = ''
  26             def _reader(fileobj, dialect=excel):
  27                 # note real readers take an iterable but 2.1 doesn't
  28                 # support iterable access to file objects.
  29                 result = []
  30                 p = csv.parser(field_sep=dialect.delimiter)
  31
  32                 while 1:
  33                     line = fileobj.readline()
  34                     if not line: break
  35
  36                     # parse lines until we get a complete entry
  37                     while 1:
  38                         fields = p.parse(line)
  39                         if fields: break
  40                         line = fileobj.readline()
  41                         if not line:
  42                             raise ValueError, "Unexpected EOF during CSV parse"
  43                     result.append(fields)
  44                 return result
  45             class writer:
  46                 def __init__(self, fileobj, dialect=excel):
  47                     self.fileobj = fileobj
  48                     self.p = csv.parser(field_sep = dialect.delimiter)
  49                 def writerow(self, fields):
  50                     print >>self.fileobj, self.p.join(fields)
  51                 def writerows(self, rows):
  52                     for fields in rows:
  53                         print >>self.fileobj, self.p.join(fields)
  54
  55 except ImportError:
  56     class excel:
  57         pass
  58
  59 class colon_separated(excel):
  60     delimiter = ':'
  61 class comma_separated(excel):
  62     delimiter = ','
  63
  64 def reader(fileobject, dialect=excel):
  65     csv_lines = [line for line in fileobject.readlines() if line.strip()]
  66     return _reader(StringIO(''.join(csv_lines)), dialect)
  67
  68 if __name__ == "__main__":
  69     f=open('testme.txt', 'r')
  70     r = reader(f, colon_separated)
  71     remember = []
  72     for record in r:
  73         print record
  74         remember.append(record)
  75     f.close()
  76     import sys
  77     w = writer(sys.stdout, colon_separated)
  78     w.writerows(remember)
  79