forked from manasRK/word2vec-recommender
-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathredis_metadata_push.py
55 lines (43 loc) · 927 Bytes
/
redis_metadata_push.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
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author: Akhil Gupta
# @Date: 2016-06-28
# @Email: [email protected] Github username: @codeorbit
# @Last Modified by: Akhil Gupta
# @Last Modified time: 2016-06-28
import redis
import gzip
robj = redis.Redis("localhost", port=6379, db=3)
def redisPush(data):
product_id = data["asin"]
dic = {}
val_list = []
try:
title = data["title"]
val_list.append(title)
except:
pass
try:
price = data["price"]
val_list.append(price)
except:
pass
try:
category = data["salesRank"].keys()[0]
val_list.append(category)
except:
pass
try:
img_url = data["imUrl"]
val_list.append(img_url)
except:
pass
robj.set(product_id, val_list)
def parse(path):
g = gzip.open(path, 'r')
for l in g:
yield eval(l)
print 'Starting loading to Redis !'
for l in parse("metadata.json.gz"):
redisPush(l)
print 'Finished loading to Redis !'