day10作业(函数实现注册''')

在猜年龄的基础上编写登录、注册方法，并且把猜年龄游戏分函数处理，如

1. 登录函数
2. 注册函数
3. 猜年龄函数
4. 选择奖品函数

'''在猜年龄的基础上编写登录、注册方法，并且把猜年龄游戏分函数处理，如

2. 登录函数
3. 注册函数
4. 猜年龄函数
5. 选择奖品函数'''
def register():
    '''登录'''
    count = 0
    while count < 3:
        user_inp = input('请输入用户名:').strip()  # 输入登录的用户名
        pwd_inp = input('请输入密码').strip()  # 输入登陆的密码


        with open('user_info.txt', 'r', encoding='utf8')as fw:
            # 1. 检查是否是未注册用户
            for i in fw:  # 逐行读入
                if user_inp == i.split(':')[0]:  # 用户名比较
                    break
            else:
                print('用户名不存在,请注册!')  # 如果没被打断说明无匹配用户名
                count = 4

            # 控制在未注册时退出循环
            if count== 4:
                break

            fw.seek(0)  # 控制文件光标回到文件头

            # 2.检查用户名和密码是否正确  (运行到这一步,说明已确定,文件中该用户名存在,即已注册)
            for line in fw:
                user, pwd = line.strip().split(':')
                if user_inp == user:
                    if pwd_inp != pwd:  # 如果用户名正确,密码错误,跳过,进行下一组用户名比对
                        continue
                    print('登陆成功!')  # 如果进行到这一步,说明,用户名\密码都正确,登录
                    count = 3  # 控制退出循环
                    guess_age()  # 进入到猜年龄函数
                    break
            else:  # 如果未被打断,说明文件中全都不匹配,即密码或用户名错误
                count += 1
                if count == 3:
                    print('输入错误3次,退出系统')
                    break
                print(f'用户名或密码错误,还剩下{3-count}次机会')


def login():
    '''注册'''
    while True:
        # 1. 注册输入用户名/密码
        user_login = input('请输入用户名:').strip()
        pwd_login = input('请输入密码:').strip()
        pwd1_login = input('请再次输入密码:').strip()

        # 2. 如果两次密码不一致,跳过,回到第一步,重新输入
        if not pwd_login == pwd1_login:
            print('两次密码不一致,请重新输入')
            continue

        # 3.打开文件,判断是否用户名存在,如果存在,提示,重新输入,否则直接写入文件
        with open('user_info.txt', 'a', encoding='utf8')as fa, \
                open('user_info.txt', 'r', encoding='utf8')as fr:
            for line in fr:
                if user_login == line.strip().split(':')[0]:
                    print('用户名已存在,请重新输入')
                    break
            else:  # 没有被打断,即用户名不是已注册过的
                with open('user_info.txt', 'a', encoding='utf8')as fw:
                    fw.write(f'{user_login}:{pwd_login}\n')
                    fw.flush()
                print('注册成功!')
                break


def guess_age():
    '''猜年龄'''
    # 1, 生成随机年龄
    import random
    age = str(int(random.random() * 100))

    while True:
        #1. 输入猜测年龄
        age_inp = input('请输入您猜的年龄').strip()

        # 2. 检测输入是否为数字.如果否,回到第一步
        if not age_inp.isdigit():
            continue

        # 3. 年龄判断
        if age_inp > age:
            print('猜大了')
        elif age_inp < age:
            print('猜小了')
        else:
            print('恭喜猜对了')
            choose_prize()  # 自动跳转选择奖品
            break


def choose_prize():
    '''选择奖品'''
    prize_dict = {'1': '95式突击步枪', '2': '120反坦克导弹', '3': '35榴', '4': '40火'}
    prize_show = '''
    奖品清单
    1    95式突击步枪
    2    120反坦克导弹
    3    35榴
    4    40火'''

    prize_count = 0  # 控制循环
    prize_car = {}  # 存储选择的奖品的清单
    while prize_count < 2:
        print(prize_show)

        # 1. 选择奖品
        chocie = input('请输入选择的奖品:').strip()

        # 2. 判断输入是否在字典的key,否,退回1
        if chocie not in prize_dict:
            print('输入错误,请重新输入')
            continue

        prize = prize_dict[chocie]  # 选择的奖品

        #3 .存储奖品清单
        if prize in prize_car:
            prize_car[prize] += 1
        else:
            prize_car[prize] = 1

        prize_count += 1
        if prize_count == 2:
            print(f'恭喜获得{prize_dict[chocie]}一把,选择结束')
            break
        print(f'恭喜获得{prize_dict[chocie]}一把,还可以选择{2-prize_count}次')

    # 4. 打印奖品清单
    print('共获得:')
    for key, i in prize_car.items():
        print(f'{key}{i}把')


tag = True
while True:
    print('''1  登录
2  注册''')

    choose = input('请选择:').strip()

    if choose == '1':  # 登录# 1. 检查是否是未注册用户
        register()

    if choose == '2':  # 注册
        login()