Mercurial > sat_legacy_website

comparison sat_website/unicodecsv.py @ 60:0d20fb28c32e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
many small changes: - set opacity 0.85 to libervia screenshot on the main page - remove videos from the repository (stored on external ftp) - more details in the README concerning the i18n - settings can be overriden by an external file (for the stats) - add in basic features.html a button to experimental features - add a link to the SàT MUC in downloads section - add link to bugzilla + improve other links in dev/community - add + symbol for projects using standard blogging system - add ~ symbol for projects being no communication tool - save subscription form results to a text file - update press.py and links.py
author souliane <souliane@mailoo.org>
date Fri, 15 May 2015 17:15:40 +0200
parents
children
comparison
equal deleted inserted replaced
59:01738ae70f2d 60:0d20fb28c32e
1 #!/usr/bin/python
2 # -*- coding: utf-8 -*-
3 #
4 # CSV Reader and Writer
5 #
6 # This is a copy paste of the examples given in Python docs:
7 # https://docs.python.org/2/library/csv.html
8
9 import csv, codecs, cStringIO
10
11 class UTF8Recoder:
12 """
13 Iterator that reads an encoded stream and reencodes the input to UTF-8
14 """
15 def __init__(self, f, encoding):
16 self.reader = codecs.getreader(encoding)(f)
17
18 def __iter__(self):
19 return self
20
21 def next(self):
22 return self.reader.next().encode("utf-8")
23
24 class UnicodeReader:
25 """
26 A CSV reader which will iterate over lines in the CSV file "f",
27 which is encoded in the given encoding.
28 """
29
30 def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
31 f = UTF8Recoder(f, encoding)
32 self.reader = csv.reader(f, dialect=dialect, **kwds)
33
34 def next(self):
35 row = self.reader.next()
36 return [unicode(s, "utf-8") for s in row]
37
38 def __iter__(self):
39 return self
40
41 class UnicodeWriter:
42 """
43 A CSV writer which will write rows to CSV file "f",
44 which is encoded in the given encoding.
45 """
46
47 def __init__(self, f, dialect=csv.excel, encoding="utf-8", **kwds):
48 # Redirect output to a queue
49 self.queue = cStringIO.StringIO()
50 self.writer = csv.writer(self.queue, dialect=dialect, **kwds)
51 self.stream = f
52 self.encoder = codecs.getincrementalencoder(encoding)()
53
54 def writerow(self, row):
55 self.writer.writerow([s.encode("utf-8") for s in row])
56 # Fetch UTF-8 output from the queue ...
57 data = self.queue.getvalue()
58 data = data.decode("utf-8")
59 # ... and reencode it into the target encoding
60 data = self.encoder.encode(data)
61 # write to the target stream
62 self.stream.write(data)
63 # empty queue
64 self.queue.truncate(0)
65
66 def writerows(self, rows):
67 for row in rows:
68 self.writerow(row)