def participatable_food_activity(restaurant_id):
with thrift_client('ers') as ers:
result = ers.get_participatable_food_activity_ids(restaurant_id)
activity_ids = []
result['8'] = set(result.values()[0]) if len(result) else set()
for day in range(1, 8):
result.setdefault(day, [])
activity_ids.extend(result[day])
result['8'] &= set(result[day])
activities = food_act_base.mget(activity_ids)
activity_map = {activity.id: activity for activity in activities}
for day in range(1, 8):
day_activity_ids = result[day]
result[day] = [activity_map[_id] for _id in day_activity_ids]
week_activity_ids = result['8']
result['8'] = [activity_map[_id] for _id in week_activity_ids]
return result
def participatable_food_activity(restaurant_id):
with thrift_client('ers') as ers:
result = ers.get_participatable_food_activity_ids(restaurant_id)
activity_ids = []
result['8'] = set(result.values()[0]) if len(result) else set()
for day in range(1,8):
result.setdefault(day,[])
activity_ids.extend(result[day])
result['8'] &= set(result[day])
activities = food_act_base.mget(activity_ids)
activity_map = {activity.id:activity for activity in activities}
for day in range(1,8):
day_activity_ids = result[day]
result[day] = [activity_map[_id] for _id in day_activity_ids]
week_activity_ids = result['8']
result['8'] = [activity_map[_id] for _id in week_activity_ids]
return result
def mget(activity_ids, category_id):
ACTIVITY_CATEGORY = thirdparty_svc.ers.SubsidyConst()
if category_id == ACTIVITY_CATEGORY.CATEGORY_RESTAURANT_ACTIVITY:
return rst_activity.mget(activity_ids)
elif category_id == ACTIVITY_CATEGORY.CATEGORY_FOOD_ACTIVITY:
return food_activity.mget(activity_ids)
