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