用python解决选择困难症

用python来解决“早上吃什么？中午吃什么？晚上吃什么？”的灵魂三问：

import random

#选择菜品
def chose_dish():
    dish = input('请输入可以选择的菜品，菜与菜之间用空格隔开：')
    dish_list = dish.split()
    choice_list = []
    num = int(input('你想吃几个菜？请输入数字：'))
    number = 0
    while True:
        choice = random.choice(dish_list)
        if choice not in choice_list:
            choice_list.append(choice)
            number += 1
        if number >=num:
            print('为你推荐以下菜品：',','.join(choice_list))
            sat = input('是否满意？满意请输入y，不满意请输入n:')
            if sat == 'y':
                break
            else:
                number = 0
                choice_list = []
                continue
    return choice_list

#选择饭店
def chose_restaurant():
    restaurant = input('请输入可以选择的饭店，饭店之间用空格隔开：')
    restaurant_list = restaurant.split()
    while True:
        choice = random.choice(restaurant_list)
        sat = input('为你推荐以下饭店:{}\n满意请输入y，不满意请输入n:'.format(choice))
        if sat == 'y':
            break
    return choice

def main():
    R_D = int(input('你想选择菜品还是饭店？选择菜品请输入1，选择饭店请输入2:'))
    if R_D == 1:
        dish_choices = chose_dish()
        print('选择结果：{}'.format(','.join(dish_choices)))
    else:
        restaurant_choice = chose_restaurant()
        print('选择结果：{}'.format(restaurant_choice))
        
if __name__ == '__main__':
    main()