Python程式設計解決“今天午飯吃什麼？”

import time
import random
food = ['雞','鴨','魚','肉','菜','蛋','面']
res = ['KFC','麥當勞','德克士','老娘舅','全聚德','達美樂']

def welcome():      #歡迎語
    print('歡迎使用“選擇困難症助手”！')
    time.sleep(1)
    print('今天中午吃什麼呢？\n1.随機食物\n2.随機餐廳\n3.自定義\n4.退出')

def print_choice(a):      #選擇
    print('系統幫您選擇了%s。您滿意麼？滿意請輸入1，不滿意請輸入2：'%a)

def end_choice(a):     #結束語
    print('您的選擇是%s,感謝使用！'%a)

def get_res():      #取得餐廳清單
    res_reserve = []
    print('請輸入您心儀的三家餐廳')
    for i in range(3):
        a = input('請輸入第%s家餐廳'%(i+1))
        res_reserve.append(a)
    return(res_reserve)

welcome()
main_choice = int(input('請選擇：'))
while main_choice == 1:
    food_try = random.choice(food)
    print_choice(food_try)
    a = int(input('請選擇'))
    if a == 1:
        end_choice(food_try)
        break

while main_choice == 2:
    res_try = random.choice(res)
    print_choice(res_try)
    a = int(input('請選擇'))
    if a == 1:
        end_choice(res_try)
        break

if main_choice == 3:
    res3 = get_res()
    while True:
        res_try = random.choice(res3)
        print_choice(res_try)
        a = int(input('請選擇'))
        if a == 1:
            end_choice(res_try)
            break

elif main_choice == 4:
    end_choice('“結束”')