def main(): print('Start running script: group_ungrouped_rsts') parser = argparse.ArgumentParser() parser.add_argument("city_id", help="city id", type=int) args = parser.parse_args() group_ungrouped_rsts(args.city_id) print('End of script: group_ungrouped_rsts')
def group_rsts(city_id): count = group_base.group_ungrouped_rsts(int(city_id)) return {'msg': 'Successfully group {} restaurants!'.format(count)}