- initdb.py now deals with initial white spaces

Miguel Barão
1 parent b2e894fe
Showing 1 changed file with 2 additions and 1 deletions Show diff stats
initdb.py
@@ -20,6 +20,7 @@ def fix(name):
 #   Parse command line options
 argparser = argparse.ArgumentParser(description='Create new database from a CSV file (SIIUE format)')
 argparser.add_argument('--db', default='students.db', type=str, help='database filename')
+# argparser.add_argument('--demo', default='students.db', type=str, help='add demo students')
 argparser.add_argument('csvfile', nargs='?', type=str, default='', help='CSV filename')
 args = argparser.parse_args()
  
@@ -42,7 +43,7 @@ try:
     if args.csvfile:
         # from csv file if available
         try:
-            csvreader = csv.DictReader(open(args.csvfile, encoding='iso-8859-1'), delimiter=';', quotechar='"')
+            csvreader = csv.DictReader(open(args.csvfile, encoding='iso-8859-1'), delimiter=';', quotechar='"', skipinitialspace=True)
         except EnvironmentError:
             print('Error: CSV file "{0}" not found.'.format(args.csvfile))
             session.rollback()
...	...	@@ -20,6 +20,7 @@ def fix(name):
20	20	# Parse command line options
21	21	argparser = argparse.ArgumentParser(description='Create new database from a CSV file (SIIUE format)')
22	22	argparser.add_argument('--db', default='students.db', type=str, help='database filename')
	23	+# argparser.add_argument('--demo', default='students.db', type=str, help='add demo students')
23	24	argparser.add_argument('csvfile', nargs='?', type=str, default='', help='CSV filename')
24	25	args = argparser.parse_args()
25	26
...	...	@@ -42,7 +43,7 @@ try:
42	43	if args.csvfile:
43	44	# from csv file if available
44	45	try:
45		- csvreader = csv.DictReader(open(args.csvfile, encoding='iso-8859-1'), delimiter=';', quotechar='"')
	46	+ csvreader = csv.DictReader(open(args.csvfile, encoding='iso-8859-1'), delimiter=';', quotechar='"', skipinitialspace=True)
46	47	except EnvironmentError:
47	48	print('Error: CSV file "{0}" not found.'.format(args.csvfile))
48	49	session.rollback()
...	...