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()