-
Notifications
You must be signed in to change notification settings - Fork 0
/
main.py
96 lines (75 loc) · 2.93 KB
/
main.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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
import sys
import getopt
from datetime import datetime
from openpyxl import load_workbook
def main():
table = ''
path = ''
filename = ''
script = ''
options, remainder = getopt.getopt(sys.argv[1:], '', ['table=', 'path='])
for opt, arg in options:
if opt == '--table':
table = arg
if opt == '--path':
path = arg
# Read given xlsx
wb = load_workbook(filename = path)
ws = wb.active
row_count = ws.max_row
column_count = ws.max_column
#Create a filename for the script
now = datetime.now()
filename = table + "_" + now.strftime("%Y-%m-%d") + ".sql"
#Create a new file
f = open(filename, "w")
script += "INSERT INTO " + table
#Iterate through the xlsx file and create the SQL file
rows = ws.rows
for i, row in enumerate(rows, 1):
for j, cell in enumerate(row, 1):
#First row of the xlsx, table columns names
if i == 1:
if j == 1:
script += "(`" + str(cell.value) + "`, "
elif j == column_count:
script += "`" + str(cell.value) + "`) VALUES "
else:
script += "`"+ str(cell.value) + "`, "
#Last row of the xlsx, ; is needed at the end of the file
elif i == row_count:
if str(cell.value) == "NULL" or str(cell.value) == "null":
if j == 1:
script += "(" + str(cell.value) + ", "
elif j == column_count:
script += str(cell.value) + ");"
else:
script += str(cell.value) + ", "
else:
if j == 1:
script += "('" + str(cell.value) + "', "
elif j == column_count:
script += "'" + str(cell.value) + "');"
else:
script += "'" + str(cell.value) + "', "
#Normal rows
else:
if str(cell.value) == "NULL" or str(cell.value) == "null":
if j == 1:
script += "(" + str(cell.value) + ", "
elif j == column_count:
script += str(cell.value) + "), "
else:
script += str(cell.value) + ", "
else:
if j == 1:
script += "('" + str(cell.value) + "', "
elif j == column_count:
script += "'" + str(cell.value) + "'), "
else:
script += "'" + str(cell.value) + "', "
f.write(script)
f.close()
return 0
if __name__ == '__main__':
sys.exit(main())