From 6ca1a82f45ae4f87f71152402b2255f03e2be04b Mon Sep 17 00:00:00 2001 From: Sebastian Reichel Date: Fri, 14 Sep 2012 20:56:07 +0200 Subject: update user import script to new csv data format --- invoice/import-users.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'invoice') diff --git a/invoice/import-users.py b/invoice/import-users.py index 096b071..4ed484c 100755 --- a/invoice/import-users.py +++ b/invoice/import-users.py @@ -2,11 +2,11 @@ import csv, sqlite3, sys title_to_gender = { - "Herr": "masculinum", - "Frau": "femininum" + "m": "masculinum", + "w": "femininum" } -data = csv.reader(open(sys.argv[1], 'r', encoding='iso-8859-1'), delimiter=';', quotechar='"') +data = csv.reader(open(sys.argv[1], 'r', encoding='utf-8'), delimiter=';', quotechar='"') connection = sqlite3.connect('shop.db') c = connection.cursor() @@ -15,8 +15,8 @@ data.__next__() for row in data: print(row) - gender = title_to_gender.get(row[2], "unknown") - t = (int(row[0]), row[1], row[3], row[4], gender, row[5], int(row[6]), row[7]) + gender = title_to_gender.get(row[7], "unknown") + t = (int(row[0]), row[1], row[2], row[3], gender, row[4], int(row[5]), row[6]) c.execute("INSERT OR REPLACE INTO users ('id', 'email', 'firstname', 'lastname', 'gender', 'street', 'plz', 'city') VALUES (?, ?, ?, ?, ?, ?, ?, ?);", t) connection.commit() -- cgit v1.2.3