veniasilente/veekun_pokedex
1
0
Fork 0
mirror of https://github.com/veekun/pokedex.git synced 2024-08-20 18:16:34 +00:00
Code Issues Projects Releases Wiki Activity

CSV import now respects NULLability of columns.

Browse source 
Empty strings loaded into NULL columns are changed to NULL instead.
This commit is contained in:
Eevee 2009-03-08 21:34:48 -04:00
parent 20c9c23f51
commit ac325b620d
1 changed files with 10 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pokedex/__init__.py
View file

@ -43,14 +43,20 @@ def csvimport(engine_uri, dir='.'):
print table_name print table_name
reader = csv.reader(open("%s/%s.csv" % (dir, table_name), 'rb'), lineterminator='\n') reader = csv.reader(open("%s/%s.csv" % (dir, table_name), 'rb'), lineterminator='\n')
columns = [unicode(column) for column in reader.next()] column_names = [unicode(column) for column in reader.next()]
for csvs in reader: for csvs in reader:
row = table() row = table()
for column, value in zip(columns, csvs): for column_name, value in zip(column_names, csvs):
if table.__table__.c[column_name].nullable and value == '':
# Empty string in a nullable column really means NULL
value = None
else:
# Otherwise, unflatten from bytes
value = value.decode('utf-8') value = value.decode('utf-8')
setattr(row, column, value)
setattr(row, column_name, value)
session.add(row) session.add(row)