add possibility to create a dialect for the csv file

This commit is contained in:
florian-dacosta
2014-12-10 18:44:29 +01:00
parent 0a20feecaa
commit 8bb62071f3
2 changed files with 11 additions and 5 deletions

View File

@@ -29,11 +29,15 @@ def UnicodeDictReader(utf8_data, **kwargs):
pos = utf8_data.tell()
sample_data = utf8_data.read(2048)
utf8_data.seek(pos)
dialect = sniffer.sniff(sample_data, delimiters=',;\t')
if not kwargs.get('dialect'):
dialect = sniffer.sniff(sample_data, delimiters=',;\t')
del kwargs['dialect']
else:
dialect = kwargs.pop('dialect')
csv_reader = csv.DictReader(utf8_data, dialect=dialect, **kwargs)
for row in csv_reader:
yield dict([(key, unicode(value, 'utf-8')) for key, value in
row.iteritems()])
yield dict([(unicode(key, 'utf-8'), unicode(value, 'utf-8'))
for key, value in row.iteritems() if key])
class BankStatementImportParser(object):