def main():
shop_set = set()
data = getdata.get_shop()
for shop in data["hits"]["hits"]:
shop_set.add(shop["_id"])
for shop_id in shop_set:
print shop_id
p = multiprocessing.Process(target=apriori.mineAssosiationRule,args=(shop_id,))
p.start()
p.join(10)
if p.is_alive():
print shop_id+"is killed."
p.terminate()
p.join()
def init():
"Add food count to db"
shops = getdata.get_shop()
for shop in shops['hits']['hits']:
rid = shop['_id']
print rid
foods = {}
orders = getdata.get_order(rid)
for order in orders['hits']['hits']:
#print order
groups = order['_source']['detail']['group']
for group in groups:
for f in group:
key = f['id']
if foods.has_key(key):
foods[key]['count'] += f['quantity']
else:
foods[key] = {'fid': f['id'], 'count': f['quantity'], 'name': f['name'] }
#print foods
for key in foods:
# print foods[key]
f = Food(rid, foods[key]['fid'], foods[key]['name'], foods[key]['count'])
db.session.add(f)
db.session.commit()
