import json
real_age = 18
prize_list = ['好迪洗发水', '绿箭侠', '小猪佩奇', '布娃娃', '再来一次!']
import random
user_prize_dict = {}
import os

def register():
  while True:
    username = input('输入用户名>(q退出):').strip().lower()
    if username=='q':break
    password = input('请输入密码>：').strip()
    re_password = input('请再次确认密码>:').strip()
    if not password == re_password:
      print('密码不一致，请重输！')
      continue
    user_dic = {'name': username, 'password': password}
    json_user_dic = json.dumps(user_dic)
    with open(f"{username}.txt", 'w', encoding='utf-8')as f:
      f.write(json_user_dic)
      f.flush()
      print('注册成功！')
      break

def login():
  count = 0
  while True:
    if count == 3:
      print('错误输入次数过多！')
      break
    username = input('请输入用户名>:').strip()
    if not os.path.exists(username + '.txt'):
      print('该用户不存在！')
      continue
    password = input('请输入密码>：').strip()
    with open(f"{username}.txt", 'r', encoding='utf-8') as f:
      user_json_dic = f.read()
      user_dic = json.loads(user_json_dic)
      if username == user_dic['name'] and password == user_dic['password']:
        print('登录成功！')
        guess_age()
        break
      else:
        print('用户名或密码错误！')
    count += 1

def guess_age():
  count = 0
  print('现在进入猜年龄游戏环节.......\n')
  while True:
    count += 1
    if count == 4:
      print('抱歉！你三次都猜错了！')
      again_guess_age = input('请问是否要继续猜3次（y继续，n退出）>:').strip().lower()
      if again_guess_age == 'y':
        count = 0
        continue
      break
    age = input('请输入你的年龄>：').strip()
    if not age.isdigit():
      print('请输入纯数字！')
      continue

    age = int(age)
    if age > real_age:
      print('猜大了！')
    elif age < real_age:
      print('猜小了！')
    else:
      print('恭喜你！猜对了！\n')
      choice_prize()
      break

def choice_prize():
  count = 1
  print('进入抽奖环节.....,您共有两次机会！\n 奖品如下：')
  while True:
    for index, prize in enumerate(prize_list, 1):
      print(index, prize)
    choice = input('请按下按钮y随机选择奖品>:').strip().lower()
    if not choice == 'y':
      print('非法输入！')
      continue
    prize_choice = random.randint(1, 15)
    if prize_choice in [6, 7, 8]:
      prize_choice = 4
    elif prize_choice in [9, 10, 11, 12, 13, 14, 15]:
      prize_choice = 5
    prize = prize_list[prize_choice - 1]
    if prize in user_prize_dict:
      user_prize_dict[prize] += 1
    else:
      user_prize_dict[prize] = 1
    print(f'本次获得奖品为：{prize},您还有{2-count}次机会！\n')
    if count == 2:
      if user_prize_dict.get('再来一次!'):
        user_prize_dict.pop('再来一次!')
      print(f'总共获得的奖品为：{user_prize_dict}')
      break
    count += 1


user_func_dic = {
  '1': register,
  '2': login,
}
while True:
  print('''
    先注册，登陆后才能玩猜年龄游戏哦！
    1. 注册
    2. 登录
  '''
     )
  choice = input('请选择功能编号(q退出)>:').strip().lower()
  if choice == 'q' : break
  if not choice in user_func_dic:
    print('错误输入')
    continue
  user_func_dic.get(choice)()