写完这段程序后在密码验证处输入错误三次后没有结束程序而是继续执行输入用户名的指令,求帮我修改
"""
身份验证和验证码校验
其他功能:查询功能 输入验证码可检测且有次数限制
"""
#模块加载
import random
#用户信息设置
user_id = 'root','visitor'
user_pwd = '200401010725'
info_dict = {'yijie':1001,'bowen':1002}
#身份验证功能
def check_user(name,pwd):
if name in user_id and pwd == user_pwd:
print("=====用户身份验证通过=====")
return True
else:
print("用户名或者密码输入错误")
return False
#生成验证码
def random_code():
code = ""
for i in range(4):
num = random.randint(0,99)
if num % 3 == 0:
temp = chr(random.randint(97,122))
elif num % 3 == 1:
temp = str(random.randint(0,9))
else:
temp = chr(random.randint(65,90))
code += temp
return code
#验证码的检测和输入次数限制
def check_code(code):
in_code = input("当前验证码为%s,请输入:"%code)
if in_code == code:
return True
else:
return False
for i in range(3):#限制尝试次数为三次
rd_code =random_code()
if check_code(rd_code):
print("验证通过")
break
else:
print("验证不通过")
#查询功能
def check_info(name):
if name in info_dict:
print(info_dict[name])
else:
print("所查询用户不存在")
#验证功能是否正常
print("-----欢迎进入系统-----")
in_username = input("请输入用户名:")
in_useer_pwd = input("请输入密码:")
if check_user(in_username,in_userpwd):
num = eval(input("""
请选择功能:
1.查询功能
2.其他功能(未开放)
"""))
if num == 1:
code = random_code()
if check_code(code):
in_name = input("请输入查询人名:")
check_info(in_name)
else:
print("已退出")