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'
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)
32 while 1:
33 line = fileobj.readline()
34 if not line: break
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)
55 except ImportError:
56 class excel:
57 pass
59 class colon_separated(excel):
60 delimiter = ':'
61 class comma_separated(excel):
62 delimiter = ','
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)
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)