猜数游戏

import random


def guess(answer, times=None):
    """
    限定次数和无限次数二合一猜数字游戏

    参数 answer: int, 答案
    参数 times: int or None, 猜的次数，要大于0，None则一直猜到对
    """
    if times is None:
        count = 1
    else:
        count = times
    while count:
        try:
            num = int(input("请输入答案："))
        except:
            print("输入有误，请输入整数。")
            continue
        if num > answer:
            print("你输的值太大")
        elif num < answer:
            print("你输的值太小")
        else:
            if times is None:
                return "你猜了%d轮，终于成功啦！" % count
            else:
                return "恭喜你答对了！"
        if times is None:
            count += 1
        else:
            count -= 1
    return "游戏失败"


if __name__ == "__main__":
    answer = random.randint(1, 100)  # 随机答案
    # print(guess(answer, 5)) # 猜5次
    print(guess(answer))  # 猜到对为止