-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmain.py
52 lines (39 loc) · 1.33 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
import sqlite3 as sql
def list_to_str(row):
ret = '('
for item in row:
if(type(item) == int):
ret += str(item)
elif(type(item) == str):
ret += '\'%s\'' % item
else:
ret += str(item)
ret += ','
return ret[:-1]+')'
if __name__ == '__main__':
source_path = str(input('Source browser profile address: '))
destination_path = str(input('Destination browser profile address: '))
conn1 = sql.connect(source_path)
c1 = conn1.cursor()
cursor = c1.execute('select * from keywords;')
output_table = [['Short Name', 'Keyword', 'URL']]
record_num = 0
insert_values = []
for row in cursor:
print([row[1], row[2], row[4]])
insert_values.append(list_to_str(row))
record_num += 1
conn1.close()
print('%d records in total.' % record_num)
insert_values_str = ''
for i in range(0, len(insert_values)-1):
insert_values_str += insert_values[i]+','
insert_values_str += insert_values[-1]
yn = input('Continue to import(Y/n)?')
if('n' not in yn and 'N' not in yn):
conn2 = sql.connect(destination_path)
c2 = conn2.cursor()
c2.execute('insert or ignore into keywords values %s;' % insert_values_str)
print('Insert succeeded!')
conn2.commit()
conn2.close()