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
# -*- coding: utf-8 -*-
"""
Created on Sun Sep 30 10:45:11 2018
@author: fangyucheng
"""
import pymysql
from crawler_sys.proxy_pool import make_up_sql
def write_dic_into_database(data_dic,
host='192.168.18.11',
user='root',
passwd='csm@1234',
database_name='proxy_pool',
table_name='proxy_pool'):
"""
write dict into mysql database
"""
connection = pymysql.connect(host=host, user=user, passwd=passwd,
db=database_name, port=3306,
cursorclass=pymysql.cursors.DictCursor)
cursor = connection.cursor()
write_into_sql = make_up_sql.make_up_replace_sql(table_name=table_name, input_dic=data_dic)
cursor.execute(write_into_sql)
connection.commit()
print('write %s into database' % data_dic['ip_address'])
def write_lst_into_database(data_lst,
host='192.168.18.11',
user='root',
passwd='csm@1234',
database_name='proxy_pool',
table_name='proxy_pool'):
"""
write list into mysql database
"""
connection = pymysql.connect(host=host, user=user, passwd=passwd,
db=database_name, port=3306,
cursorclass=pymysql.cursors.DictCursor)
cursor = connection.cursor()
length = len(data_lst)
for line in data_lst:
write_into_sql = make_up_sql.make_up_replace_sql(table_name=table_name, input_dic=line)
cursor.execute(write_into_sql)
connection.commit()
print('write %s pieces of data into database' % length)