Mercurial > p > roundup > code

annotate roundup/rcsv.py @ 2204:10bb5de34eec maint-0.6

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
hyperlink designators in message display [SF#931828]
author Richard Jones <richard@users.sourceforge.net>
date Sat, 10 Apr 2004 22:10:25 +0000
parents 7b06875e3774
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
1778
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
1 """
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
2 Supplies a Python-2.3 Object Craft csv module work-alike to the extent
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
3 needed by Roundup using the Python 2.3 csv module.
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
4
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
5 """
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
6
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
7 from roundup.i18n import _
2071
3555941caea0 nicer message
Richard Jones <richard@users.sourceforge.net>
parents: 1778
diff changeset
8 error = """
3555941caea0 nicer message
Richard Jones <richard@users.sourceforge.net>
parents: 1778
diff changeset
9 Sorry, you need a csv module. Either upgrade your Python to 2.3 or later,
3555941caea0 nicer message
Richard Jones <richard@users.sourceforge.net>
parents: 1778
diff changeset
10 or get and install the csv module from:
1778
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
11 http://www.object-craft.com.au/projects/csv/
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
12 """
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
13 try:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
14 import csv
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
15 try:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
16 reader = csv.reader
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
17 writer = csv.writer
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
18 excel = csv.excel
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
19 error = ''
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
20 except AttributeError:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
21 # fake it all up using the Object-Craft CSV module
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
22 class excel:
2111
Richard Jones <richard@users.sourceforge.net>
parents: 2071
diff changeset
23 delimiter = ','
1778
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
24 if hasattr(csv, 'parser'):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
25 error = ''
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
26 def reader(fileobj, dialect=excel):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
27 # note real readers take an iterable but 2.1 doesn't
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
28 # support iterable access to file objects.
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
29 result = []
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
30 p = csv.parser(field_sep=dialect.delimiter)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
31
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
32 while 1:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
33 line = fileobj.readline()
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
34 if not line: break
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
35
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
36 # parse lines until we get a complete entry
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
37 while 1:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
38 fields = p.parse(line)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
39 if fields: break
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
40 line = fileobj.readline()
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
41 if not line:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
42 raise ValueError, "Unexpected EOF during CSV parse"
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
43 result.append(fields)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
44 return result
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
45 class writer:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
46 def __init__(self, fileobj, dialect=excel):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
47 self.fileobj = fileobj
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
48 self.p = csv.parser(field_sep = dialect.delimiter)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
49 def writerow(self, fields):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
50 print >>self.fileobj, self.p.join(fields)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
51 def writerows(self, rows):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
52 for fields in rows:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
53 print >>self.fileobj, self.p.join(fields)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
54
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
55 except ImportError:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
56 class excel:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
57 pass
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
58
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
59 class colon_separated(excel):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
60 delimiter = ':'
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
61 class comma_separated(excel):
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
62 delimiter = ','
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
63
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
64
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
65 if __name__ == "__main__":
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
66 f=open('testme.txt', 'r')
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
67 r = reader(f, colon_separated)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
68 remember = []
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
69 for record in r:
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
70 print record
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
71 remember.append(record)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
72 f.close()
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
73 import sys
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
74 w = writer(sys.stdout, colon_separated)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
75 w.writerows(remember)
d85cb1ad9bc9 merge from HEAD
Richard Jones <richard@users.sourceforge.net>
parents:
diff changeset
76
Roundup Issue Tracker: http://roundup-tracker.org/