-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathimporter.py
49 lines (38 loc) · 1.4 KB
/
importer.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
import csv
import mysql.connector
# data format example:
# "2348";"Meyer Hans";;"0049123456567";
source_csv = 'some_file.csv'
# database connection
cnx = mysql.connector.connect(user='root', database='geheim')
cursor = cnx.cursor()
with open(source_csv, newline='') as csvfile:
csv_data = csv.reader(csvfile, delimiter=';', quotechar='"')
for row in csv_data:
try:
# manipulate data
short_code = row[0]
name = row[1]
some_field = row[2]
phone_number = row[3]
query = ("""
INSERT INTO tablename
(short_code, name, some_field, phone_number)
VALUES
(%s, %s, %s, %s)
ON DUPLICATE KEY UPDATE
-- no need to update the PK
name = VALUES(%s),
some_field = VALUES(%s),
phone_number = VALUES(%s) ;
"""
)
data = (short_code, name, some_field, phone_number,
name, some_field, phone_number,)
cursor.execute(query, data)
cursor.close()
except mysql.connector.Error as err:
print("Something went wrong: {}".format(err))
cnx.rollback()
cnx.commit()
cnx.close()