def prepData(self,restaurant_id): data = getdata.get_order(restaurant_id) orders = data["hits"]["hits"] for order in orders: for foods in order["_source"]["detail"]["group"]: self.numItems += 1 key = order["_source"]["order_id"] list = [] for i, item in enumerate(foods): food_id = item["id"] quantity = item["quantity"] # print key,food_id,quantity self.name[food_id] = item["name"] list.append(food_id) self.transList.append(list)
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()