def set_bd_restaurant_director(user_id, rst_ids, notice_enabled=True, in_charge=True): old_rst_dir = rst_dir_base.query_restaurant_director(director_ids=[user_id], offset=0, limit=1000) new_rst_dir = rst_dir_base.query_restaurant_director(restaurant_ids=rst_ids, offset=0, limit=1000) old_rst_ids = [rst_dir.restaurant_id for rst_dir in old_rst_dir] new_rst_ids = [rst_dir.restaurant_id for rst_dir in new_rst_dir] rm_rst_ids = _diff(old_rst_ids, new_rst_ids) add_rst_ids = _diff(new_rst_ids, old_rst_ids) rst_dir_base.rm_restaurant_director(user_id, rm_rst_ids) rst_dir_base.set_restaurant_director(user_id, add_rst_ids, notice_enabled, in_charge)
def set_bd_restaurant_director(user_id, rst_ids, notice_enabled=True, in_charge=True): old_rst_dir = rst_dir_base\ .query_restaurant_director(director_ids=[user_id,], offset=0, limit=1000) new_rst_dir = rst_dir_base\ .query_restaurant_director(restaurant_ids=rst_ids, offset=0, limit=1000) old_rst_ids = [rst_dir.restaurant_id for rst_dir in old_rst_dir] new_rst_ids = [rst_dir.restaurant_id for rst_dir in new_rst_dir] rm_rst_ids = _diff(old_rst_ids, new_rst_ids) add_rst_ids = _diff(new_rst_ids, old_rst_ids) rst_dir_base.rm_restaurant_director(user_id, rm_rst_ids) rst_dir_base.set_restaurant_director(user_id, add_rst_ids, notice_enabled, in_charge)
def set_notification(restaurant_ids, notice_enabled=None, in_charge=None): director_base.set_restaurant_director(current_user.id, restaurant_ids, notice_enabled, in_charge)