Python就業班——Python函數與模塊——練習——yci_assistant生活小助理系統（慕K）

import random

chose_project = {1: '雙色球', 2: '號碼', 3: '天氣', 0: '退出'}


# 驗證輸入
def check_num(num):
    if num.isdigit():
        num = int(num)
        return num
    else:
        print('請輸入數字')


# 雙色球
def generate_lottery(num):
    for j in range(0, int(num)):
        lst1 = []
        # 紅球會出現重複數字
        # for i in range(0, 6):
        #     n = random.randint(1,33)
        #     lst1.append(n)
        # n = random.randint(1,16)
        # lst1.append(n)
        # 紅球不會出現重複數字
        nums_lst = random.sample(range(1, 33), 6)
        lst1.extend(nums_lst)
        nums_lst = random.randint(1, 16)
        lst1.append(nums_lst)
        print(lst1)


phone_numbers_str = '匪警[110],火警[119],急救[120]'


# 號碼
def find_phone(keyword):
    phone_lst = phone_numbers_str.split(',')
    result_lst = []
    for p in phone_lst:
        # if num in p:
        #     result_lst.append(p)
        if p.find(keyword) != -1:
            result_lst.append(p)
    return result_lst


weather_str = '北京,晴天,1℃=上海,下雨,4℃=廣州,下雨,4℃'


# 天氣
def get_weather(city):
    weather_dict = {}
    weather_lst = weather_str.split('=')
    for i in range(0, len(weather_lst)):
        w = weather_lst[i].split(',')
        weather = {'name': w[0], 'weather': w[1], 'max': w[2]}
        weather_dict[weather['name']] = weather
    if city in weather_dict:
        return weather_dict.get(city)
    else:
        return {}


while True:
    for k, v in chose_project.items():
        print(k, v)
    n = input('輸入：')
    n = check_num(n)
    if n == 1:
        n = input('隨機幾組號碼：')
        generate_lottery(n)
    elif n == 2:
        n = input('輸入要查詢的機構或號碼：')
        result = find_phone(keyword=n)
        if result:
            # print(result)
            for phone_num in result:
                print(phone_num)
        else:
            print('查詢不到')
    elif n == 3:
        n = input('輸入要查詢天氣的城市：')
        result = get_weather(n)
        if result:
            print('{name}，{weather}，{max}'.format_map(result))
        else:
            print('暫無{}天氣'.format(n))
    elif n == 0:
        print('退出')
        print('—' * 36)
        break
    else:
        print('請重新輸入')
    print('—' * 36)
print('感謝使用！')