Commit df5d44ecfa6f350224df0b1964536eebe39d57a2
1 parent
b2e894fe
Exists in
master
and in
1 other branch
- initdb.py now deals with initial white spaces
Showing
1 changed file
with
2 additions
and
1 deletions
Show diff stats
initdb.py
| @@ -20,6 +20,7 @@ def fix(name): | @@ -20,6 +20,7 @@ def fix(name): | ||
| 20 | # Parse command line options | 20 | # Parse command line options |
| 21 | argparser = argparse.ArgumentParser(description='Create new database from a CSV file (SIIUE format)') | 21 | argparser = argparse.ArgumentParser(description='Create new database from a CSV file (SIIUE format)') |
| 22 | argparser.add_argument('--db', default='students.db', type=str, help='database filename') | 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 | argparser.add_argument('csvfile', nargs='?', type=str, default='', help='CSV filename') | 24 | argparser.add_argument('csvfile', nargs='?', type=str, default='', help='CSV filename') |
| 24 | args = argparser.parse_args() | 25 | args = argparser.parse_args() |
| 25 | 26 | ||
| @@ -42,7 +43,7 @@ try: | @@ -42,7 +43,7 @@ try: | ||
| 42 | if args.csvfile: | 43 | if args.csvfile: |
| 43 | # from csv file if available | 44 | # from csv file if available |
| 44 | try: | 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 | except EnvironmentError: | 47 | except EnvironmentError: |
| 47 | print('Error: CSV file "{0}" not found.'.format(args.csvfile)) | 48 | print('Error: CSV file "{0}" not found.'.format(args.csvfile)) |
| 48 | session.rollback() | 49 | session.rollback() |