python函数嵌套调用_登录验证码校验

python函数嵌套调用_登录验证码校验

import random
def generate_checkcode(n):
    code = ''
    str = 'abcdefghijklnmopqrstuvwxyzABCDEFGHIJKLNMOPQRSTUVWXYZ0123456789'
    for i in range(n):
        ran = random.randint(0, len(str) - 1)
        code += str[ran]
    return code


def login():
    while True:
        username = input('请输入用户名:')
        password = input('请输入密码:')
        code = generate_checkcode(4)
        print(f'验证码是：{code}')
        code1 = input('请输入接收到的验证码：')
        if code.upper() == code1.upper():
            if username == 'admin' and password == '123456':
                print('登录成功')
                break
            elif username != 'admin':
                print('用户名不存在！')
            else:
                print('密码错误')
        else:
            print('验证码错误')
login()