import numpy as np
import pandas as pd
from JsonToDF import get_data
from review import reviews_df
from review.review_score import get_reviews
from business import get_business_score
import pickle
count = 1000
df = get_data(count)
print df.shape
id_list = np.unique(reviews_df.business_id)
data = []
for _id in id_list:
_dict = {}
_dict['id'] = _id
_dict['name'] = np.unique(reviews_df[reviews_df.business_id == _id].name)[0]
print _id
reviews = get_reviews(_id)
try:
cat_score = get_business_score(reviews,_id)
_dict['cat_score'] = cat_score
data.append(_dict)
except:
pass
pickle.dump(data,open('data.p','w'))
