/
begin.py
70 lines (63 loc) · 2.01 KB
/
begin.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
from snapdealstuff import snapdeal
from flipkart import flipkart
from reviews import review
import MySQLdb
from csvthingy import links
def insert(product_id, skuid, reviews, score, user_id):
db = MySQLdb.connect("localhost", "root", "root", "reviews")
cursor = db.cursor()
try:
sql = "INSERT into product (product_id, skuid, reviews,score,user_id) values ('{0}',{1},{2},{3},{4})".format(product_id, skuid, repr(reviews),score,user_id)
# print sql
print reviews
cursor.execute(sql)
db.commit()
except MySQLdb.Error as e:
print e
db.rollback()
def complete(user_id):
db = MySQLdb.connect("localhost", "root", "root", "reviews")
cursor = db.cursor()
try:
sql = "UPDATE user SET status='complete' where user_id=" + str(user_id)
# print MySQLdb
cursor.execute(sql)
db.commit()
except MySQLdb.Error as e:
print e
db.rollback()
def clear(user_id):
db = MySQLdb.connect("localhost", "root", "root", "reviews")
cursor = db.cursor()
try:
sql = "DELETE from product where user_id={0}".format(user_id)
# print sql
cursor.execute(sql)
db.commit()
except MySQLdb.Error as e:
print e
db.rollback()
def start(filename):
user_id = 1
product_id = "1"
skuid = 1
reviews = "dsadas'dsd"
score = 1.1
clear(user_id)
# insert(product_id, skuid, reviews, score, user_id)
# for link in links('renka_product_url_fk_snapdeal.csv'):
for link in links(filename):
for rev in flipkart(link[1]):
product_id = rev[0]
skuid = link[0]
reviews = rev[1]
score = rev[2]
user_id = 1
insert(product_id, skuid, reviews, score, user_id)
for rev in snapdeal(link[2]):
product_id = rev[0]
skuid = link[0]
reviews = rev[1]
score = rev[2]
user_id = 1
insert(product_id, skuid, reviews, score, user_id)